From 782a24b04dcd63f00cb3baf8ba9a69c011905974 Mon Sep 17 00:00:00 2001 From: Novice <857526207@qq.com> Date: Sun, 15 Dec 2024 21:18:14 +0800 Subject: [PATCH] fix: remove the unused QueueWorkflowPartialSuccessEvent handle in workflow (#11669) Co-authored-by: Novice Lee --- .../apps/workflow/generate_task_pipeline.py | 22 ------------------- 1 file changed, 22 deletions(-) diff --git a/api/core/app/apps/workflow/generate_task_pipeline.py b/api/core/app/apps/workflow/generate_task_pipeline.py index bdece7e6657873..2330229f439b97 100644 --- a/api/core/app/apps/workflow/generate_task_pipeline.py +++ b/api/core/app/apps/workflow/generate_task_pipeline.py @@ -394,28 +394,6 @@ def _process_stream_response( # save workflow app log self._save_workflow_app_log(workflow_run) - yield self._workflow_finish_to_stream_response( - task_id=self._application_generate_entity.task_id, workflow_run=workflow_run - ) - elif isinstance(event, QueueWorkflowPartialSuccessEvent): - if not workflow_run: - raise Exception("Workflow run not initialized.") - - if not graph_runtime_state: - raise Exception("Graph runtime state not initialized.") - workflow_run = self._handle_workflow_run_partial_success( - workflow_run=workflow_run, - start_at=graph_runtime_state.start_at, - total_tokens=graph_runtime_state.total_tokens, - total_steps=graph_runtime_state.node_run_steps, - outputs=event.outputs, - exceptions_count=event.exceptions_count, - conversation_id=None, - trace_manager=trace_manager, - ) - # save workflow app log - self._save_workflow_app_log(workflow_run) - yield self._workflow_finish_to_stream_response( task_id=self._application_generate_entity.task_id, workflow_run=workflow_run )