diff --git a/.env.example b/.env.example index 2edb817..96e1193 100644 --- a/.env.example +++ b/.env.example @@ -13,6 +13,8 @@ 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_PASSWORD=xxxx # milvus数据库连接用户的密码 MILVUS_DIMENSION=2560 # milvus数据库向量维度 LLM_KEY_LOCAL=123 diff --git a/app/config/env.py b/app/config/env.py index dc388c1..af20fe5 100644 --- a/app/config/env.py +++ b/app/config/env.py @@ -21,6 +21,8 @@ class Settings(BaseSettings): 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") llm_key_local: str = Field(..., env="LLM_KEY_LOCAL") diff --git a/app/utils/milvus_utils.py b/app/utils/milvus_utils.py index 1a96f34..390b0e7 100644 --- a/app/utils/milvus_utils.py +++ b/app/utils/milvus_utils.py @@ -29,6 +29,8 @@ class MilvusService: similarity_metric="COSINE", consistency_level="Strong", overwrite=False, + user=env.milvus_username, + password=env.milvus_password ) return self._milvus_vector_store @@ -48,7 +50,7 @@ class MilvusService: parser = HierarchicalNodeParser.from_defaults(chunk_sizes=[2048, 512, 128]) # nodes = await parser.aget_nodes_from_documents(documents) # 使用同步方法并包装在 asyncio.to_thread 中 - nodes = await asyncio.to_thread(parser.get_nodes_from_documents,documents) + nodes = await asyncio.to_thread(parser.get_nodes_from_documents, documents) await vector_index.ainsert_nodes(nodes) return vector_index