diff --git a/rag.py b/rag.py index 290c292..0df57a3 100644 --- a/rag.py +++ b/rag.py @@ -23,6 +23,8 @@ import faiss, numpy as np, requests from FlagEmbedding import BGEM3FlagModel from rich import print +ROOT="Corpus" + # ------------------------- CLI ------------------------------------------- p = argparse.ArgumentParser() p.add_argument("--index", nargs="*", default=["chap.idx"], @@ -58,7 +60,9 @@ print(f"Passages chargés : {total_passages} (agrégat de {len(indexes)} index)" DOCS: dict[int,str] = {} for base_offset, meta in zip(start_offset, metas): for i, m in enumerate(meta): - DOCS[base_offset + i] = Path(m["path"]).read_text(encoding="utf-8") + rel_path = m.get("path") or m.get("file") + full_path = ROOT / rel_path + DOCS[base_offset + i] = full_path.read_text(encoding="utf-8") print("[dim]Cache texte préchargé.[/]") # ------------------------- modèle embeddings -----------------------------