Add application column to tasks (#1291)

Co-authored-by: Muhammed Salih Altun <muhammedsalihaltun@gmail.com>

skyvern/forge/sdk/db/client.py

@@ -114,6 +114,7 @@ class AgentDB:
         max_steps_per_run: int | None = None,
         error_code_mapping: dict[str, str] | None = None,
         task_type: str = TaskType.general,
+        application: str | None = None,
     ) -> Task:
         try:
             async with self.Session() as session:
@@ -138,6 +139,7 @@ class AgentDB:
                     retry=retry,
                     max_steps_per_run=max_steps_per_run,
                     error_code_mapping=error_code_mapping,
+                    application=application,
                 )
                 session.add(new_task)
                 await session.commit()
@@ -478,6 +480,7 @@ class AgentDB:
         workflow_run_id: str | None = None,
         organization_id: str | None = None,
         only_standalone_tasks: bool = False,
+        application: str | None = None,
         order_by_column: OrderBy = OrderBy.created_at,
         order: SortDirection = SortDirection.desc,
     ) -> list[Task]:
@@ -505,6 +508,8 @@ class AgentDB:
                     query = query.filter(TaskModel.workflow_run_id == workflow_run_id)
                 if only_standalone_tasks:
                     query = query.filter(TaskModel.workflow_run_id.is_(None))
+                if application:
+                    query = query.filter(TaskModel.application == application)
                 order_by_col = getattr(TaskModel, order_by_column)
                 query = (
                     query.order_by(order_by_col.desc() if order == SortDirection.desc else order_by_col.asc())

skyvern/forge/sdk/db/models.py

@@ -71,6 +71,7 @@ class TaskModel(Base):
     error_code_mapping = Column(JSON, nullable=True)
     errors = Column(JSON, default=[], nullable=False)
     max_steps_per_run = Column(Integer, nullable=True)
+    application = Column(String, nullable=True)
     created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False, index=True)
     modified_at = Column(
         DateTime,

skyvern/forge/sdk/db/utils.py

@@ -82,6 +82,7 @@ def convert_to_task(task_obj: TaskModel, debug_enabled: bool = False) -> Task:
         max_steps_per_run=task_obj.max_steps_per_run,
         error_code_mapping=task_obj.error_code_mapping,
         errors=task_obj.errors,
+        application=task_obj.application,
     )
     return task