Adrien commited on
Commit
b7b8732
·
1 Parent(s): 9a276ad
Files changed (1) hide show
  1. rag_demo/rag/source_annotator.py +1 -1
rag_demo/rag/source_annotator.py CHANGED
@@ -31,7 +31,7 @@ class SourceAnnotator:
31
  # Could also use a score cut-off instead of max()
32
  max_score = max(scores, key=lambda x: x["score"])
33
 
34
- annotated_response += f"{sentence} [filename: {max_score['filename']}, chunk_id: {max_score['chunk_id']}, chunk_part_with_answer: {max_score['answer']}] "
35
 
36
  return annotated_response
37
 
 
31
  # Could also use a score cut-off instead of max()
32
  max_score = max(scores, key=lambda x: x["score"])
33
 
34
+ annotated_response += f"{sentence} [filename: {max_score['filename']}, chunk_id: {max_score['chunk_id']} "
35
 
36
  return annotated_response
37