From 22862043543e55fa0467c739714230eae3425512 Mon Sep 17 00:00:00 2001 From: Cameron Vetter Date: Sun, 18 Jun 2023 19:39:06 -0500 Subject: [PATCH] Correct AzureSearch Vector Store not applying search_kwargs when searching (#6132) Fixes #6131 Simply passes kwargs forward from similarity_search to helper functions so that search_kwargs are applied to search as originally intended. See bug for repro steps. #### Who can review? @hwchase17 @dev2049 Twitter: poshporcupine --- langchain/vectorstores/azuresearch.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/langchain/vectorstores/azuresearch.py b/langchain/vectorstores/azuresearch.py index 94ac1172..bb3bf3ce 100644 --- a/langchain/vectorstores/azuresearch.py +++ b/langchain/vectorstores/azuresearch.py @@ -243,11 +243,11 @@ class AzureSearch(VectorStore): ) -> List[Document]: search_type = kwargs.get("search_type", self.search_type) if search_type == "similarity": - docs = self.vector_search(query, k=k) + docs = self.vector_search(query, k=k, **kwargs) elif search_type == "hybrid": - docs = self.hybrid_search(query, k=k) + docs = self.hybrid_search(query, k=k, **kwargs) elif search_type == "semantic_hybrid": - docs = self.semantic_hybrid_search(query, k=k) + docs = self.semantic_hybrid_search(query, k=k, **kwargs) else: raise ValueError(f"search_type of {search_type} not allowed.") return docs