diff --git a/integrations/langchain/src/databricks_langchain/vectorstores.py b/integrations/langchain/src/databricks_langchain/vectorstores.py index 1f247f3cc..f433fdd7a 100644 --- a/integrations/langchain/src/databricks_langchain/vectorstores.py +++ b/integrations/langchain/src/databricks_langchain/vectorstores.py @@ -653,7 +653,7 @@ def max_marginal_relevance_search( query_vector = self._embeddings.embed_query(query) # type: ignore[union-attr] docs = self.max_marginal_relevance_search_by_vector( - query, + query_text, query_vector, k, fetch_k,