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

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,
-        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