feat: 连接milvus数据库的用户名和密码
This commit is contained in:
@@ -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")
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user