FabNum/Code
4
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

Update rag.py

Browse Source
This commit is contained in:
Stéphan Peccini 2025-05-19 07:59:15 +02:00
parent 4f61b37db1
commit 86a902de9d
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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 --------------------------- # reconstruire docs dans le même ordre que l'index ---------------------------
docs = [] docs = []
for m in meta: 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]")) docs.append(cache.get(key, "[passage manquant]"))
print(f"[dim]Passages rechargés : {len(docs)} (ordre conforme à l'index).[/]") print(f"[dim]Passages rechargés : {len(docs)} (ordre conforme à l'index).[/]")
@ -104,6 +108,6 @@ try:
print("\n[dim]--- contexte utilisé ---[/]") print("\n[dim]--- contexte utilisé ---[/]")
for rank, idx_id in enumerate(hits, 1): for rank, idx_id in enumerate(hits, 1):
m = meta[int(idx_id)] 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: except Exception as e:
print("[red]Erreur :", e) print("[red]Erreur :", e)