From 86a902de9dd325644d1547d9fcfe4389300bac5b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?St=C3=A9phan?= Date: Mon, 19 May 2025 07:59:15 +0200 Subject: [PATCH] Update rag.py --- rag.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/rag.py b/rag.py index 7ab2bc0..1ed322c 100644 --- a/rag.py +++ b/rag.py @@ -52,7 +52,11 @@ for fp in sorted(ROOT.rglob("*")): # reconstruire docs dans le même ordre que l'index --------------------------- docs = [] for m in meta: - key = (m["file"], m["part"]) + # compatibilité avec l’ancien et le nouveau format + path = m.get("file") or m.get("path") # nouvelle clé : "path" + part = m["part"] + key = (Path(path).name, part) # on garde le nom court pour le cache + docs.append(cache.get(key, "[passage manquant]")) print(f"[dim]Passages rechargés : {len(docs)} (ordre conforme à l'index).[/]") @@ -104,6 +108,6 @@ try: print("\n[dim]--- contexte utilisé ---[/]") for rank, idx_id in enumerate(hits, 1): m = meta[int(idx_id)] - print(f"[{rank}] {m['file']} · part {m['part']} → {docs[int(idx_id)][:120]}…") + print(f"[{rank}] {Path(m.get('file') or m.get('path')).name} · part {m['part']} → …") except Exception as e: print("[red]Erreur :", e)