alihan/RAG_Techniques
1
0
Fork 0
mirror of https://github.com/NirDiamant/RAG_Techniques.git synced 2025-04-07 00:48:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
eliavs
2024-09-07 18:16:41 +03:00
parent 8c257e3f07
commit 9f123c707c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
all_rag_techniques_runnable_scripts/crag.py
View File

@@ -155,12 +155,12 @@ class CRAG:
max_score = max(eval_scores)
sources = []
if max_score > 0.7:
if max_score > self.upper_threshold:
print("\nAction: Correct - Using retrieved document")
best_doc = retrieved_docs[eval_scores.index(max_score)]
final_knowledge = best_doc
sources.append(("Retrieved document", ""))
elif max_score < 0.3:
elif max_score < self.lower_threshold:
print("\nAction: Incorrect - Performing web search")
final_knowledge, sources = self.perform_web_search(query)
else: