Update rag.py

This commit is contained in:
Stéphan Peccini 2025-05-19 07:59:15 +02:00
parent 4f61b37db1
commit 86a902de9d

8
rag.py
View File

@ -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 lancien 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)