|
@@ -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
|
|
|
)
|