stagehand 4.2.2a0.dev1615__tar.gz → 4.3.0a0.dev1620__tar.gz

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stagehand
-Version: 4.2.2a0.dev1615
+Version: 4.3.0a0.dev1620
 Summary: Browserbase's SDK for building browser agents
 Keywords: ai,agent,ai-agent,browser,browser-agent,browser-automation,computer-use,headless,playwright,cdp,llm,automation,web-scraping,testing
 Author: Browserbase

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "stagehand"
-version = "4.2.2a0.dev1615"
+version = "4.3.0a0.dev1620"
 description = "Browserbase's SDK for building browser agents"
 readme = "README.md"
 requires-python = ">=3.11"

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/src/stagehand/_extension/service-worker.js

@@ -7363,7 +7363,7 @@ function isRecord$1(value) {
 }
 var package_default$1 = {
 	name: "@browserbasehq/stagehand-protocol",
-	version: "3.0.0-alpha-c6b6f278c880d2f8b770ac0c1ecc63f248515968",
+	version: "3.0.0-alpha-76cd7e0f2ff04b0fb5c07e27e0db70746d0ad3c8",
 	"private": true,
 	type: "module",
 	exports: {
@@ -12529,7 +12529,22 @@ strictObject({
 }).meta({ id: "LocalBrowserLaunchOptions" });
 strictObject({
 	cdpUrl: string().min(1),
-	extensionId: string().optional()
+	extensionId: string().optional(),
+	/**
+	* Browser-level CDP WebSocket URL dialed by the extension inside Chrome.
+	* Defaults to the client's URL. Set this when the client connects through
+	* a gateway the extension cannot or should not use.
+	*/
+	extensionCdpUrl: url({
+		protocol: /^wss?$/,
+		hostname: /.+/
+	}).refine((value) => {
+		try {
+			return new URL(value).port !== "0";
+		} catch {
+			return false;
+		}
+	}, { message: "Port must be between 1 and 65535" }).optional()
 }).meta({ id: "LocalBrowserConnectOptions" });
 var DEFAULT_BROWSERBASE_URL = "https://api.browserbase.com";
 var BrowserbaseClientOptionsSchema = strictObject({
@@ -54456,7 +54471,7 @@ var WebTracerProvider = class extends BasicTracerProvider {
 };
 var package_default = {
 	name: "@browserbasehq/stagehand-extension",
-	version: "1.2.0-alpha-c6b6f278c880d2f8b770ac0c1ecc63f248515968",
+	version: "1.2.0-alpha-76cd7e0f2ff04b0fb5c07e27e0db70746d0ad3c8",
 	"private": true,
 	type: "module",
 	scripts: {
@@ -58376,12 +58391,14 @@ var StagehandRuntime = class {
 		const { cdpUrl } = params;
 		const previousSession = this.browserSession;
 		this.browserSession = void 0;
+		this.browserSessionCdpUrl = void 0;
 		this.disposeAllPageEventSubscriptions();
 		this.pagesById.clear();
 		this.responseHandles.clear();
 		await previousSession?.close();
 		try {
 			this.browserSession = await this.adapters.browserSessionFactory(cdpUrl, this.logger, bootstrapLogger);
+			this.browserSessionCdpUrl = cdpUrl;
 		} catch (error) {
 			await this.browserSession?.close();
 			this.browserSession = void 0;
@@ -58396,7 +58413,7 @@ var StagehandRuntime = class {
 				const state = this.state.getState();
 				if (state.status !== "idle") throw new Error("A Stagehand instance is already initialized");
 				this.logger.setLevel(params.logLevel);
-				if (!this.browserSession?.connected) {
+				if (!this.browserSession?.connected || params.browserCdpUrl !== void 0 && params.browserCdpUrl !== this.browserSessionCdpUrl) {
 					if (!params.browserCdpUrl) throw new Error("stagehand.init requires browserCdpUrl until resident mode is active");
 					await this.replaceBrowserConnection({ cdpUrl: params.browserCdpUrl }, logger);
 				}
@@ -58798,6 +58815,7 @@ var StagehandRuntime = class {
 		await this.enqueueLifecycle(async () => {
 			const session = this.browserSession;
 			this.browserSession = void 0;
+			this.browserSessionCdpUrl = void 0;
 			this.clearStagehandInstance();
 			await session?.close();
 		});

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/src/stagehand/_generated/protocol_version.py

@@ -1,4 +1,4 @@
 """Generated from packages/protocol/package.json. Do not edit."""
 
 PROTOCOL_SEMVER_PATTERN = '^(0|[1-9]\\d*)\\.(0|[1-9]\\d*)\\.(0|[1-9]\\d*)(?:-((?:0|[1-9]\\d*|\\d*[A-Za-z-][0-9A-Za-z-]*)(?:\\.(?:0|[1-9]\\d*|\\d*[A-Za-z-][0-9A-Za-z-]*))*))?(?:\\+([0-9A-Za-z-]+(?:\\.[0-9A-Za-z-]+)*))?$'
-STAGEHAND_PROTOCOL_VERSION = '3.0.0-alpha-c6b6f278c880d2f8b770ac0c1ecc63f248515968'
+STAGEHAND_PROTOCOL_VERSION = '3.0.0-alpha-76cd7e0f2ff04b0fb5c07e27e0db70746d0ad3c8'

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/src/stagehand/_validation.py

@@ -16,6 +16,19 @@ def _validate_url(value: str) -> str:
 WireUrl = Annotated[str, AfterValidator(_validate_url)]
 
 
+def _validate_websocket_url(value: str) -> str:
+    parsed = urlparse(value)
+    if parsed.scheme not in ("ws", "wss") or not parsed.hostname:
+        raise ValueError("must be a ws:// or wss:// URL with a host")
+    _url_adapter.validate_python(value)
+    if parsed.port == 0:
+        raise ValueError("port must be between 1 and 65535")
+    return value
+
+
+WireWebSocketUrl = Annotated[str, AfterValidator(_validate_websocket_url)]
+
+
 class WireModel(BaseModel):
     model_config = ConfigDict(validate_assignment=True, allow_inf_nan=False)
 

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/src/stagehand/browser.py

@@ -85,6 +85,8 @@ _CHROME_REQUEST_TIMEOUT_SECONDS = 0.1
 class _WorkerInitMetadata:
     api_key: str | None
     browser: BrowserSessionMetadata | None
+    browser_cdp_url: str | None = None
+    """The CDP WebSocket the extension dials, when it is not the client's own."""
 
 
 @dataclass(frozen=True)
@@ -533,18 +535,25 @@ class LocalBrowser:
         *,
         cdp_url: str,
         extension_id: str | None = None,
+        extension_cdp_url: str | None = None,
     ) -> StagehandBrowser:
         """Connect to an existing browser.
 
         Args:
             extension_id: Deprecated and ignored. Omit this argument; Stagehand
                 discovers the installed extension automatically.
+            extension_cdp_url: The browser-level CDP WebSocket that Stagehand's
+                extension dials from inside the browser. Defaults to the one this
+                client connected to. Set it when the client reaches Chrome through
+                a proxy the browser cannot or should not use, such as an
+                authenticated gateway whose URL carries credentials.
         """
         options = LocalBrowserConnectOptions.model_validate({
             name: value
             for name, value in (
                 ("cdp_url", cdp_url),
                 ("extension_id", extension_id),
+                ("extension_cdp_url", extension_cdp_url),
             )
             if value is not None
         })
@@ -553,7 +562,11 @@ class LocalBrowser:
             origin="connected",
             source=_ConnectedBrowserSource(options.cdp_url),
             local_extension_dir=lambda: str(extension_directory()),
-            worker_init_metadata=_WorkerInitMetadata(api_key=None, browser=None),
+            worker_init_metadata=_WorkerInitMetadata(
+                api_key=None,
+                browser=None,
+                browser_cdp_url=options.extension_cdp_url,
+            ),
         )
 
 

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/src/stagehand/client_models.py

@@ -18,7 +18,7 @@ from ._generated.models import (
     StagehandResultMetadata,
     TelemetryConfig,
 )
-from ._validation import WireModel, WireUrl
+from ._validation import WireModel, WireUrl, WireWebSocketUrl
 from .browserbase_session import DEFAULT_BROWSERBASE_URL
 from .client_types import Cache as CacheInput
 
@@ -108,6 +108,8 @@ class LocalBrowserConnectOptions(WireModel):
     cdp_url: Annotated[str, Field(min_length=1)]
     extension_id: str | None = None
     """Deprecated and ignored. Omit this field; Stagehand discovers the extension automatically."""
+    extension_cdp_url: WireWebSocketUrl | None = None
+    """Browser-level CDP WebSocket the extension dials, when it differs from `cdp_url`."""
 
 
 class BrowserbaseClientOptions(WireModel):

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/src/stagehand/client_types.py

@@ -67,6 +67,8 @@ class LocalBrowserConnectOptions(TypedDict):
     cdp_url: str
     extension_id: NotRequired[str]
     """Deprecated and ignored. Omit this field; Stagehand discovers the extension automatically."""
+    extension_cdp_url: NotRequired[str]
+    """Browser-level CDP WebSocket the extension dials, when it differs from `cdp_url`."""
 
 
 class BrowserbaseClientOptions(TypedDict, total=False):

{stagehand-4.2.2a0.dev1615 → stagehand-4.3.0a0.dev1620}/src/stagehand/stagehand.py

@@ -503,7 +503,10 @@ class Stagehand:
     def _worker_init_params(
         self, claimed: _ClaimedBrowser, *, has_browserbase_connection: bool
     ) -> StagehandInitParams:
-        browser_cdp_url = claimed.cdp_client.web_socket_debugger_url
+        browser_cdp_url = (
+            claimed.worker_init_metadata.browser_cdp_url
+            or claimed.cdp_client.web_socket_debugger_url
+        )
         if browser_cdp_url is None:
             raise RuntimeError("The browser CDP WebSocket URL is unavailable")
         values = self._create_config.model_dump(