Spaces:
Sleeping
Sleeping
Commit ·
99b1b18
1
Parent(s): a4fcb7d
update
Browse files
server/utils/RetrieveQuery.py
CHANGED
|
@@ -38,7 +38,7 @@ def QueryRetriever(queries,vector_retriever,keyword_retriever,chunks_dict):
|
|
| 38 |
# sort on basis of values
|
| 39 |
rrf_scores= sorted(rrf_scores.items(), key=lambda item: item[1], reverse=True)
|
| 40 |
# select top 10 documents
|
| 41 |
-
for chunk_id,score in rrf_scores[:
|
| 42 |
if chunk_id in chunks_dict:
|
| 43 |
documents.append(final_documents[chunk_id])
|
| 44 |
|
|
|
|
| 38 |
# sort on basis of values
|
| 39 |
rrf_scores= sorted(rrf_scores.items(), key=lambda item: item[1], reverse=True)
|
| 40 |
# select top 10 documents
|
| 41 |
+
for chunk_id,score in rrf_scores[:12]:
|
| 42 |
if chunk_id in chunks_dict:
|
| 43 |
documents.append(final_documents[chunk_id])
|
| 44 |
|