Update rag.py
This commit is contained in:
parent
4f61b37db1
commit
86a902de9d
8
rag.py
8
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)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user