[Backend] Add City and State targeting for Massive geo-targeting (#4133)

skyvern/forge/sdk/schemas/task_v2.py

@@ -5,7 +5,7 @@ from typing import Any
 from pydantic import BaseModel, ConfigDict, Field, field_validator
 
 from skyvern.config import settings
-from skyvern.schemas.runs import ProxyLocation
+from skyvern.schemas.runs import ProxyLocationInput
 from skyvern.utils.url_validators import validate_url
 
 DEFAULT_WORKFLOW_TITLE = "New Workflow"
@@ -40,7 +40,7 @@ class TaskV2(BaseModel):
     output: dict[str, Any] | list | str | None = None
     totp_verification_url: str | None = None
     totp_identifier: str | None = None
-    proxy_location: ProxyLocation | None = None
+    proxy_location: ProxyLocationInput = None
     webhook_callback_url: str | None = None
     webhook_failure_reason: str | None = None
     extracted_information_schema: dict | list | str | None = None
@@ -149,7 +149,7 @@ class TaskV2Request(BaseModel):
     webhook_callback_url: str | None = None
     totp_verification_url: str | None = None
     totp_identifier: str | None = None
-    proxy_location: ProxyLocation | None = None
+    proxy_location: ProxyLocationInput = None
     publish_workflow: bool = False
     extracted_information_schema: dict | list | str | None = None
     error_code_mapping: dict[str, str] | None = None

skyvern/forge/sdk/schemas/tasks.py

@@ -18,7 +18,7 @@ from skyvern.exceptions import (
 from skyvern.forge.sdk.db.enums import TaskType
 from skyvern.forge.sdk.schemas.files import FileInfo
 from skyvern.schemas.docs.doc_strings import PROXY_LOCATION_DOC_STRING
-from skyvern.schemas.runs import ProxyLocation
+from skyvern.schemas.runs import ProxyLocationInput
 from skyvern.utils.url_validators import validate_url
 
 
@@ -69,7 +69,7 @@ class TaskBase(BaseModel):
             }
         ],
     )
-    proxy_location: ProxyLocation | None = Field(
+    proxy_location: ProxyLocationInput = Field(
         default=None,
         description=PROXY_LOCATION_DOC_STRING,
     )