youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 09:28:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
parent a83ccccffc
commit 86594851cb
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/relyt/relyt_vector.py
View File

@ -224,7 +224,7 @@ class RelytVector(BaseVector):
def search_by_vector(self, query_vector: list[float], **kwargs: Any) -> list[Document]:
results = self.similarity_search_with_score_by_vector(
k=int(kwargs.get("top_k")), embedding=query_vector, filter=kwargs.get("filter")
k=int(kwargs.get("top_k", 4)), embedding=query_vector, filter=kwargs.get("filter")
)
# Organize results.