Adrien
commited on
Commit
·
c56c0fd
1
Parent(s):
9cb6543
fix
Browse files
rag_demo/rag/source_annotator.py
CHANGED
@@ -23,7 +23,7 @@ class SourceAnnotator:
|
|
23 |
for sentence in sentences:
|
24 |
scores = []
|
25 |
for chunk in reranked_chunks:
|
26 |
-
score = self.source_annotator(sentence
|
27 |
score["filename"] = chunk.metadata["filename"].split(".pdf")[0]
|
28 |
score["chunk_id"] = chunk.chunk_id
|
29 |
scores.append(score)
|
|
|
23 |
for sentence in sentences:
|
24 |
scores = []
|
25 |
for chunk in reranked_chunks:
|
26 |
+
score = self.source_annotator(sentence, chunk.content)
|
27 |
score["filename"] = chunk.metadata["filename"].split(".pdf")[0]
|
28 |
score["chunk_id"] = chunk.chunk_id
|
29 |
scores.append(score)
|