From 9eb4b46bdc79c50342845ee24ab315f88f045834 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Mon, 12 May 2025 12:56:48 -0700 Subject: [PATCH] add max_steps to task v2 table (#2293) --- ...499ecbce_add_max_steps_to_task_v2_table.py | 31 +++++++++++++++++++ skyvern/forge/sdk/db/models.py | 1 + 2 files changed, 32 insertions(+) create mode 100644 alembic/versions/2025_05_05_1008-07cb499ecbce_add_max_steps_to_task_v2_table.py diff --git a/alembic/versions/2025_05_05_1008-07cb499ecbce_add_max_steps_to_task_v2_table.py b/alembic/versions/2025_05_05_1008-07cb499ecbce_add_max_steps_to_task_v2_table.py new file mode 100644 index 00000000..2dd5b879 --- /dev/null +++ b/alembic/versions/2025_05_05_1008-07cb499ecbce_add_max_steps_to_task_v2_table.py @@ -0,0 +1,31 @@ +"""add max_steps to task v2 table + +Revision ID: 07cb499ecbce +Revises: e8285b6ddcf0 +Create Date: 2025-05-05 10:08:04.242755+00:00 + +""" + +from typing import Sequence, Union + +import sqlalchemy as sa + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "07cb499ecbce" +down_revision: Union[str, None] = "e8285b6ddcf0" +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("observer_cruises", sa.Column("max_steps", sa.Integer(), nullable=True)) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column("observer_cruises", "max_steps") + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py index 7863bbea..0e57df60 100644 --- a/skyvern/forge/sdk/db/models.py +++ b/skyvern/forge/sdk/db/models.py @@ -587,6 +587,7 @@ class TaskV2Model(Base): proxy_location = Column(String, nullable=True) extracted_information_schema = Column(JSON, nullable=True) error_code_mapping = Column(JSON, nullable=True) + max_steps = Column(Integer, nullable=True) created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False) modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)