Kshitijk20 commited on
Commit
da21908
·
1 Parent(s): 78f5d91

changes k value to 15

Browse files
Files changed (1) hide show
  1. app/retrieval/retriever.py +1 -1
app/retrieval/retriever.py CHANGED
@@ -13,7 +13,7 @@ class Retriever:
13
  self.llm = llm
14
  self.dense_retriever = self.vector_store.as_retriever(
15
  search_type="similarity",
16
- search_kwargs={"k": 5,"namespace": self.namespace, "filter": self.metadata}
17
  )
18
  self.hybrid_retriever = EnsembleRetriever(
19
  retrievers=[self.dense_retriever, sparse_retriever], # Use .retriever attribute
 
13
  self.llm = llm
14
  self.dense_retriever = self.vector_store.as_retriever(
15
  search_type="similarity",
16
+ search_kwargs={"k": 15,"namespace": self.namespace, "filter": self.metadata}
17
  )
18
  self.hybrid_retriever = EnsembleRetriever(
19
  retrievers=[self.dense_retriever, sparse_retriever], # Use .retriever attribute