From eaeae5f1f5f837603ec8c8d5c2790cbdf2491006 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?St=C3=A9phan?= <stephan-pro@peccini.fr>
Date: Mon, 19 May 2025 08:18:24 +0200
Subject: [PATCH] Update rag.py

---
 rag.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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