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
commit 67ac53242d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions

View File

@ -45,7 +45,7 @@ class MilvusKBService(KBService):
def do_drop_kb(self):
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
self._load_milvus(embeddings=embeddings)
return self.milvus.similarity_search_with_score(query, top_k)
@ -70,6 +70,7 @@ class MilvusKBService(KBService):
self.milvus.col.delete(expr=f'pk in {delete_list}')
def do_clear_vs(self):
if not self.milvus.col:
self.milvus.col.drop()