get proxy_location enum back in the tasks table (#1736)

This commit is contained in:
Shuchang Zheng
2025-02-06 17:35:59 +08:00
committed by GitHub
parent 0e1fa58484
commit 68e0e73020
2 changed files with 79 additions and 1 deletions

View File

@@ -44,6 +44,7 @@ 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):
@@ -69,7 +70,7 @@ class TaskModel(Base):
navigation_payload = Column(JSON)
extracted_information = Column(JSON)
failure_reason = Column(String)
proxy_location = Column(String)
proxy_location = Column(Enum(ProxyLocation))
extracted_information_schema = Column(JSON)
workflow_run_id = Column(String, ForeignKey("workflow_runs.workflow_run_id"), index=True)
order = Column(Integer, nullable=True)