From e0b67536e0742ba56f7ab7e1159b42d7688e7783 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 bdece7e665..2330229f43 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 @@ class WorkflowAppGenerateTaskPipeline(BasedGenerateTaskPipeline, WorkflowCycleMa # 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 )