[Backend] Add City and State targeting for Massive geo-targeting (#4133)
This commit is contained in:
@@ -41,7 +41,7 @@ from skyvern.exceptions import (
|
||||
)
|
||||
from skyvern.forge.sdk.api.files import get_download_dir, make_temp_directory
|
||||
from skyvern.forge.sdk.core.skyvern_context import current, ensure_context
|
||||
from skyvern.schemas.runs import ProxyLocation, get_tzinfo_from_proxy
|
||||
from skyvern.schemas.runs import ProxyLocation, ProxyLocationInput, get_tzinfo_from_proxy
|
||||
from skyvern.webeye.utils.page import ScreenshotMode, SkyvernFrame
|
||||
|
||||
LOG = structlog.get_logger()
|
||||
@@ -680,7 +680,7 @@ class BrowserState:
|
||||
async def check_and_fix_state(
|
||||
self,
|
||||
url: str | None = None,
|
||||
proxy_location: ProxyLocation | None = None,
|
||||
proxy_location: ProxyLocationInput = None,
|
||||
task_id: str | None = None,
|
||||
workflow_run_id: str | None = None,
|
||||
script_id: str | None = None,
|
||||
@@ -886,7 +886,7 @@ class BrowserState:
|
||||
async def get_or_create_page(
|
||||
self,
|
||||
url: str | None = None,
|
||||
proxy_location: ProxyLocation | None = None,
|
||||
proxy_location: ProxyLocationInput = None,
|
||||
task_id: str | None = None,
|
||||
workflow_run_id: str | None = None,
|
||||
script_id: str | None = None,
|
||||
|
||||
@@ -9,7 +9,7 @@ from skyvern.exceptions import MissingBrowserState
|
||||
from skyvern.forge import app
|
||||
from skyvern.forge.sdk.schemas.tasks import Task
|
||||
from skyvern.forge.sdk.workflow.models.workflow import WorkflowRun
|
||||
from skyvern.schemas.runs import ProxyLocation
|
||||
from skyvern.schemas.runs import ProxyLocation, ProxyLocationInput
|
||||
from skyvern.webeye.browser_factory import BrowserContextFactory, BrowserState, VideoArtifact
|
||||
|
||||
LOG = structlog.get_logger()
|
||||
@@ -26,7 +26,7 @@ class BrowserManager:
|
||||
|
||||
@staticmethod
|
||||
async def _create_browser_state(
|
||||
proxy_location: ProxyLocation | None = None,
|
||||
proxy_location: ProxyLocationInput = None,
|
||||
url: str | None = None,
|
||||
task_id: str | None = None,
|
||||
workflow_run_id: str | None = None,
|
||||
|
||||
@@ -16,7 +16,7 @@ from skyvern.forge.sdk.schemas.persistent_browser_sessions import (
|
||||
PersistentBrowserSessionStatus,
|
||||
is_final_status,
|
||||
)
|
||||
from skyvern.schemas.runs import ProxyLocation
|
||||
from skyvern.schemas.runs import ProxyLocation, ProxyLocationInput
|
||||
from skyvern.webeye.browser_factory import BrowserState
|
||||
|
||||
LOG = structlog.get_logger()
|
||||
@@ -254,7 +254,7 @@ class PersistentSessionsManager:
|
||||
runnable_id: str | None = None,
|
||||
runnable_type: str | None = None,
|
||||
timeout_minutes: int | None = None,
|
||||
proxy_location: ProxyLocation | None = ProxyLocation.RESIDENTIAL,
|
||||
proxy_location: ProxyLocationInput = ProxyLocation.RESIDENTIAL,
|
||||
) -> PersistentBrowserSession:
|
||||
"""Create a new browser session for an organization and return its ID with the browser state."""
|
||||
|
||||
|
||||
Reference in New Issue
Block a user