ソースを参照

fix: remove the unused QueueWorkflowPartialSuccessEvent handle in workflow (#11669)

Co-authored-by: Novice Lee <novicelee@NoviPro.local>
Novice 4 ヶ月 前
コミット
e0b67536e0
1 ファイル変更0 行追加22 行削除
  1. 0 22
      api/core/app/apps/workflow/generate_task_pipeline.py

+ 0 - 22
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
                 )