stagehand 4.2.0a0.dev1587__tar.gz → 4.2.0a0.dev1588__tar.gz

{stagehand-4.2.0a0.dev1587 → stagehand-4.2.0a0.dev1588}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stagehand
-Version: 4.2.0a0.dev1587
+Version: 4.2.0a0.dev1588
 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.0a0.dev1587 → stagehand-4.2.0a0.dev1588}/pyproject.toml

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

{stagehand-4.2.0a0.dev1587 → stagehand-4.2.0a0.dev1588}/src/stagehand/_extension/service-worker.js

@@ -7363,7 +7363,7 @@ function isRecord$1(value) {
 }
 var package_default$1 = {
 	name: "@browserbasehq/stagehand-protocol",
-	version: "2.1.0-alpha-0f3d0df1b028b50564bf56ff47070ddfd9fdf823",
+	version: "2.1.0-alpha-f2060a689d150957768d92e41eda8decc68bc5bf",
 	"private": true,
 	type: "module",
 	exports: {
@@ -12557,7 +12557,7 @@ strictObject({
 }).meta({ id: "LocalBrowserLaunchOptions" });
 strictObject({
 	cdpUrl: string().min(1),
-	extensionId: string().min(1).optional()
+	extensionId: string().optional()
 }).meta({ id: "LocalBrowserConnectOptions" });
 var DEFAULT_BROWSERBASE_URL = "https://api.browserbase.com";
 var BrowserbaseClientOptionsSchema = strictObject({
@@ -12579,7 +12579,7 @@ strictObject({
 	baseUrl: url().default(DEFAULT_BROWSERBASE_URL),
 	clientOptions: BrowserbaseClientOptionsSchema.optional(),
 	sessionId: string().min(1),
-	extensionId: string().min(1).optional()
+	extensionId: string().optional()
 }).meta({ id: "BrowserbaseConnectOptions" });
 var BrowserbaseCredentialsShape = {
 	apiKey: string().min(1),
@@ -54394,7 +54394,7 @@ var WebTracerProvider = class extends BasicTracerProvider {
 };
 var package_default = {
 	name: "@browserbasehq/stagehand-extension",
-	version: "1.1.0-alpha-0f3d0df1b028b50564bf56ff47070ddfd9fdf823",
+	version: "1.1.0-alpha-f2060a689d150957768d92e41eda8decc68bc5bf",
 	"private": true,
 	type: "module",
 	scripts: {

{stagehand-4.2.0a0.dev1587 → stagehand-4.2.0a0.dev1588}/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 = '2.1.0-alpha-0f3d0df1b028b50564bf56ff47070ddfd9fdf823'
+STAGEHAND_PROTOCOL_VERSION = '2.1.0-alpha-f2060a689d150957768d92e41eda8decc68bc5bf'

{stagehand-4.2.0a0.dev1587 → stagehand-4.2.0a0.dev1588}/src/stagehand/browser.py

@@ -305,6 +305,7 @@ async def _connect_browser(
     origin: Literal["launched", "connected"],
     source: _BrowserConnectionSource,
     extension_dir: str | None = None,
+    local_extension_dir: Callable[[], str] | None = None,
     extension_id: str | None = None,
     preloaded_extension: bool = False,
     after_connect: Callable[[CDPClient], Awaitable[None]] | None = None,
@@ -316,6 +317,7 @@ async def _connect_browser(
         cdp_client = await CDPClient.connect(
             cdp_url=source.cdp_url,
             extension_dir=extension_dir,
+            local_extension_dir=local_extension_dir,
             extension_id=extension_id,
             preloaded_extension=preloaded_extension,
             service_worker_url_includes="service-worker.js",
@@ -532,6 +534,12 @@ class LocalBrowser:
         cdp_url: str,
         extension_id: str | None = None,
     ) -> StagehandBrowser:
+        """Connect to an existing browser.
+
+        Args:
+            extension_id: Deprecated and ignored. Omit this argument; Stagehand
+                discovers the installed extension automatically.
+        """
         options = LocalBrowserConnectOptions.model_validate({
             name: value
             for name, value in (
@@ -540,13 +548,11 @@ class LocalBrowser:
             )
             if value is not None
         })
-        extension_dir = None if options.extension_id is not None else str(extension_directory())
         return await _connect_browser(
             provider="local",
             origin="connected",
             source=_ConnectedBrowserSource(options.cdp_url),
-            extension_dir=extension_dir,
-            extension_id=options.extension_id,
+            local_extension_dir=lambda: str(extension_directory()),
             worker_init_metadata=_WorkerInitMetadata(api_key=None, browser=None),
         )
 
@@ -646,6 +652,12 @@ class BrowserbaseBrowser:
         session_id: str,
         extension_id: str | None = None,
     ) -> StagehandBrowser:
+        """Connect to an existing browser.
+
+        Args:
+            extension_id: Deprecated and ignored. Omit this argument; Stagehand
+                discovers the installed extension automatically.
+        """
         options = BrowserbaseConnectOptions.model_validate({
             name: value
             for name, value in (
@@ -668,8 +680,7 @@ class BrowserbaseBrowser:
                 keep_alive=True,
                 _close_callback=connection.close,
             ),
-            extension_id=options.extension_id,
-            preloaded_extension=options.extension_id is None,
+            preloaded_extension=True,
             worker_init_metadata=_WorkerInitMetadata(
                 api_key=options.api_key,
                 browser=_browser_session_metadata(connection.session_id, connection.region),

{stagehand-4.2.0a0.dev1587 → stagehand-4.2.0a0.dev1588}/src/stagehand/cdp_client.py

@@ -4,7 +4,7 @@ import asyncio
 import json
 import re
 import time
-from collections.abc import Mapping
+from collections.abc import Callable, Mapping
 from contextlib import suppress
 from dataclasses import dataclass
 from typing import Any, Literal, Protocol, cast
@@ -253,6 +253,7 @@ class CDPClient:
         *,
         cdp_url: str,
         extension_dir: str | None = None,
+        local_extension_dir: Callable[[], str] | None = None,
         extension_id: str | None = None,
         preloaded_extension: bool = False,
         service_worker_url_includes: str | None = None,
@@ -263,9 +264,18 @@ class CDPClient:
         lifecycle. Direct internal callers must provide cancellation themselves.
         CDP command acknowledgements inherit that same caller-owned cancellation.
         """
-        if sum((bool(extension_dir), bool(extension_id), preloaded_extension)) != 1:
+        if (
+            sum((
+                bool(extension_dir),
+                local_extension_dir is not None,
+                bool(extension_id),
+                preloaded_extension,
+            ))
+            != 1
+        ):
             raise ValueError(
-                "Exactly one of extension_dir, extension_id, or preloaded_extension is required"
+                "Exactly one of extension_dir, local_extension_dir, extension_id, "
+                "or preloaded_extension is required"
             )
 
         web_socket_debugger_url = await _resolve_browser_web_socket_url(cdp_url)
@@ -274,10 +284,14 @@ class CDPClient:
 
         try:
             resolved_extension_id = extension_id
-            if extension_dir is not None:
+            if local_extension_dir is not None:
+                resolved_extension_id = await client._resolve_extension(
+                    load_if_not_found=True, extension_dir=local_extension_dir
+                )
+            elif extension_dir is not None:
                 resolved_extension_id = await client._load_unpacked_extension(extension_dir)
-            if preloaded_extension:
-                resolved_extension_id = await client._discover_installed_stagehand_extension_id()
+            elif preloaded_extension:
+                resolved_extension_id = await client._resolve_extension(load_if_not_found=False)
             if resolved_extension_id is None:
                 raise RuntimeError("Stagehand extension ID was not resolved")
 
@@ -491,12 +505,41 @@ class CDPClient:
             if error.code == -32601 or "method not found" in str(error).lower():
                 raise RuntimeError(
                     "This Chrome build does not support Extensions.loadUnpacked. "
-                    "Launch with --load-extension and connect using extension_id instead."
+                    "Preload the Stagehand extension and connect using a Chrome build "
+                    "that supports Extensions.getExtensions."
                 ) from error
             raise
         return _required_string(loaded, "id", "Extensions.loadUnpacked")
 
-    async def _discover_installed_stagehand_extension_id(self) -> str:
+    async def _resolve_extension(
+        self,
+        *,
+        load_if_not_found: bool,
+        extension_dir: Callable[[], str] | None = None,
+    ) -> str:
+        extension_id: str | None = None
+        try:
+            extension_id = await self._get_installed_stagehand_extension_id()
+        except _CDPCommandError as error:
+            unsupported = error.method == "Extensions.getExtensions" and (
+                error.code == -32601
+                or "method not found" in str(error).lower()
+                or "wasn't found" in str(error).lower()
+            )
+            if not load_if_not_found or not unsupported:
+                raise
+        if extension_id is not None:
+            return extension_id
+        if load_if_not_found:
+            if extension_dir is None:
+                raise ValueError("extension_dir is required to load the Stagehand extension")
+            return await self._load_unpacked_extension(extension_dir())
+        raise RuntimeError(
+            "Stagehand extension is not installed in the connected browser. "
+            "The extension must be included when the Browserbase session is created."
+        )
+
+    async def _get_installed_stagehand_extension_id(self) -> str | None:
         response = await self.send_command("Extensions.getExtensions")
         installed = [
             extension
@@ -514,10 +557,7 @@ class CDPClient:
             raise RuntimeError(
                 "Stagehand extension is installed in the connected browser but is disabled."
             )
-        raise RuntimeError(
-            "Stagehand extension is not installed in the connected browser. "
-            "The extension must be included when the Browserbase session is created."
-        )
+        return None
 
     async def _wait_for_service_worker(
         self,

{stagehand-4.2.0a0.dev1587 → stagehand-4.2.0a0.dev1588}/src/stagehand/client_models.py

@@ -105,7 +105,8 @@ class LocalBrowserConnectOptions(WireModel):
     model_config = ConfigDict(extra="forbid", strict=True)
 
     cdp_url: Annotated[str, Field(min_length=1)]
-    extension_id: Annotated[str | None, Field(min_length=1)] = None
+    extension_id: str | None = None
+    """Deprecated and ignored. Omit this field; Stagehand discovers the extension automatically."""
 
 
 class BrowserbaseClientOptions(WireModel):
@@ -125,7 +126,8 @@ class BrowserbaseConnectOptions(WireModel):
     base_url: Annotated[str, Field(min_length=1)] = DEFAULT_BROWSERBASE_URL
     client_options: BrowserbaseClientOptions | None = None
     session_id: Annotated[str, Field(min_length=1)]
-    extension_id: Annotated[str | None, Field(min_length=1)] = None
+    extension_id: str | None = None
+    """Deprecated and ignored. Omit this field; Stagehand discovers the extension automatically."""
 
 
 class _BrowserbaseSearchOptions(WireModel):

{stagehand-4.2.0a0.dev1587 → stagehand-4.2.0a0.dev1588}/src/stagehand/client_types.py

@@ -66,6 +66,7 @@ class LocalBrowserLaunchOptions(TypedDict, total=False):
 class LocalBrowserConnectOptions(TypedDict):
     cdp_url: str
     extension_id: NotRequired[str]
+    """Deprecated and ignored. Omit this field; Stagehand discovers the extension automatically."""
 
 
 class BrowserbaseClientOptions(TypedDict, total=False):
@@ -85,6 +86,7 @@ class BrowserbaseConnectOptions(TypedDict):
     client_options: NotRequired[BrowserbaseClientOptions]
     session_id: str
     extension_id: NotRequired[str]
+    """Deprecated and ignored. Omit this field; Stagehand discovers the extension automatically."""
 
 
 class ClientLLM(TypedDict):