Merge branch 'main' into local_change1

Author: leagend

CHANGELOG.md

@@ -1,5 +1,13 @@
 # Stagehand Python Changelog
 
+## 0.5.3
+
+### Patch Changes
+
+[#196](https://github.com/browserbase/stagehand-python/pull/196) [`93f5c97`](https://github.com/browserbase/stagehand-python/commit/93f5c97) Thanks @chrisreadsf, @miguelg719 and Derek Meegan! - remove duplicate project id if already passed to Stagehand
+[#203](https://github.com/browserbase/stagehand-python/pull/203) [`82c6fed`](https://github.com/browserbase/stagehand-python/commit/82c6fed) Thanks @miguelg719! - Bump openai dependency version
+[#198](https://github.com/browserbase/stagehand-python/pull/198) [`057b38b`](https://github.com/browserbase/stagehand-python/commit/057b38b) Thanks @Zach10za! - Fix draw_overlay on env:LOCAL
+
 ## 0.5.2
 
 ### Patch Changes

pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "stagehand"
-version = "0.5.2"
+version = "0.5.3"
 description = "Python SDK for Stagehand"
 readme = "README.md"
 classifiers = [ "Programming Language :: Python :: 3", "License :: OSI Approved :: MIT License", "Operating System :: OS Independent",]
 requires-python = ">=3.9"
-dependencies = [ "python-dotenv>=1.0.0", "pydantic>=1.10.0", "playwright>=1.42.1", "rich>=13.7.0", "openai>=1.83.0,<1.99.6", "anthropic>=0.51.0", "litellm>=1.72.0,<1.75.0", "nest-asyncio>=1.6.0",]
+dependencies = [ "httpx>=0.24.0", "python-dotenv>=1.0.0", "pydantic>=1.10.0", "playwright>=1.42.1", "requests>=2.31.0", "browserbase>=1.4.0", "rich>=13.7.0", "openai>=1.99.6", "anthropic>=0.51.0", "litellm>=1.72.0,<1.75.0", "nest-asyncio>=1.6.0",]
 [[project.authors]]
 name = "Browserbase, Inc."
 email = "[email protected]"

stagehand/config.py

@@ -1,7 +1,7 @@
 import os
-from typing import Any, Callable, Optional
-from urllib.parse import urlparse
+from typing import Any, Callable, Literal, Optional, Union
 
+from browserbase.types import SessionCreateParams as BrowserbaseSessionCreateParams
 from pydantic import BaseModel, ConfigDict, Field, field_validator, ValidationError
 
 from stagehand.schemas import AvailableModel
@@ -78,7 +78,14 @@ class StagehandConfig(BaseModel):
     dom_settle_timeout_ms: Optional[int] = Field(
         3000,
         alias="domSettleTimeoutMs",
-        description="Timeout in milliseconds to wait for DOM to settle after actions",
+        description="Timeout for DOM to settle (in ms)",
+    )
+    browserbase_session_create_params: Optional[
+        Union[BrowserbaseSessionCreateParams, dict[str, Any]]
+    ] = Field(
+        None,
+        alias="browserbaseSessionCreateParams",
+        description="Browserbase session create params",
     )
     enable_caching: Optional[bool] = Field(
         False, 
@@ -171,6 +178,16 @@ def validate_dom_settle_timeout(cls, v):
         """Validate DOM settle timeout is positive."""
         if v is not None and v < 0:
             raise ValueError("dom_settle_timeout_ms must be non-negative")
+
+    @field_validator("browserbase_session_create_params", mode="before")
+    @classmethod
+    def validate_browserbase_params(cls, v, info):
+        """Validate and convert browserbase session create params."""
+        if isinstance(v, dict) and "project_id" not in v:
+            values = info.data
+            project_id = values.get("project_id") or values.get("projectId")
+            if project_id:
+                v = {**v, "project_id": project_id}
         return v
 
     def with_overrides(self, **overrides) -> "StagehandConfig":

stagehand/handlers/observe_handler.py

@@ -118,7 +118,10 @@ async def observe(
 
         # Draw overlay if requested
         if options.draw_overlay:
-            await draw_observe_overlay(self.stagehand_page, elements_with_selectors)
+            await draw_observe_overlay(
+                page=self.stagehand_page,
+                elements=[el.model_dump() for el in elements_with_selectors],
+            )
 
         # Return the list of results without trying to attach _llm_response
         return elements_with_selectors

stagehand/utils.py

@@ -79,7 +79,7 @@ def format_simplified_tree(node: AccessibilityNode, level: int = 0) -> str:
     return result
 
 
-async def draw_observe_overlay(page, elements):
+async def draw_observe_overlay(page, elements: list[dict]):
     """
     Draw an overlay on the page highlighting the observed elements.