Fix SelfQueryRetriever, passing new query to vector store (#4774)

# Fix SelfQueryRetriever, passing new query to vector store
1 year ago · 08876ad066
parent 8fd4d5d117
commit 08876ad066
1 changed files with 1 additions and 1 deletions
--- a/langchain/retrievers/self_query/base.py
+++ b/langchain/retrievers/self_query/base.py
@ -81,7 +81,7 @@ class SelfQueryRetriever(BaseRetriever, BaseModel):
            new_kwargs["k"] = structured_query.limit

        search_kwargs = {**self.search_kwargs, **new_kwargs}
-        docs = self.vectorstore.search(query, self.search_type, **search_kwargs)
+        docs = self.vectorstore.search(new_query, self.search_type, **search_kwargs)
        return docs

    async def aget_relevant_documents(self, query: str) -> List[Document]: