Adrien commited on
Commit
9a276ad
·
1 Parent(s): c56c0fd

feat: add chunk part with answer

Browse files
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']}] "
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']}, chunk_part_with_answer: {max_score['answer']}] "
35
 
36
  return annotated_response
37