WizardCoder2007 commited on
Commit
99b1b18
·
1 Parent(s): a4fcb7d
Files changed (1) hide show
  1. server/utils/RetrieveQuery.py +1 -1
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[:8]:
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