add two params to config and fix test

Author: filip-michalsky

stagehand/config.py

@@ -30,6 +30,8 @@ class StagehandConfig(BaseModel):
         headless (bool): Run browser in headless mode
         system_prompt (Optional[str]): System prompt to use for LLM interactions.
         local_browser_launch_options (Optional[dict[str, Any]]): Local browser launch options.
+        use_api (bool): Whether to use API mode.
+        experimental (bool): Enable experimental features.
     """
 
     env: Literal["BROWSERBASE", "LOCAL"] = "BROWSERBASE"
@@ -94,6 +96,12 @@ class StagehandConfig(BaseModel):
         alias="localBrowserLaunchOptions",
         description="Local browser launch options",
     )
+    use_api: bool = Field(
+        True, alias="useAPI", description="Whether to use API mode"
+    )
+    experimental: bool = Field(
+        False, description="Enable experimental features"
+    )
 
     model_config = ConfigDict(populate_by_name=True)
 

tests/integration/api/test_core_api.py

@@ -33,7 +33,7 @@ def browserbase_config(self):
     @pytest_asyncio.fixture
     async def stagehand_api(self, browserbase_config):
         """Create a Stagehand instance for BROWSERBASE API testing"""
-        if not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_URL_API")):
+        if not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_API_URL")):
             pytest.skip("Browserbase credentials and API URL not available")
 
         stagehand = Stagehand(config=browserbase_config)
@@ -45,7 +45,7 @@ async def stagehand_api(self, browserbase_config):
     @pytest.mark.integration
     @pytest.mark.api
     @pytest.mark.skipif(
-        not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_URL_API")),
+        not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_API_URL")),
         reason="Browserbase credentials and API URL are not available for API integration tests",
     )
     async def test_stagehand_api_initialization(self, stagehand_api):
@@ -56,7 +56,7 @@ async def test_stagehand_api_initialization(self, stagehand_api):
     @pytest.mark.integration
     @pytest.mark.api
     @pytest.mark.skipif(
-        not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_URL_API")),
+        not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_API_URL")),
         reason="Browserbase credentials and API URL are not available for API integration tests",
     )
     async def test_api_observe_and_act_workflow(self, stagehand_api):
@@ -97,7 +97,7 @@ async def test_api_observe_and_act_workflow(self, stagehand_api):
     @pytest.mark.integration
     @pytest.mark.api
     @pytest.mark.skipif(
-        not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_URL_API")),
+        not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_API_URL")),
         reason="Browserbase credentials and API URL are not available for API integration tests",
     )
     async def test_api_basic_navigation_and_observe(self, stagehand_api):
@@ -123,7 +123,7 @@ async def test_api_basic_navigation_and_observe(self, stagehand_api):
     @pytest.mark.integration
     @pytest.mark.api
     @pytest.mark.skipif(
-        not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_URL_API")),
+        not (os.getenv("BROWSERBASE_API_KEY") and os.getenv("BROWSERBASE_PROJECT_ID") and os.getenv("STAGEHAND_API_URL")),
         reason="Browserbase credentials and API URL are not available for API integration tests",
     )
     async def test_api_extraction_functionality(self, stagehand_api):