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

pull/7584/head
张皮皮 2 years ago committed by GitHub
parent ad13011043
commit f29685f8a1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -614,7 +614,7 @@ class DatasetRetrieval:
top_k: int, score_threshold: float) -> list[Document]:
filter_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)
if not filter_documents:
return []

Loading…
Cancel
Save