Adrien commited on
Commit
9cb6543
·
1 Parent(s): bccb279

fix: bypassing reranking for now

Browse files
Files changed (1) hide show
  1. rag_demo/rag/retriever.py +2 -1
rag_demo/rag/retriever.py CHANGED
@@ -63,7 +63,8 @@ class RAGPipeline:
63
  logger.info(f"{len(n_k_documents)} documents retrieved successfully")
64
 
65
  if len(n_k_documents) > 0:
66
- k_documents = self.rerank(query, chunks=n_k_documents, keep_top_k=k)
 
67
  else:
68
  k_documents = []
69
 
 
63
  logger.info(f"{len(n_k_documents)} documents retrieved successfully")
64
 
65
  if len(n_k_documents) > 0:
66
+ # k_documents = self.rerank(query, chunks=n_k_documents, keep_top_k=k)
67
+ k_documents = n_k_documents[:k]
68
  else:
69
  k_documents = []
70