Add search_kwargs option for VectorDBQAWithSourcesChain (#657)

Allows for passing additional vectorstore params like namespace, etc. to
VectorDBQAWithSourcesChain

Example:
`chain = VectorDBQAWithSourcesChain.from_llm(OpenAI(temperature=0),
vectorstore=store, search_kwargs={"namespace": namespace})`
harrison/document-split
iocuydi 1 year ago committed by GitHub
parent bfb23f4608
commit 207e319a70
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -14,7 +14,12 @@ class VectorDBQAWithSourcesChain(BaseQAWithSourcesChain, BaseModel):
vectorstore: VectorStore
"""Vector Database to connect to."""
k: int = 4
"""Number of results to return from store"""
search_kwargs: Dict[str, Any] = {}
"""Extra search args"""
def _get_docs(self, inputs: Dict[str, Any]) -> List[Document]:
question = inputs[self.question_key]
return self.vectorstore.similarity_search(question, k=self.k)
return self.vectorstore.similarity_search(
question, k=self.k, **self.search_kwargs
)

Loading…
Cancel
Save