Spaces:
Runtime error
Runtime error
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 |
|