From db11ece30d9fff4f35583e9f759213a1b4a52754 Mon Sep 17 00:00:00 2001 From: miguel Date: Tue, 24 Jun 2025 12:17:53 -0700 Subject: [PATCH] fix bug on api_url load from .env --- stagehand/config.py | 3 ++- stagehand/main.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/stagehand/config.py b/stagehand/config.py index b4963c1b..a577230d 100644 --- a/stagehand/config.py +++ b/stagehand/config.py @@ -1,3 +1,4 @@ +import os from typing import Any, Callable, Literal, Optional from browserbase.types import SessionCreateParams as BrowserbaseSessionCreateParams @@ -42,7 +43,7 @@ class StagehandConfig(BaseModel): None, alias="projectId", description="Browserbase project ID" ) api_url: Optional[str] = Field( - "https://api.stagehand.browserbase.com/v1", + os.environ.get("STAGEHAND_API_URL", "https://api.stagehand.browserbase.com/v1"), alias="apiUrl", description="Stagehand API URL", ) diff --git a/stagehand/main.py b/stagehand/main.py index 6e384999..0de682e0 100644 --- a/stagehand/main.py +++ b/stagehand/main.py @@ -67,7 +67,7 @@ def __init__( self.config = config # Handle non-config parameters - self.api_url = self.config.api_url or os.getenv("STAGEHAND_API_URL") + self.api_url = self.config.api_url self.model_api_key = self.config.model_api_key or os.getenv("MODEL_API_KEY") self.model_name = self.config.model_name