Adrien
commited on
Commit
·
9cb6543
1
Parent(s):
bccb279
fix: bypassing reranking for now
Browse files
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 |
|