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

2024-08-23 16:59:34 +08:00
parent ad13011043
commit f29685f8a1
1 changed files with 1 additions and 1 deletions
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@@ -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 []