better catch exceptions (#434)

2024-06-13 21:53:46 +08:00
parent 81619dd850
commit 4bb7403f61
2 changed files with 53 additions and 23 deletions
--- a/skyvern/forge/sdk/schemas/tasks.py
+++ b/skyvern/forge/sdk/schemas/tasks.py
@@ -95,8 +95,13 @@ class TaskStatus(StrEnum):
                TaskStatus.queued,
                TaskStatus.running,
                TaskStatus.timed_out,
+                TaskStatus.failed,
+            },
+            TaskStatus.queued: {
+                TaskStatus.running,
+                TaskStatus.timed_out,
+                TaskStatus.failed,
            },
-            TaskStatus.queued: {TaskStatus.running, TaskStatus.timed_out},
            TaskStatus.running: {
                TaskStatus.completed,
                TaskStatus.failed,