From 1eccb541994bd1d4ed328a0ed70b2eeb04978a9a Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Thu, 3 Apr 2025 10:57:12 +0800 Subject: [PATCH] fix: Knowledge base retrieval segmentation sorted by relevance (#2791) --- .../search_dataset_node/impl/base_search_dataset_node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py b/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py index a74e5cc8e..5107d4ce2 100644 --- a/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py +++ b/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py @@ -88,7 +88,7 @@ class BaseSearchDatasetNode(ISearchDatasetStepNode): 'is_hit_handling_method_list': [row for row in result if row.get('is_hit_handling_method')], 'data': '\n'.join( [f"{reset_title(paragraph.get('title', ''))}{paragraph.get('content')}" for paragraph in - paragraph_list])[0:dataset_setting.get('max_paragraph_char_number', 5000)], + result])[0:dataset_setting.get('max_paragraph_char_number', 5000)], 'directly_return': '\n'.join( [paragraph.get('content') for paragraph in result if