jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: update the default values of top-k parameter in vdb to be consistent (#9367)

Browse source
This commit is contained in:
zhuhao 2024-10-16 16:00:21 +08:00 • committed by GitHub
commit 86594851cb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 9 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/rag/datasource/vdb/myscale/myscale_vector.py
View file

@ -121,7 +121,7 @@ class MyScaleVector(BaseVector):
return self._search(f"TextSearch('enable_nlq=false')(text, '{query}')", SortOrder.DESC, **kwargs)
def _search(self, dist: str, order: SortOrder, **kwargs: Any) -> list[Document]:
top_k = kwargs.get("top_k", 5)
top_k = kwargs.get("top_k", 4)
score_threshold = float(kwargs.get("score_threshold") or 0.0)
where_str = (
f"WHERE dist < {1 - score_threshold}"