allow passing chrome path as init param not only env var

Author: monadoid

src/stagehand/_client.py

@@ -66,6 +66,7 @@ def __init__(
         local_host: str = "127.0.0.1",
         local_port: int = 0,
         local_headless: bool = True,
+        local_chrome_path: str | None = None,
         local_ready_timeout_s: float = 10.0,
         local_openai_api_key: str | None = None,
         local_shutdown_on_close: bool = True,
@@ -100,6 +101,7 @@ def __init__(
         self._local_host = local_host
         self._local_port = local_port
         self._local_headless = local_headless
+        self._local_chrome_path = local_chrome_path
         self._local_ready_timeout_s = local_ready_timeout_s
         self._local_openai_api_key = local_openai_api_key
         self._local_shutdown_on_close = local_shutdown_on_close
@@ -134,6 +136,7 @@ def __init__(
                     headless=local_headless,
                     ready_timeout_s=local_ready_timeout_s,
                     openai_api_key=openai_api_key,
+                    chrome_path=local_chrome_path,
                     shutdown_on_close=local_shutdown_on_close,
                 ),
                 _local_stagehand_binary_path=_local_stagehand_binary_path,
@@ -350,6 +353,7 @@ def __init__(
         local_host: str = "127.0.0.1",
         local_port: int = 0,
         local_headless: bool = True,
+        local_chrome_path: str | None = None,
         local_ready_timeout_s: float = 10.0,
         local_openai_api_key: str | None = None,
         local_shutdown_on_close: bool = True,
@@ -384,6 +388,7 @@ def __init__(
         self._local_host = local_host
         self._local_port = local_port
         self._local_headless = local_headless
+        self._local_chrome_path = local_chrome_path
         self._local_ready_timeout_s = local_ready_timeout_s
         self._local_openai_api_key = local_openai_api_key
         self._local_shutdown_on_close = local_shutdown_on_close
@@ -417,6 +422,7 @@ def __init__(
                     headless=local_headless,
                     ready_timeout_s=local_ready_timeout_s,
                     openai_api_key=openai_api_key,
+                    chrome_path=local_chrome_path,
                     shutdown_on_close=local_shutdown_on_close,
                 ),
                 _local_stagehand_binary_path=_local_stagehand_binary_path,

src/stagehand/lib/sea_server.py

@@ -25,6 +25,7 @@ class SeaServerConfig:
     headless: bool
     ready_timeout_s: float
     openai_api_key: str | None
+    chrome_path: str | None
     shutdown_on_close: bool
 
 
@@ -179,6 +180,9 @@ def _start_sync(self) -> tuple[str, subprocess.Popen[bytes]]:
         proc_env["HEADLESS"] = "true" if self._config.headless else "false"
         if self._config.openai_api_key:
             proc_env["OPENAI_API_KEY"] = self._config.openai_api_key
+        if self._config.chrome_path:
+            proc_env["CHROME_PATH"] = self._config.chrome_path
+            proc_env["LIGHTHOUSE_CHROMIUM_PATH"] = self._config.chrome_path
 
         preexec_fn = None
         creationflags = 0
@@ -226,6 +230,9 @@ async def _start_async(self) -> tuple[str, subprocess.Popen[bytes]]:
         proc_env["HEADLESS"] = "true" if self._config.headless else "false"
         if self._config.openai_api_key:
             proc_env["OPENAI_API_KEY"] = self._config.openai_api_key
+        if self._config.chrome_path:
+            proc_env["CHROME_PATH"] = self._config.chrome_path
+            proc_env["LIGHTHOUSE_CHROMIUM_PATH"] = self._config.chrome_path
 
         preexec_fn = None
         creationflags = 0