reverting more files except test folder

Author: filip-michalsky

.env.example

@@ -2,4 +2,4 @@ MODEL_API_KEY = "your-favorite-llm-api-key"
 BROWSERBASE_API_KEY = "browserbase-api-key"
 BROWSERBASE_PROJECT_ID = "browserbase-project-id"
 STAGEHAND_API_URL = "api_url"
-STAGEHAND_ENV= "LOCAL or BROWSERBASE"
+STAGEHAND_ENV= "LOCAL or BROWSERBASE"

run_integration_tests.sh

@@ -166,4 +166,4 @@ async def _extract_page_text(self) -> ExtractResult:
 
         tree = await get_accessibility_tree(self.stagehand_page, self.logger)
         output_string = tree["simplified"]
-        return ExtractResult(data=output_string)
+        return ExtractResult(data=output_string)

stagehand/handlers/extract_handler.py

@@ -88,27 +88,7 @@ def __init__(
         self.wait_for_captcha_solves = self.config.wait_for_captcha_solves
         self.system_prompt = self.config.system_prompt
         self.verbose = self.config.verbose
-
-        # Smart environment detection
-        if self.config.env:
-            self.env = self.config.env.upper()
-        else:
-            # Auto-detect environment based on available configuration
-            has_browserbase_config = bool(
-                self.browserbase_api_key and self.browserbase_project_id
-            )
-            has_local_config = bool(self.config.local_browser_launch_options)
-
-            if has_local_config and not has_browserbase_config:
-                # Local browser options specified but no Browserbase config
-                self.env = "LOCAL"
-            elif not has_browserbase_config and not has_local_config:
-                # No configuration specified, default to LOCAL for easier local development
-                self.env = "LOCAL"
-            else:
-                # Default to BROWSERBASE if Browserbase config is available
-                self.env = "BROWSERBASE"
-
+        self.env = self.config.env.upper() if self.config.env else "BROWSERBASE"
         self.local_browser_launch_options = (
             self.config.local_browser_launch_options or {}
         )
@@ -218,14 +198,9 @@ def cleanup_handler(sig, frame):
                 return
 
             self.__class__._cleanup_called = True
-            if self.env == "BROWSERBASE":
-                print(
-                    f"\n[{signal.Signals(sig).name}] received. Ending Browserbase session..."
-                )
-            else:
-                print(
-                    f"\n[{signal.Signals(sig).name}] received. Cleaning up Stagehand resources..."
-                )
+            print(
+                f"\n[{signal.Signals(sig).name}] received. Ending Browserbase session..."
+            )
 
             try:
                 # Try to get the current event loop
@@ -264,15 +239,9 @@ async def _async_cleanup(self):
         """Async cleanup method called from signal handler."""
         try:
             await self.close()
-            if self.env == "BROWSERBASE" and self.session_id:
-                print(f"Session {self.session_id} ended successfully")
-            else:
-                print("Stagehand resources cleaned up successfully")
+            print(f"Session {self.session_id} ended successfully")
         except Exception as e:
-            if self.env == "BROWSERBASE":
-                print(f"Error ending Browserbase session: {str(e)}")
-            else:
-                print(f"Error cleaning up Stagehand resources: {str(e)}")
+            print(f"Error ending Browserbase session: {str(e)}")
         finally:
             # Force exit after cleanup completes (or fails)
             # Use os._exit to avoid any further Python cleanup that might hang

stagehand/main.py

@@ -14,9 +14,7 @@
     TreeResult,
 )
 from .agent import (
-    AgentConfig,
-    AgentExecuteOptions,
-    AgentResult,
+    AgentConfig
 )
 from .llm import (
     ChatMessage,