vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Introduce new TaskStatuses: queued and timed_out (#170)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-04-08 22:56:49 -07:00
committed by GitHub
parent a7351f2212
commit 81a1478e2f
3 changed files with 45 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
skyvern/forge/sdk/schemas/tasks.py
View File

@@ -73,19 +73,23 @@ class TaskRequest(BaseModel):
class TaskStatus(StrEnum):
created = "created"
queued = "queued"
running = "running"
timed_out = "timed_out"
failed = "failed"
terminated = "terminated"
completed = "completed"
def is_final(self) -> bool:
return self in {TaskStatus.failed, TaskStatus.terminated, TaskStatus.completed}
return self in {TaskStatus.failed, TaskStatus.terminated, TaskStatus.completed, TaskStatus.timed_out}
def can_update_to(self, new_status: TaskStatus) -> bool:
allowed_transitions: dict[TaskStatus, set[TaskStatus]] = {
TaskStatus.created: {TaskStatus.running},
TaskStatus.running: {TaskStatus.completed, TaskStatus.failed, TaskStatus.terminated},
TaskStatus.created: {TaskStatus.queued, TaskStatus.running, TaskStatus.timed_out},
TaskStatus.queued: {TaskStatus.running, TaskStatus.timed_out},
TaskStatus.running: {TaskStatus.completed, TaskStatus.failed, TaskStatus.terminated, TaskStatus.timed_out},
TaskStatus.failed: set(),
TaskStatus.terminated: set(),
TaskStatus.completed: set(),
}
return new_status in allowed_transitions[self]
@@ -97,6 +101,7 @@ class TaskStatus(StrEnum):
def cant_have_extracted_info(self) -> bool:
status_cant_have_extracted_information = {
TaskStatus.created,
TaskStatus.queued,
TaskStatus.running,
TaskStatus.failed,
TaskStatus.terminated,