Update rag.py

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 -----------------------------