From 2abb947611967d78196f98bfdb06bcead904bb3e Mon Sep 17 00:00:00 2001 From: Jonathan Dobson Date: Fri, 19 Sep 2025 10:28:17 -0400 Subject: [PATCH] add run_with to workflow_runs table (#3478) --- ...e91_add_run_with_to_workflow_runs_table.py | 31 +++++++++++++++++++ skyvern/forge/sdk/db/models.py | 1 + 2 files changed, 32 insertions(+) create mode 100644 alembic/versions/2025_09_19_1414-6692fb5a3e91_add_run_with_to_workflow_runs_table.py diff --git a/alembic/versions/2025_09_19_1414-6692fb5a3e91_add_run_with_to_workflow_runs_table.py b/alembic/versions/2025_09_19_1414-6692fb5a3e91_add_run_with_to_workflow_runs_table.py new file mode 100644 index 00000000..ba48c018 --- /dev/null +++ b/alembic/versions/2025_09_19_1414-6692fb5a3e91_add_run_with_to_workflow_runs_table.py @@ -0,0 +1,31 @@ +"""add run_with to workflow_runs table + +Revision ID: 6692fb5a3e91 +Revises: 4937dff73027 +Create Date: 2025-09-19 14:14:56.090820+00:00 + +""" + +from typing import Sequence, Union + +import sqlalchemy as sa + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "6692fb5a3e91" +down_revision: Union[str, None] = "4937dff73027" +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.add_column("workflow_runs", sa.Column("run_with", sa.String(), nullable=True)) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column("workflow_runs", "run_with") + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py index 3e607b01..3de91aff 100644 --- a/skyvern/forge/sdk/db/models.py +++ b/skyvern/forge/sdk/db/models.py @@ -286,6 +286,7 @@ class WorkflowRunModel(Base): browser_address = Column(String, nullable=True) script_run = Column(JSON, nullable=True) job_id = Column(String, nullable=True) + run_with = Column(String, nullable=True) # 'agent' or 'code' queued_at = Column(DateTime, nullable=True) started_at = Column(DateTime, nullable=True)