From b7af52d037a1630993ce586ee340154cf0bf3d73 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 26 Jun 2024 17:58:10 -0700 Subject: [PATCH] when updating a step to canceled status, we can skip terminal status check when is_last = true (#524) --- skyvern/forge/sdk/models.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/skyvern/forge/sdk/models.py b/skyvern/forge/sdk/models.py index 46c1c8fe..a0bc39fd 100644 --- a/skyvern/forge/sdk/models.py +++ b/skyvern/forge/sdk/models.py @@ -66,6 +66,9 @@ class Step(BaseModel): if status and not old_status.can_update_to(status): raise ValueError(f"invalid_status_transition({old_status},{status},{self.step_id})") + if status == StepStatus.canceled: + return + if status and status.requires_output() and output is None: raise ValueError(f"status_requires_output({status},{self.step_id})")