weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
hzg0601 2023-12-13 15:52:11 +08:00
parent 95c09981a6
commit db008c1af8
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/knowledge_base/kb_service/milvus_kb_service.py
View File

@ -48,7 +48,11 @@ class MilvusKBService(KBService):
def _load_milvus(self): def _load_milvus(self):
self.milvus = Milvus(embedding_function=EmbeddingsFunAdapter(self.embed_model), 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): def do_init(self):
self._load_milvus() self._load_milvus()