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

Skyvern-AI · Feb 6, 2025 · 9f14224 · 9f14224
1 parent 68e0e73
commit 9f14224
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 79 deletions.
diff --git a/alembic/versions/2025_02_06_0923-8a981a8df4bb_get_proxy_location_enum_back_for_tasks.py b/alembic/versions/2025_02_06_0923-8a981a8df4bb_get_proxy_location_enum_back_for_tasks.py
diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py
@@ -44,7 +44,6 @@
     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)