MODEL_API_KEY is the public facing env, but either MODEL_API_KEY or OPENAI_API_KEY can be used - one is required to be set.

Author: monadoid

examples/local_example.py

@@ -6,7 +6,8 @@
 - BROWSERBASE_API_KEY (can be any value in local mode)
 - BROWSERBASE_PROJECT_ID (can be any value in local mode)
 - MODEL_API_KEY (used for client configuration even in local mode)
-- OPENAI_API_KEY (used by the SEA server for LLM access)
+
+You can also set `OPENAI_API_KEY` if you prefer, but the example defaults to `MODEL_API_KEY`.
 
 Install the published wheel before running this script:
   `pip install stagehand-alpha`
@@ -22,13 +23,15 @@
 
 
 def main() -> None:
+    model_key = os.environ.get("MODEL_API_KEY")
     openai_key = os.environ.get("OPENAI_API_KEY")
-    if not openai_key:
-        sys.exit("Set the OPENAI_API_KEY environment variable to run the local server.")
+
+    if not model_key and not openai_key:
+        sys.exit("Set MODEL_API_KEY to run the local server.")
 
     client = Stagehand(
         server="local",
-        local_openai_api_key=openai_key,
+        local_openai_api_key=openai_key or model_key,
         local_ready_timeout_s=30.0,
     )
 

test_local_mode.py

@@ -10,9 +10,9 @@
 from stagehand import Stagehand
 
 # Set required API key for LLM operations
-if not os.environ.get("OPENAI_API_KEY"):
-    print("❌ Error: OPENAI_API_KEY environment variable not set")  # noqa: T201
-    print("   Set it with: export OPENAI_API_KEY='sk-proj-...'")  # noqa: T201
+if not os.environ.get("MODEL_API_KEY") and not os.environ.get("OPENAI_API_KEY"):
+    print("❌ Error: MODEL_API_KEY or OPENAI_API_KEY environment variable not set")  # noqa: T201
+    print("   Set it with: export MODEL_API_KEY='sk-proj-...'")  # noqa: T201
     sys.exit(1)
 
 print("🚀 Testing local server mode...")  # noqa: T201
@@ -24,7 +24,7 @@
         server="local",
         browserbase_api_key="local",  # Dummy value - not used in local mode
         browserbase_project_id="local",  # Dummy value - not used in local mode
-        model_api_key=os.environ["OPENAI_API_KEY"],
+        model_api_key=os.environ.get("MODEL_API_KEY") or os.environ["OPENAI_API_KEY"],
         local_headless=True,
         local_port=0,  # Auto-pick free port
         local_ready_timeout_s=15.0,  # Give it time to start