feat: 更改milvus相关环境变量名称

.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_USERNAME=root                    # milvus数据库连接用户名
-MILVUS_DATABASE=xxxx                    # milvus数据库数据库名称
-MILVUS_USERNAME=xxxx                    # 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

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")

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,
+        user=env.milvus_username,
-        collection_name=env.milvus_collection,
+        password=env.milvus_password,
-        db_name=env.milvus_database,
+
+        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