fix annotation bugs

2026-05-06 02:18:08 +08:00 · 2024-03-18 17:57:10 +08:00
parent 0ea233edbe
commit 09cfbe117e
4 changed files with 27 additions and 152 deletions
--- a/api/core/app/task_pipeline/workflow_cycle_manage.py
+++ b/api/core/app/task_pipeline/workflow_cycle_manage.py
@ -425,8 +425,11 @@ class WorkflowCycleManage:
        return workflow_node_execution

    def _handle_workflow_finished(self, event: QueueStopEvent | QueueWorkflowSucceededEvent | QueueWorkflowFailedEvent) \
-            -> WorkflowRun:
+            -> Optional[WorkflowRun]:
        workflow_run = db.session.query(WorkflowRun).filter(WorkflowRun.id == self._task_state.workflow_run_id).first()
+        if not workflow_run:
+            return None
+
        if isinstance(event, QueueStopEvent):
            workflow_run = self._workflow_run_failed(
                workflow_run=workflow_run,