diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index 457376f3..5d6fcda9 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -1262,6 +1262,7 @@ async def _cancel_workflow_run(workflow_run_id: str, organization_id: str, x_api WorkflowRunStatus.running, WorkflowRunStatus.created, WorkflowRunStatus.queued, + WorkflowRunStatus.paused, ]: continue await app.WORKFLOW_SERVICE.mark_workflow_run_as_canceled(child_workflow_run.workflow_run_id) diff --git a/skyvern/forge/sdk/routes/streaming_verify.py b/skyvern/forge/sdk/routes/streaming_verify.py index 5f28542d..7d43a9aa 100644 --- a/skyvern/forge/sdk/routes/streaming_verify.py +++ b/skyvern/forge/sdk/routes/streaming_verify.py @@ -179,7 +179,12 @@ async def verify_workflow_run( return None, None - if workflow_run.status not in [WorkflowRunStatus.created, WorkflowRunStatus.queued, WorkflowRunStatus.running]: + if workflow_run.status not in [ + WorkflowRunStatus.created, + WorkflowRunStatus.queued, + WorkflowRunStatus.running, + WorkflowRunStatus.paused, + ]: LOG.info( "Workflow run is not running.", workflow_run_status=workflow_run.status, diff --git a/skyvern/forge/sdk/workflow/models/workflow.py b/skyvern/forge/sdk/workflow/models/workflow.py index 5671b22b..ddf72612 100644 --- a/skyvern/forge/sdk/workflow/models/workflow.py +++ b/skyvern/forge/sdk/workflow/models/workflow.py @@ -110,6 +110,7 @@ class WorkflowRunStatus(StrEnum): canceled = "canceled" timed_out = "timed_out" completed = "completed" + paused = "paused" def is_final(self) -> bool: return self in [ diff --git a/skyvern/services/run_service.py b/skyvern/services/run_service.py index 2e13e2b6..a77f2977 100644 --- a/skyvern/services/run_service.py +++ b/skyvern/services/run_service.py @@ -119,6 +119,7 @@ async def cancel_workflow_run( WorkflowRunStatus.running, WorkflowRunStatus.created, WorkflowRunStatus.queued, + WorkflowRunStatus.paused, ]: continue await app.WORKFLOW_SERVICE.mark_workflow_run_as_canceled(child_workflow_run.workflow_run_id)