diff --git a/deployment/docker/datamate/docker-compose.yml b/deployment/docker/datamate/docker-compose.yml
index e2d97e41..1bb368b0 100644
--- a/deployment/docker/datamate/docker-compose.yml
+++ b/deployment/docker/datamate/docker-compose.yml
@@ -30,6 +30,7 @@ services:
       - log_level=DEBUG
       - pgsql_password=${DB_PASSWORD:-password}
       - datamate_jwt_enable=${DATAMATE_JWT_ENABLE:-false}
+      - milvus_uri=${MILVUS_URI:-http://milvus:19530}
     volumes:
       - dataset_volume:/dataset
       - flow_volume:/flow
diff --git a/deployment/helm/datamate/values.yaml b/deployment/helm/datamate/values.yaml
index a95bdf34..fb5044ef 100644
--- a/deployment/helm/datamate/values.yaml
+++ b/deployment/helm/datamate/values.yaml
@@ -140,6 +140,8 @@ backend-python:
           key: DB_PASSWORD
     - name: datamate_jwt_enable
       value: *DATAMATE_JWT_ENABLE
+    - name: milvus_uri
+      value: "http://milvus:19530"
   volumes:
     - *datasetVolume
     - *flowVolume
diff --git a/runtime/datamate-python/.env.example b/runtime/datamate-python/.env.example
index d1883839..2f95c434 100644
--- a/runtime/datamate-python/.env.example
+++ b/runtime/datamate-python/.env.example
@@ -20,3 +20,10 @@ LABEL_STUDIO_BASE_URL=http://localhost:30001
 LABEL_STUDIO_USER_TOKEN="demo_dev_token"
 
 DATAMATE_JWT_ENABLE=false
+
+# Milvus settings (Vector Database)
+# Development: use localhost
+MILVUS_URI=http://localhost:19530
+MILVUS_TOKEN=
+# Production: use service name
+# MILVUS_URI=http://milvus:19530
diff --git a/runtime/datamate-python/app/core/config.py b/runtime/datamate-python/app/core/config.py
index 49dd3320..77139ab6 100644
--- a/runtime/datamate-python/app/core/config.py
+++ b/runtime/datamate-python/app/core/config.py
@@ -78,7 +78,7 @@ def build_database_url(self):
     datamate_jwt_enable: bool = False
 
     # Milvus 配置
-    milvus_uri: str = "http://localhost:19530"
+    milvus_uri: str = "http://milvus:19530"
     milvus_token: str = ""
 
     # 文件存储配置（共享文件系统）