github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-24 20:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(milvus): Fix milvus store search with topk always set to 4 (#2131)

Browse Source
This commit is contained in:
slp 2024-11-18 09:46:03 +08:00 committed by GitHub
parent 1d784c5b15
commit 2decf02c12
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dbgpt/storage/vector_store/milvus_store.py
View File

@ -458,7 +458,7 @@ class MilvusStore(VectorStoreBase):
# convert to milvus expr filter.
milvus_filter_expr = self.convert_metadata_filters(filters) if filters else None
_, docs_and_scores = self._search(
query=text, topk=topk, expr=milvus_filter_expr
query=text, k=topk, expr=milvus_filter_expr
)
if any(score < 0.0 or score > 1.0 for _, score, id in docs_and_scores):
logger.warning(