feat: 连接milvus数据库的用户名和密码

This commit is contained in:
martsforever
2025-09-07 13:40:46 +08:00
parent de360da4d3
commit 31993219f3
3 changed files with 7 additions and 1 deletions
+2
View File
@@ -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")
+3 -1
View File
@@ -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