From 375daa3ee824533bd8b9fe92f75e02818448721b Mon Sep 17 00:00:00 2001
From: martsforever <martsforever@aliyun.com>
Date: Sun, 7 Sep 2025 14:21:04 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=9B=B4=E6=94=B9milvus=E7=9B=B8?=
 =?UTF-8?q?=E5=85=B3=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F=E5=90=8D=E7=A7=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env.example              |  8 ++++----
 app/config/env.py         |  6 +++---
 app/utils/milvus_utils.py | 12 +++++++-----
 3 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/.env.example b/.env.example
index 96e1193..8be95bb 100644
--- a/.env.example
+++ b/.env.example
@@ -11,11 +11,11 @@ PG_DB_PASSWORD=xxx                      # postgres数据库连接密码
 PG_DB_DATABASE=xxx                      # postgres数据库连接的数据库名
 
 MILVUS_URI=http://xxxx:7323             # milvus数据库的连接uri
-MILVUS_COLLECTION=xxxx                  # milvus数据库集合名称
-MILVUS_DATABASE=xxxx                    # milvus数据库数据库名称
-MILVUS_USERNAME=xxxx                    # milvus数据库连接用户名
+MILVUS_USERNAME=root                    # milvus数据库连接用户名
 MILVUS_PASSWORD=xxxx                    # milvus数据库连接用户的密码
-MILVUS_DIMENSION=2560                   # milvus数据库向量维度
+LLAMA_INDEX_DATABASE=langgraph          # llama-index所使用的数据名称（需要提前创建）
+LLAMA_INDEX_COLLECTION=llama_index      # llama-index所使用的集合名称（不需要提前创建，会自动创建）
+LLAMA_INDEX_DIMENSION=1024              # llama-index所使用嵌入模型的向量维度
 
 LLM_KEY_LOCAL=123
 LLM_KEY_HUOSHAN=a0311f2a-ba85-4428-b158-xxxxxxxxxxxx 	# 火山引擎模型服务平台key
diff --git a/app/config/env.py b/app/config/env.py
index af20fe5..55029a5 100644
--- a/app/config/env.py
+++ b/app/config/env.py
@@ -19,11 +19,11 @@ class Settings(BaseSettings):
   pg_db_database: str = Field(..., env="PG_DB_DATABASE")
 
   milvus_uri: str = Field(..., env="MILVUS_URI")
-  milvus_collection: str = Field(..., env="MILVUS_COLLECTION")
-  milvus_database: str = Field(..., env="MILVUS_DATABASE")
   milvus_username: str = Field(..., env="MILVUS_USERNAME")
   milvus_password: str = Field(..., env="MILVUS_PASSWORD")
-  milvus_dimension: str = Field(..., env="MILVUS_DIMENSION")
+  llama_index_database: str = Field(..., env="LLAMA_INDEX_DATABASE")
+  llama_index_collection: str = Field(..., env="LLAMA_INDEX_COLLECTION")
+  llama_index_dimension: str = Field(..., env="LLAMA_INDEX_DIMENSION")
 
   llm_key_local: str = Field(..., env="LLM_KEY_LOCAL")
   llm_key_huoshan: str = Field(..., env="LLM_KEY_HUOSHAN")
diff --git a/app/utils/milvus_utils.py b/app/utils/milvus_utils.py
index 390b0e7..1e5d3b4 100644
--- a/app/utils/milvus_utils.py
+++ b/app/utils/milvus_utils.py
@@ -21,16 +21,18 @@ class MilvusService:
       print("create new_vector_store...")
       self._milvus_vector_store = MilvusVectorStore(
         uri=env.milvus_uri,
-        dim=env.milvus_dimension,
-        collection_name=env.milvus_collection,
-        db_name=env.milvus_database,
+        user=env.milvus_username,
+        password=env.milvus_password,
+
+        db_name=env.llama_index_database,
+        collection_name=env.llama_index_collection,
+        dim=env.llama_index_dimension,
+
         embedding_field="embedding",
         id_field="id",
         similarity_metric="COSINE",
         consistency_level="Strong",
         overwrite=False,
-        user=env.milvus_username,
-        password=env.milvus_password
       )
     return self._milvus_vector_store