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 06:34:10 +02:00
parent 8c99fd2da3
commit f8c630cdfe
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
rag.py
View File

@ -17,7 +17,7 @@ from FlagEmbedding import BGEM3FlagModel
from rich import print from rich import print
ROOT = Path("Fiches") # dossier racine des fiches ROOT = Path("Fiches") # dossier racine des fiches
K = 4 # nombre de passages remis au LLM K = 10 # nombre de passages remis au LLM
# ------------------ utilitaires de découpe identiques à l'index ------------- # ------------------ utilitaires de découpe identiques à l'index -------------
CHUNK, OVERLAP = 800, 100 # garder cohérent avec index.py CHUNK, OVERLAP = 800, 100 # garder cohérent avec index.py
@ -80,8 +80,10 @@ try:
D, I = idx.search(q_emb.astype("float32").reshape(1, -1), K) D, I = idx.search(q_emb.astype("float32").reshape(1, -1), K)
hits = I[0] hits = I[0]
# réordonne pour mettre en tête les passages contenant “Seuil”
hits = sorted(hits, key=lambda i: "Seuil" not in docs[int(i)])
context = "\n\n".join(docs[int(i)] for i in hits) context = "\n\n".join(docs[int(i)] for i in hits[:K])
prompt = ( prompt = (
"<system>Réponds en français, de façon précise, et uniquement à partir du contexte fourni. Si l'information n'est pas dans le contexte, réponds : 'Je ne sais pas'.</system>\n" "<system>Réponds en français, de façon précise, et uniquement à partir du contexte fourni. Si l'information n'est pas dans le contexte, réponds : 'Je ne sais pas'.</system>\n"
f"<context>{context}</context>\n<user>{q}</user>" f"<context>{context}</context>\n<user>{q}</user>"