diff --git a/alembic/versions/2025_08_07_1530-d67d7c00ab02_add_finished_at_to_steps_table.py b/alembic/versions/2025_08_07_1530-d67d7c00ab02_add_finished_at_to_steps_table.py new file mode 100644 index 00000000..987ad734 --- /dev/null +++ b/alembic/versions/2025_08_07_1530-d67d7c00ab02_add_finished_at_to_steps_table.py @@ -0,0 +1,31 @@ +"""add finished_at to steps table + +Revision ID: d67d7c00ab02 +Revises: 0135ee8b36b0 +Create Date: 2025-08-07 15:30:06.604933+00:00 + +""" + +from typing import Sequence, Union + +import sqlalchemy as sa + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "d67d7c00ab02" +down_revision: Union[str, None] = "0135ee8b36b0" +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("steps", sa.Column("finished_at", sa.DateTime(), nullable=True)) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column("steps", "finished_at") + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py index 62ff0de8..9ba398d2 100644 --- a/skyvern/forge/sdk/db/models.py +++ b/skyvern/forge/sdk/db/models.py @@ -134,6 +134,7 @@ class StepModel(Base): reasoning_token_count = Column(Integer, default=0) cached_token_count = Column(Integer, default=0) step_cost = Column(Numeric, default=0) + finished_at = Column(DateTime, nullable=True) class OrganizationModel(Base):