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

8
skyvern/forge/sdk/db/models.py
View File

@@ -30,7 +30,7 @@ class TaskModel(Base):
task_id = Column(String, primary_key=True, index=True, default=generate_task_id)
organization_id = Column(String, ForeignKey("organizations.organization_id"))
status = Column(String)
status = Column(String, index=True)
webhook_callback_url = Column(String)
title = Column(String)
url = Column(String)
@@ -46,8 +46,10 @@ class TaskModel(Base):
retry = Column(Integer, nullable=True)
error_code_mapping = Column(JSON, nullable=True)
errors = Column(JSON, default=[], nullable=False)
created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)
created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False, index=True)
modified_at = Column(
DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False, index=True
)
class StepModel(Base):

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,