Spaces:
Runtime error
Runtime error
Updated code
Browse files
utils.py
CHANGED
|
@@ -286,10 +286,10 @@ def load_retriver(chroma_vectorstore):
|
|
| 286 |
return compression_retriever
|
| 287 |
|
| 288 |
|
| 289 |
-
def load_retriver(
|
| 290 |
"""Load cohere rerank method for retrieval"""
|
| 291 |
-
bm25_retriever = BM25Retriever.from_documents(text_chunks)
|
| 292 |
-
bm25_retriever.k = 2
|
| 293 |
chroma_retriever = chroma_vectorstore.as_retriever(search_kwargs={"k": 3})
|
| 294 |
# ensemble_retriever = EnsembleRetriever(retrievers=[bm25_retriever, chroma_retriever], weights=[0.3, 0.7])
|
| 295 |
logging.basicConfig()
|
|
|
|
| 286 |
return compression_retriever
|
| 287 |
|
| 288 |
|
| 289 |
+
def load_retriver(chroma_vectorstore):
|
| 290 |
"""Load cohere rerank method for retrieval"""
|
| 291 |
+
# bm25_retriever = BM25Retriever.from_documents(text_chunks)
|
| 292 |
+
# bm25_retriever.k = 2
|
| 293 |
chroma_retriever = chroma_vectorstore.as_retriever(search_kwargs={"k": 3})
|
| 294 |
# ensemble_retriever = EnsembleRetriever(retrievers=[bm25_retriever, chroma_retriever], weights=[0.3, 0.7])
|
| 295 |
logging.basicConfig()
|