vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Revert "get proxy_location enum back in the tasks table" (#1737)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-02-06 17:53:43 +08:00
committed by GitHub
parent 68e0e73020
commit 9f14224589
2 changed files with 1 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
alembic/versions/2025_02_06_0923-8a981a8df4bb_get_proxy_location_enum_back_for_tasks.py
View File

@@ -1,77 +0,0 @@
"""get proxy_location enum back for tasks
Revision ID: 8a981a8df4bb
Revises: ebf093461132
Create Date: 2025-02-06 09:23:24.003503+00:00
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "8a981a8df4bb"
down_revision: Union[str, None] = "ebf093461132"
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.alter_column(
"tasks",
"proxy_location",
existing_type=sa.VARCHAR(),
type_=sa.Enum(
"US_CA",
"US_NY",
"US_TX",
"US_FL",
"US_WA",
"RESIDENTIAL",
"RESIDENTIAL_ES",
"RESIDENTIAL_IE",
"RESIDENTIAL_GB",
"RESIDENTIAL_IN",
"RESIDENTIAL_JP",
"RESIDENTIAL_FR",
"RESIDENTIAL_DE",
"NONE",
name="proxylocation",
),
postgresql_using="proxy_location::proxylocation",
existing_nullable=True,
)
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.alter_column(
"tasks",
"proxy_location",
existing_type=sa.Enum(
"US_CA",
"US_NY",
"US_TX",
"US_FL",
"US_WA",
"RESIDENTIAL",
"RESIDENTIAL_ES",
"RESIDENTIAL_IE",
"RESIDENTIAL_GB",
"RESIDENTIAL_IN",
"RESIDENTIAL_JP",
"RESIDENTIAL_FR",
"RESIDENTIAL_DE",
"NONE",
name="proxylocation",
),
type_=sa.VARCHAR(),
postgresql_using="proxy_location::text",
existing_nullable=True,
)
# ### end Alembic commands ###

3
skyvern/forge/sdk/db/models.py
View File

@@ -44,7 +44,6 @@ from skyvern.forge.sdk.db.id import (
generate_workflow_run_id,
)
from skyvern.forge.sdk.schemas.observers import ObserverThoughtType
from skyvern.forge.sdk.schemas.tasks import ProxyLocation
class Base(AsyncAttrs, DeclarativeBase):
@@ -70,7 +69,7 @@ class TaskModel(Base):
navigation_payload = Column(JSON)
extracted_information = Column(JSON)
failure_reason = Column(String)
proxy_location = Column(Enum(ProxyLocation))
proxy_location = Column(String)
extracted_information_schema = Column(JSON)
workflow_run_id = Column(String, ForeignKey("workflow_runs.workflow_run_id"), index=True)
order = Column(Integer, nullable=True)