diff --git a/alembic/versions/2025_01_06_1214-d47a586d7036_make_browser_session_runnable_type_and_.py b/alembic/versions/2025_01_06_1214-d47a586d7036_make_browser_session_runnable_type_and_.py new file mode 100644 index 00000000..66617ef8 --- /dev/null +++ b/alembic/versions/2025_01_06_1214-d47a586d7036_make_browser_session_runnable_type_and_.py @@ -0,0 +1,35 @@ +"""make browser session runnable type and id nullable + +Revision ID: d47a586d7036 +Revises: 32e2f138f7fd +Create Date: 2025-01-06 12:14:00.216039+00:00 + +""" + +from typing import Sequence, Union + +import sqlalchemy as sa + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "d47a586d7036" +down_revision: Union[str, None] = "32e2f138f7fd" +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("persistent_browser_sessions", sa.Column("browser_id", sa.String(), nullable=True)) + op.alter_column("persistent_browser_sessions", "runnable_type", existing_type=sa.VARCHAR(), nullable=True) + op.alter_column("persistent_browser_sessions", "runnable_id", existing_type=sa.VARCHAR(), nullable=True) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.alter_column("persistent_browser_sessions", "runnable_id", existing_type=sa.VARCHAR(), nullable=False) + op.alter_column("persistent_browser_sessions", "runnable_type", existing_type=sa.VARCHAR(), nullable=False) + op.drop_column("persistent_browser_sessions", "browser_id") + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py index 8fd15f4e..1c501042 100644 --- a/skyvern/forge/sdk/db/models.py +++ b/skyvern/forge/sdk/db/models.py @@ -570,8 +570,9 @@ class PersistentBrowserSessionModel(Base): persistent_browser_session_id = Column(String, primary_key=True, default=generate_persistent_browser_session_id) organization_id = Column(String, ForeignKey("organizations.organization_id"), nullable=False) - runnable_type = Column(String, nullable=False) - runnable_id = Column(String, nullable=False) + runnable_type = Column(String, nullable=True) + runnable_id = Column(String, nullable=True) + browser_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) deleted_at = Column(DateTime, nullable=True)