support proxy select for pbs (#3461)

This commit is contained in:
LawyZheng
2025-09-18 17:21:46 +08:00
committed by GitHub
parent 763eb7bd4f
commit 3e12133af9
8 changed files with 50 additions and 0 deletions

View File

@@ -12,6 +12,7 @@ from skyvern.exceptions import BrowserSessionNotRenewable, MissingBrowserAddress
from skyvern.forge.sdk.db.client import AgentDB
from skyvern.forge.sdk.db.polls import wait_on_persistent_browser_address
from skyvern.forge.sdk.schemas.persistent_browser_sessions import PersistentBrowserSession, is_final_status
from skyvern.schemas.runs import ProxyLocation
from skyvern.webeye.browser_factory import BrowserState
LOG = structlog.get_logger()
@@ -237,6 +238,7 @@ class PersistentSessionsManager:
runnable_id: str | None = None,
runnable_type: str | None = None,
timeout_minutes: int | None = None,
proxy_location: ProxyLocation | None = ProxyLocation.RESIDENTIAL,
) -> PersistentBrowserSession:
"""Create a new browser session for an organization and return its ID with the browser state."""
@@ -250,6 +252,7 @@ class PersistentSessionsManager:
runnable_type=runnable_type,
runnable_id=runnable_id,
timeout_minutes=timeout_minutes,
proxy_location=proxy_location,
)
return browser_session_db