Adrien
commited on
Commit
·
b7b8732
1
Parent(s):
9a276ad
fix
Browse files
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']} "
|
35 |
|
36 |
return annotated_response
|
37 |
|