From cc1e53008f88fd5230f498f2fae92af654f645c8 Mon Sep 17 00:00:00 2001
From: Cherilyn Buren <88433283+NiuBlibing@users.noreply.github.com>
Date: Tue, 17 Jun 2025 21:21:54 +0800
Subject: [PATCH] langchain: add missing milvus branch for self_query (#31630)

Fix #29603
---
 libs/langchain/langchain/retrievers/self_query/base.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/libs/langchain/langchain/retrievers/self_query/base.py b/libs/langchain/langchain/retrievers/self_query/base.py
index 801990d1859..42bebc50ab6 100644
--- a/libs/langchain/langchain/retrievers/self_query/base.py
+++ b/libs/langchain/langchain/retrievers/self_query/base.py
@@ -154,6 +154,14 @@ def _get_builtin_translator(vectorstore: VectorStore) -> Visitor:
             if isinstance(vectorstore, PineconeVectorStore):
                 return PineconeTranslator()
 
+        try:
+            from langchain_milvus import Milvus
+        except ImportError:
+            pass
+        else:
+            if isinstance(vectorstore, Milvus):
+                return MilvusTranslator()
+
         try:
             from langchain_mongodb import MongoDBAtlasVectorSearch
         except ImportError: