diff --git a/alembic/versions/2025_11_03_0424-2903b6252f31_add_workflow_run_id_workflow_run_block_.py b/alembic/versions/2025_11_03_0424-2903b6252f31_add_workflow_run_id_workflow_run_block_.py new file mode 100644 index 00000000..0f9528b8 --- /dev/null +++ b/alembic/versions/2025_11_03_0424-2903b6252f31_add_workflow_run_id_workflow_run_block_.py @@ -0,0 +1,33 @@ +"""add workflow_run_id & workflow_run_block_id to script_blocks table + +Revision ID: 2903b6252f31 +Revises: d135e472622c +Create Date: 2025-11-03 04:24:52.201400+00:00 + +""" + +from typing import Sequence, Union + +import sqlalchemy as sa + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "2903b6252f31" +down_revision: Union[str, None] = "d135e472622c" +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("script_blocks", sa.Column("workflow_run_id", sa.String(), nullable=True)) + op.add_column("script_blocks", sa.Column("workflow_run_block_id", sa.String(), nullable=True)) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column("script_blocks", "workflow_run_block_id") + op.drop_column("script_blocks", "workflow_run_id") + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py index f67fc737..cbf62a85 100644 --- a/skyvern/forge/sdk/db/models.py +++ b/skyvern/forge/sdk/db/models.py @@ -975,6 +975,8 @@ class ScriptBlockModel(Base): script_block_label = Column(String, nullable=False) script_file_id = Column(String, nullable=True) run_signature = Column(String, nullable=True) + workflow_run_id = Column(String, nullable=True) + workflow_run_block_id = Column(String, 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)