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()
|