Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 11:42:29 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix score_threshold is none, return all top K documents (#7581)

Browse Source
This commit is contained in:
张皮皮 2024-08-23 16:59:34 +08:00 committed by GitHub
parent ad13011043
commit f29685f8a1
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
api/core/rag/retrieval/dataset_retrieval.py
View File

@ -614,7 +614,7 @@ class DatasetRetrieval:
top_k: int, score_threshold: float) -> list[Document]: top_k: int, score_threshold: float) -> list[Document]:
filter_documents = [] filter_documents = []
for document in all_documents: for document in all_documents:
if score_threshold and document.metadata['score'] >= score_threshold: if score_threshold is None or document.metadata['score'] >= score_threshold:
filter_documents.append(document) filter_documents.append(document)
if not filter_documents: if not filter_documents:
return [] return []