Merge pull request #2640 from chatchat-space/hzg0601-patch-1

Update milvus_kb_service.py
This commit is contained in:
Zhi-guo Huang 2024-01-12 16:58:44 +08:00 committed by GitHub
commit 3e959e5b0f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -53,7 +53,7 @@ class MilvusKBService(KBService):
self.milvus = Milvus(embedding_function=EmbeddingsFunAdapter(self.embed_model),
collection_name=self.kb_name,
connection_args=kbs_config.get("milvus"),
index_params=kbs_config.ge("milvus_kwargs")["index_params"],
index_params=kbs_config.get("milvus_kwargs")["index_params"],
search_params=kbs_config.get("milvus_kwargs")["search_params"]
)