add model to Task and TaskV2; expose it to run_task endpoint; thread … (#2540)

skyvern/forge/sdk/db/client.py

@@ -140,6 +140,7 @@ class AgentDB:
         task_type: str = TaskType.general,
         application: str | None = None,
         include_action_history_in_verification: bool | None = None,
+        model: dict[str, Any] | None = None,
     ) -> Task:
         try:
             async with self.Session() as session:
@@ -166,6 +167,7 @@ class AgentDB:
                     error_code_mapping=error_code_mapping,
                     application=application,
                     include_action_history_in_verification=include_action_history_in_verification,
+                    model=model,
                 )
                 session.add(new_task)
                 await session.commit()
@@ -2363,6 +2365,7 @@ class AgentDB:
         webhook_callback_url: str | None = None,
         extracted_information_schema: dict | list | str | None = None,
         error_code_mapping: dict | None = None,
+        model: dict[str, Any] | None = None,
     ) -> TaskV2:
         async with self.Session() as session:
             new_task_v2 = TaskV2Model(
@@ -2378,6 +2381,7 @@ class AgentDB:
                 extracted_information_schema=extracted_information_schema,
                 error_code_mapping=error_code_mapping,
                 organization_id=organization_id,
+                model=model,
             )
             session.add(new_task_v2)
             await session.commit()

skyvern/forge/sdk/db/models.py

@@ -92,6 +92,7 @@ class TaskModel(Base):
         nullable=False,
         index=True,
     )
+    model = Column(JSON, nullable=True)
 
 
 class StepModel(Base):
@@ -593,6 +594,7 @@ class TaskV2Model(Base):
 
     created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
     modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)
+    model = Column(JSON, nullable=True)
 
 
 class ThoughtModel(Base):

skyvern/forge/sdk/db/utils.py

@@ -91,6 +91,7 @@ def convert_to_task(task_obj: TaskModel, debug_enabled: bool = False, workflow_p
         error_code_mapping=task_obj.error_code_mapping,
         errors=task_obj.errors,
         application=task_obj.application,
+        model=task_obj.model,
     )
     return task