diff --git a/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py b/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py index 70d8ddb69be..36dd2144aee 100644 --- a/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py +++ b/apps/application/flow/step_node/reranker_node/impl/base_reranker_node.py @@ -107,7 +107,9 @@ def execute(self, question, reranker_setting, reranker_list, reranker_model_id, max_paragraph_char_number = reranker_setting.get('max_paragraph_char_number', 5000) result = reset_result_list(result, documents) r = filter_result(result, max_paragraph_char_number, top_n, similarity) - return NodeResult({'result_list': r, 'result': ''.join([item.get('page_content') for item in r])}, {}) + return NodeResult({'result_list': r, 'result': ''.join([item.get('page_content') for item in r]), + 'is_hit_handling_method_list': [r for row in r if + row.get('metadata').get('is_hit_handling_method')]}, {}) def get_details(self, index: int, **kwargs): return { diff --git a/ui/src/workflow/common/data.ts b/ui/src/workflow/common/data.ts index 0a8f3707fed..0bd073f1032 100644 --- a/ui/src/workflow/common/data.ts +++ b/ui/src/workflow/common/data.ts @@ -372,6 +372,10 @@ export const rerankerNode = { label: t('workflow.nodes.rerankerNode.result'), value: 'result', }, + { + label: t('workflow.nodes.searchKnowledgeNode.is_hit_handling_method_list'), + value: 'is_hit_handling_method_list', + } ], }, },