Spaces:
Runtime error
Runtime error
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 |
|