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

Merge pull request #1233 from Pickupppp/master 

修复 milvus 作为默认数据库初始化异常和调用search方法异常
This commit is contained in:
zqt996 2023-08-24 22:28:38 +08:00 committed by GitHub
parent bc3dbb443e 050a585a9d
commit 67ac53242d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -45,7 +45,7 @@ class MilvusKBService(KBService):
def do_drop_kb(self): def do_drop_kb(self):
self.milvus.col.drop() self.milvus.col.drop()
def do_search(self, query: str, top_k: int, embeddings: Embeddings): def do_search(self, query: str, top_k: int,score_threshold: float, embeddings: Embeddings):
# todo: support score threshold # todo: support score threshold
self._load_milvus(embeddings=embeddings) self._load_milvus(embeddings=embeddings)
return self.milvus.similarity_search_with_score(query, top_k) return self.milvus.similarity_search_with_score(query, top_k)
@ -70,7 +70,8 @@ class MilvusKBService(KBService):
self.milvus.col.delete(expr=f'pk in {delete_list}') self.milvus.col.delete(expr=f'pk in {delete_list}')
def do_clear_vs(self): def do_clear_vs(self):
self.milvus.col.drop() if not self.milvus.col:
self.milvus.col.drop()
if __name__ == '__main__': if __name__ == '__main__':