diff --git a/alembic/versions/2024_08_06_1515-8f237f00faeb_add_gb_proxy.py b/alembic/versions/2024_08_06_1515-8f237f00faeb_add_gb_proxy.py new file mode 100644 index 00000000..71dcb7c0 --- /dev/null +++ b/alembic/versions/2024_08_06_1515-8f237f00faeb_add_gb_proxy.py @@ -0,0 +1,26 @@ +"""Add GB proxy + +Revision ID: 8f237f00faeb +Revises: c5ed5a3a14eb +Create Date: 2024-08-06 15:15:15.369986+00:00 + +""" +from typing import Sequence, Union + +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision: str = '8f237f00faeb' +down_revision: Union[str, None] = 'c5ed5a3a14eb' +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade() -> None: + op.execute("ALTER TYPE proxylocation ADD VALUE 'RESIDENTIAL_GB'") + + +def downgrade() -> None: + pass diff --git a/skyvern/forge/sdk/schemas/tasks.py b/skyvern/forge/sdk/schemas/tasks.py index 1ddd2e95..a448cde2 100644 --- a/skyvern/forge/sdk/schemas/tasks.py +++ b/skyvern/forge/sdk/schemas/tasks.py @@ -18,6 +18,7 @@ class ProxyLocation(StrEnum): RESIDENTIAL = "RESIDENTIAL" RESIDENTIAL_ES = "RESIDENTIAL_ES" RESIDENTIAL_IE = "RESIDENTIAL_IE" + RESIDENTIAL_GB = "RESIDENTIAL_GB" NONE = "NONE"