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

2025-05-30 20:07:12 -07:00
parent aee129a0a8
commit 2ed14f42e7
14 changed files with 103 additions and 5 deletions
--- a/skyvern/forge/sdk/db/models.py
+++ b/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):