also allow overriding BROWSERBASE_API_URL via env

Author: pirate

.github/workflows/test.yml

@@ -22,6 +22,7 @@ jobs:
       - name: Run tests
         run: cargo test --verbose
         env:
+          BROWSERBASE_API_URL: ${{ secrets.BROWSERBASE_API_URL }}
           STAGEHAND_API_URL: ${{ secrets.STAGEHAND_API_URL }}
           BROWSERBASE_API_KEY: ${{ secrets.BROWSERBASE_API_KEY }}
           BROWSERBASE_PROJECT_ID: ${{ secrets.BROWSERBASE_PROJECT_ID }}

README.md

@@ -179,8 +179,9 @@ GROQ_API_KEY=your_groq_key                          # Groq
 CEREBRAS_API_KEY=your_cerebras_key                  # Cerebras
 DEEPSEEK_API_KEY=your_deepseek_key                  # DeepSeek
 
-# Optional: Custom Stagehand API URL (defaults to https://api.stagehand.browserbase.com/v1)
-STAGEHAND_API_URL=https://api.stagehand.browserbase.com/v1
+# Optional: Custom API URLs
+STAGEHAND_API_URL=https://api.stagehand.browserbase.com/v1  # Stagehand API (default)
+BROWSERBASE_API_URL=https://api.browserbase.com/v1          # Browserbase API (default)
 ```
 
 The SDK checks for model API keys in the order listed above and uses the first one found.

src/lib.rs

@@ -271,6 +271,9 @@ pub struct V3Options {
 /// Default Stagehand API URL
 pub const DEFAULT_STAGEHAND_API_URL: &str = "https://api.stagehand.browserbase.com/v1";
 
+/// Default Browserbase API URL
+pub const DEFAULT_BROWSERBASE_API_URL: &str = "https://api.browserbase.com/v1";
+
 /// Transport selection for connecting to Stagehand API
 #[derive(Debug, Clone, PartialEq)]
 pub enum TransportChoice {
@@ -1032,9 +1035,12 @@ impl Stagehand {
         let api_key = std::env::var("BROWSERBASE_API_KEY")
             .map_err(|_| StagehandError::MissingApiKey("BROWSERBASE_API_KEY".to_string()))?;
 
+        let browserbase_api_url = std::env::var("BROWSERBASE_API_URL")
+            .unwrap_or_else(|_| DEFAULT_BROWSERBASE_API_URL.to_string());
+
         let client = reqwest::Client::new();
         let response = client
-            .get(format!("https://api.browserbase.com/v1/sessions/{}", session_id))
+            .get(format!("{}/sessions/{}", browserbase_api_url, session_id))
             .header("x-bb-api-key", &api_key)
             .send()
             .await