From 47e0f92c0fcbefe9cfdd76881a3cd1e46334a2a9 Mon Sep 17 00:00:00 2001 From: Rain Wang Date: Wed, 18 Jun 2025 14:35:12 +0800 Subject: [PATCH] Fixes #20748 KnowledgeRetrievalNode return all external documents when reranker disabled even top-k configed (#20762) --- api/core/rag/retrieval/dataset_retrieval.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/core/rag/retrieval/dataset_retrieval.py b/api/core/rag/retrieval/dataset_retrieval.py index 697886052..38c0b540d 100644 --- a/api/core/rag/retrieval/dataset_retrieval.py +++ b/api/core/rag/retrieval/dataset_retrieval.py @@ -496,6 +496,8 @@ class DatasetRetrieval: all_documents = self.calculate_keyword_score(query, all_documents, top_k) elif index_type == "high_quality": all_documents = self.calculate_vector_score(all_documents, top_k, score_threshold) + else: + all_documents = all_documents[:top_k] if top_k else all_documents self._on_query(query, dataset_ids, app_id, user_from, user_id)