From e5bcb08b6dd364d792a68add83f5ff97df11ed97 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Mon, 4 Nov 2024 11:20:31 -0800 Subject: [PATCH] Add new proxies RESIDENTIAL_IN, RESIDENTIAL_JP (#1126) Co-authored-by: Muhammed Salih Altun --- ...8f9e09e181d_add_in_and_jp_proxylocation.py | 30 +++++++++++++++++++ skyvern/forge/sdk/schemas/tasks.py | 2 ++ 2 files changed, 32 insertions(+) create mode 100644 alembic/versions/2024_11_04_1914-b8f9e09e181d_add_in_and_jp_proxylocation.py diff --git a/alembic/versions/2024_11_04_1914-b8f9e09e181d_add_in_and_jp_proxylocation.py b/alembic/versions/2024_11_04_1914-b8f9e09e181d_add_in_and_jp_proxylocation.py new file mode 100644 index 000000000..26db3e605 --- /dev/null +++ b/alembic/versions/2024_11_04_1914-b8f9e09e181d_add_in_and_jp_proxylocation.py @@ -0,0 +1,30 @@ +"""Add IN and JP proxylocation + +Revision ID: b8f9e09e181d +Revises: 2873c5c8c41e +Create Date: 2024-11-04 19:14:36.603689+00:00 + +""" + +from typing import Sequence, Union + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "b8f9e09e181d" +down_revision: Union[str, None] = "2873c5c8c41e" +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.execute("ALTER TYPE proxylocation ADD VALUE 'RESIDENTIAL_IN'") + op.execute("ALTER TYPE proxylocation ADD VALUE 'RESIDENTIAL_JP'") + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + pass + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/schemas/tasks.py b/skyvern/forge/sdk/schemas/tasks.py index 0c76508ab..7b449fa56 100644 --- a/skyvern/forge/sdk/schemas/tasks.py +++ b/skyvern/forge/sdk/schemas/tasks.py @@ -20,6 +20,8 @@ class ProxyLocation(StrEnum): RESIDENTIAL_ES = "RESIDENTIAL_ES" RESIDENTIAL_IE = "RESIDENTIAL_IE" RESIDENTIAL_GB = "RESIDENTIAL_GB" + RESIDENTIAL_IN = "RESIDENTIAL_IN" + RESIDENTIAL_JP = "RESIDENTIAL_JP" NONE = "NONE"