Skip to content

Commit

Permalink
fix: remove the unused QueueWorkflowPartialSuccessEvent handle in wor…
Browse files Browse the repository at this point in the history
…kflow (#11669)

Co-authored-by: Novice Lee <novicelee@NoviPro.local>
  • Loading branch information
2 people authored and iamjoel committed Dec 16, 2024
1 parent b96686c commit 782a24b
Showing 1 changed file with 0 additions and 22 deletions.
22 changes: 0 additions & 22 deletions api/core/app/apps/workflow/generate_task_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
)
Expand Down

0 comments on commit 782a24b

Please sign in to comment.