Skip to content

Commit

Permalink
fix(milvus): Fix milvus store search with topk always set to 4 (#2131)
Browse files Browse the repository at this point in the history
  • Loading branch information
slper001 authored Nov 18, 2024
1 parent 1d784c5 commit 2decf02
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dbgpt/storage/vector_store/milvus_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -458,7 +458,7 @@ def similar_search_with_scores(
# 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(
Expand Down

0 comments on commit 2decf02

Please sign in to comment.