From 4476d93741c250a484f300b883c20ec61ade8810 Mon Sep 17 00:00:00 2001
From: Stuart Lynn <stuart.lynn@databricks.com>
Date: Wed, 15 Oct 2025 15:15:27 +0100
Subject: [PATCH] fixed the bug introduced for previous mmr fix

---
 integrations/langchain/src/databricks_langchain/vectorstores.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/integrations/langchain/src/databricks_langchain/vectorstores.py b/integrations/langchain/src/databricks_langchain/vectorstores.py
index 1f247f3c..f433fdd7 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,