Mishab commited on
Commit
0288eac
1 Parent(s): cc8fe19

Updated code

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -286,10 +286,10 @@ def load_retriver(chroma_vectorstore):
286
  return compression_retriever
287
 
288
 
289
- def load_retriver(text_chunks, embeddings, 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()
 
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()