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 f86a8305e96..693495a6a78 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 @@ -73,7 +73,7 @@ def execute(self, dataset_id_list, dataset_setting, question, [f"{reset_title(paragraph.get('title', ''))}{paragraph.get('content')}" for paragraph in paragraph_list])[0:dataset_setting.get('max_paragraph_char_number', 5000)], 'directly_return': '\n'.join( - [f"{reset_title(paragraph.get('title', ''))}{paragraph.get('content')}" for paragraph in + [paragraph.get('content') for paragraph in result if paragraph.get('is_hit_handling_method')]), 'question': question}, diff --git a/apps/application/flow/workflow_manage.py b/apps/application/flow/workflow_manage.py index d5342c9c4cf..d2e99bce85d 100644 --- a/apps/application/flow/workflow_manage.py +++ b/apps/application/flow/workflow_manage.py @@ -304,11 +304,11 @@ def await_result(self, result): if chunk is None: break yield chunk - yield self.get_chunk_content('', True) finally: self.work_flow_post_handler.handler(self.params['chat_id'], self.params['chat_record_id'], self.answer, self) + yield self.get_chunk_content('', True) def run_chain_async(self, current_node): future = executor.submit(self.run_chain, current_node)