fix 1) formatting 2) test for api key

Author: chrisreadsf

stagehand/config.py

@@ -71,7 +71,9 @@ class StagehandConfig(BaseModel):
         alias="domSettleTimeoutMs",
         description="Timeout for DOM to settle (in ms)",
     )
-    browserbase_session_create_params: Optional[Union[BrowserbaseSessionCreateParams, dict[str, Any]]] = Field(
+    browserbase_session_create_params: Optional[
+        Union[BrowserbaseSessionCreateParams, dict[str, Any]]
+    ] = Field(
         None,
         alias="browserbaseSessionCreateParams",
         description="Browserbase session create params",
@@ -117,16 +119,16 @@ class StagehandConfig(BaseModel):
     )
 
     model_config = ConfigDict(populate_by_name=True)
-    
-    @field_validator('browserbase_session_create_params', mode='before')
+
+    @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:
+        if isinstance(v, dict) and "project_id" not in v:
             values = info.data
-            project_id = values.get('project_id') or values.get('projectId')
+            project_id = values.get("project_id") or values.get("projectId")
             if project_id:
-                v = {**v, 'project_id': project_id}
+                v = {**v, "project_id": project_id}
         return v
 
     def with_overrides(self, **overrides) -> "StagehandConfig":

tests/integration/local/test_core_local.py

@@ -20,7 +20,7 @@ def local_config(self):
             self_heal=True,
             wait_for_captcha_solves=False,
             system_prompt="You are a browser automation assistant for testing purposes.",
-            model_client_options={"apiKey": os.getenv("MODEL_API_KEY")},
+            model_client_options={"apiKey": os.getenv("MODEL_API_KEY") or os.getenv("OPENAI_API_KEY")},
             use_api=False,
         )