stagehand 4.2.0a0.dev1525__tar.gz → 4.2.0a0.dev1526__tar.gz

{stagehand-4.2.0a0.dev1525 → stagehand-4.2.0a0.dev1526}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stagehand
-Version: 4.2.0a0.dev1525
+Version: 4.2.0a0.dev1526
 Summary: Browserbase's SDK for building browser agents
 Keywords: ai,browser,automation,web-scraping,testing
 Author: Browserbase
@@ -8,6 +8,7 @@ License-Expression: MIT
 Requires-Dist: browserbase>=1.15,<2
 Requires-Dist: opentelemetry-api>=1.35,<2
 Requires-Dist: pydantic>=2.12,<3
+Requires-Dist: typing-extensions>=4.12,<5
 Requires-Dist: websockets>=16.1.1
 Requires-Python: >=3.11
 Project-URL: Homepage, https://stagehand.dev

{stagehand-4.2.0a0.dev1525 → stagehand-4.2.0a0.dev1526}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "stagehand"
-version = "4.2.0a0.dev1525"
+version = "4.2.0a0.dev1526"
 description = "Browserbase's SDK for building browser agents"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -8,6 +8,7 @@ dependencies = [
   "browserbase>=1.15,<2",
   "opentelemetry-api>=1.35,<2",
   "pydantic>=2.12,<3",
+  "typing_extensions>=4.12,<5",
   "websockets>=16.1.1",
 ]
 license = "MIT"
@@ -39,12 +40,13 @@ select = ["ANN", "E", "F", "I", "PYI", "UP"]
 [tool.ty.environment]
 python-version = "3.11"
 
-[tool.ty.src]
-exclude = ["src/stagehand/_generated"]
-
 [tool.ty.rules]
 missing-type-argument = "error"
-possibly-unresolved-reference = "warn"
+possibly-unresolved-reference = "error"
+unsound-return-statement = "error"
+dynamic-function-decorator-return = "error"
+missing-override-decorator = "error"
+unused-ignore-comment = "error"
 
 [tool.pytest.ini_options]
 asyncio_mode = "auto"
@@ -58,5 +60,5 @@ dev = [
   "pytest==9.0.2",
   "pytest-asyncio==1.3.0",
   "ruff==0.15.7",
-  "ty==0.0.56",
+  "ty==0.0.82",
 ]

{stagehand-4.2.0a0.dev1525 → stagehand-4.2.0a0.dev1526}/src/stagehand/_extension/service-worker.js

@@ -7359,7 +7359,7 @@ function isRecord$1(value) {
 }
 var package_default$1 = {
 	name: "@browserbasehq/stagehand-protocol",
-	version: "2.1.0-alpha-d47450802f4279a301f4c90a98788e2fb7afd85b",
+	version: "2.1.0-alpha-ea2789ca56e8a7ee154850de37657823c98604f8",
 	"private": true,
 	type: "module",
 	exports: {
@@ -53981,7 +53981,7 @@ var WebTracerProvider = class extends BasicTracerProvider {
 };
 var package_default = {
 	name: "@browserbasehq/stagehand-extension",
-	version: "1.1.0-alpha-d47450802f4279a301f4c90a98788e2fb7afd85b",
+	version: "1.1.0-alpha-ea2789ca56e8a7ee154850de37657823c98604f8",
 	"private": true,
 	type: "module",
 	scripts: {

{stagehand-4.2.0a0.dev1525 → stagehand-4.2.0a0.dev1526}/src/stagehand/_generated/models.py

@@ -2117,13 +2117,17 @@ class StagehandExtractParams(WireModel):
     )
     page_id: Annotated[StrictStr, Field(min_length=1)]
     instruction: Annotated[StrictStr, Field(min_length=1)]
-    schema_: Annotated[Optional[FieldSchema0], Field(alias="schema", validate_default=True)] = {
-        "$schema": "https://json-schema.org/draft/2020-12/schema",
-        "type": "object",
-        "properties": {"extraction": {"type": "string"}},
-        "required": ["extraction"],
-        "additionalProperties": False,
-    }
+    schema_: Annotated[Optional[FieldSchema0], Field(alias="schema", validate_default=True)] = Field(
+        default_factory=lambda: FieldSchema0.model_validate(
+            {
+                "$schema": "https://json-schema.org/draft/2020-12/schema",
+                "type": "object",
+                "properties": {"extraction": {"type": "string"}},
+                "required": ["extraction"],
+                "additionalProperties": False,
+            }
+        )
+    )
     options: Optional[ExtractOptions] = None
 
 
@@ -2147,9 +2151,13 @@ class StagehandInitParams(WireModel):
     browser: Optional[BrowserSessionMetadata] = None
     model: Optional[Union[ModelConfig, ClientModelReference]] = None
     """Default model configuration; when omitted and a Browserbase Model Gateway session is available, Browserbase selects a model automatically for inference calls"""
-    telemetry: Annotated[TelemetryConfig, Field(validate_default=True)] = {
-        "traces": {"endpoint": "https://example.com/v1/traces", "headers": {}}
-    }
+    telemetry: Annotated[TelemetryConfig, Field(validate_default=True)] = Field(
+        default_factory=lambda: TelemetryConfig.model_validate(
+            {
+                "traces": {"endpoint": "https://example.com/v1/traces", "headers": {}}
+            }
+        )
+    )
     log_level: LogLevel = LogLevel.info
     system_prompt: Optional[StrictStr] = None
     self_heal: Optional[StrictBool] = None

{stagehand-4.2.0a0.dev1525 → stagehand-4.2.0a0.dev1526}/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-d47450802f4279a301f4c90a98788e2fb7afd85b'
+STAGEHAND_PROTOCOL_VERSION = '2.1.0-alpha-ea2789ca56e8a7ee154850de37657823c98604f8'

{stagehand-4.2.0a0.dev1525 → stagehand-4.2.0a0.dev1526}/src/stagehand/_validation.py

@@ -3,6 +3,7 @@ from typing import Annotated
 from urllib.parse import urlparse
 
 from pydantic import AfterValidator, AnyUrl, BaseModel, ConfigDict, TypeAdapter, model_validator
+from typing_extensions import override
 
 _url_adapter = TypeAdapter(AnyUrl)
 
@@ -18,6 +19,7 @@ WireUrl = Annotated[str, AfterValidator(_validate_url)]
 class WireModel(BaseModel):
     model_config = ConfigDict(validate_assignment=True)
 
+    @override
     def model_post_init(self, context: object, /) -> None:
         super().model_post_init(context)
         # Protocol defaults must survive `model_dump(exclude_unset=True)` at the wire boundary.

{stagehand-4.2.0a0.dev1525 → stagehand-4.2.0a0.dev1526}/src/stagehand/browser.py

@@ -15,7 +15,7 @@ from collections.abc import Awaitable, Callable, Mapping, Sequence
 from contextlib import suppress
 from dataclasses import dataclass, field
 from pathlib import Path, PureWindowsPath
-from typing import TYPE_CHECKING, Any, Literal, Protocol
+from typing import TYPE_CHECKING, Any, Literal, Protocol, cast
 
 if TYPE_CHECKING:
     from .browser_context import BrowserContext
@@ -281,7 +281,8 @@ class _LocalBrowserOptions(Protocol):
 
 
 class _WaitableChromeProcess(Protocol):
-    returncode: int | None
+    @property
+    def returncode(self) -> int | None: ...
 
     async def wait(self) -> int: ...
 
@@ -721,7 +722,7 @@ async def _launch_local_browser(options: _LocalBrowserOptions) -> ResolvedBrowse
     process: _ChromeProcess | None = None
     try:
         try:
-            process = await asyncio.create_subprocess_exec(
+            launched_process = await asyncio.create_subprocess_exec(
                 chrome_path,
                 *flags,
                 stdin=asyncio.subprocess.DEVNULL,
@@ -729,9 +730,10 @@ async def _launch_local_browser(options: _LocalBrowserOptions) -> ResolvedBrowse
                 stderr=asyncio.subprocess.DEVNULL,
                 start_new_session=sys.platform != "win32",
             )
+            process = launched_process
         except OSError as error:
             raise RuntimeError(f"Failed to start Chrome: {error}") from error
-        await _wait_for_chrome(f"http://127.0.0.1:{port}", process)
+        await _wait_for_chrome(f"http://127.0.0.1:{port}", launched_process)
     except BaseException as launch_error:
         try:
             if process is not None:
@@ -746,7 +748,7 @@ async def _launch_local_browser(options: _LocalBrowserOptions) -> ResolvedBrowse
         raise
 
     async def close() -> None:
-        await _close_local_chrome(process, profile)
+        await _close_local_chrome(launched_process, profile)
 
     return ResolvedBrowserSource(
         cdp_url=f"http://127.0.0.1:{port}",
@@ -838,10 +840,10 @@ def _read_chrome_version(cdp_url: str) -> dict[str, object]:
         url,
         timeout=_CHROME_REQUEST_TIMEOUT_SECONDS,
     ) as response:
-        value: Any = json.load(response)
+        value: object = json.load(response)
     if not isinstance(value, dict):
         raise RuntimeError("Chrome version endpoint returned invalid JSON")
-    return value
+    return {str(key): item for key, item in value.items()}
 
 
 def _chrome_exited_before_ready_error(returncode: int | None) -> RuntimeError:
@@ -992,7 +994,8 @@ def _should_disable_chromium_sandbox(
 ) -> bool:
     platform = sys.platform if platform is None else platform
     environment = os.environ if environment is None else environment
-    getuid = getattr(os, "getuid", None) if getuid is None else getuid
+    if getuid is None:
+        getuid = cast("Callable[[], int] | None", getattr(os, "getuid", None))
     return (
         bool(environment.get("CI"))
         or options.chromium_sandbox is False

{stagehand-4.2.0a0.dev1525 → stagehand-4.2.0a0.dev1526}/src/stagehand/client_models.py

@@ -66,7 +66,8 @@ Cache = bool | CacheOptions
 def _cache_config(cache: CacheInput) -> bool | dict[str, int]:
     if isinstance(cache, bool):
         return cache
-    return CacheOptions.model_validate(cache).model_dump(exclude_none=True)
+    options = CacheOptions.model_validate(cache)
+    return {"threshold": options.threshold} if options.threshold is not None else {}
 
 
 class LocalViewport(WireModel):