为milvus增加额外索引与搜索配置

2023-12-13 15:52:11 +08:00 · 2023-12-13 15:52:11 +08:00 · db008c1af8
parent 95c09981a6
commit db008c1af8
1 changed files with 5 additions and 1 deletions
--- a/server/knowledge_base/kb_service/milvus_kb_service.py
+++ b/server/knowledge_base/kb_service/milvus_kb_service.py
@ -48,7 +48,11 @@ class MilvusKBService(KBService):

    def _load_milvus(self):
        self.milvus = Milvus(embedding_function=EmbeddingsFunAdapter(self.embed_model),
-                             collection_name=self.kb_name, connection_args=kbs_config.get("milvus"))
+                             collection_name=self.kb_name, 
+                             connection_args=kbs_config.get("milvus"),
+                             index_params=kbs_config.ge("milvus_kwargs")["index_params"],
+                             search_params=kbs_config.get("milvus_kwargs")["search_params"]
+                             )

    def do_init(self):
        self._load_milvus()