Spaces:
Sleeping
Sleeping
Commit ·
da21908
1
Parent(s): 78f5d91
changes k value to 15
Browse files
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":
|
| 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
|