fix: load `GOOGLE_API_KEY` from env #1412

seanmcguire12 · 2025-12-13T03:39:52Z

why

act, extract, & observe fail, and stagehand logs AI_LoadAPIKeyError if a user attempts to use a google LLM, and has GOOGLE_API_KEY in their .env instead of GOOGLE_GENERATIVE_AI_API_KEY or GEMINI_API_KEY

what changed

this PR widens the accepted env vars for google models to accept GOOGLE_API_KEY

Summary by cubic

Allow GOOGLE_API_KEY for Google models by expanding the env var lookup. Fixes key-loading failures in act, extract, and observe when users set GOOGLE_API_KEY in .env.

^{Written for commit 4984318. Summary will update automatically on new commits.}

changeset-bot · 2025-12-13T03:39:56Z

🦋 Changeset detected

Latest commit: 4984318

The changes in this PR will be included in the next version bump.

This PR includes changesets to release 3 packages

Name	Type
@browserbasehq/stagehand	Patch
@browserbasehq/stagehand-evals	Patch
@browserbasehq/stagehand-server	Patch

Not sure what this means? Click here to learn what changesets are.

Click here if you're a maintainer who wants to add another changeset to this PR

cubic-dev-ai

No issues found across 2 files

greptile-apps · 2025-12-13T03:43:29Z

Greptile Overview

Greptile Summary

Adds GOOGLE_API_KEY to the list of accepted environment variables for Google models specified with the google/ provider prefix (e.g., google/gemini-2.5-flash).

Changes

Updated providerEnvVarMap for the google provider to accept three environment variable names: GEMINI_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY, and GOOGLE_API_KEY
This fix prevents AI_LoadAPIKeyError when users have GOOGLE_API_KEY in their .env file instead of the previously required GEMINI_API_KEY or GOOGLE_GENERATIVE_AI_API_KEY

Architecture Context

The codebase supports two formats for specifying Google models:

With provider prefix (e.g., google/gemini-2.5-flash): Uses the updated google provider mapping
Without prefix (e.g., gemini-1.5-flash): Falls back to GoogleClient constructor's google_legacy mapping, which already supported GOOGLE_API_KEY

This PR ensures consistent behavior across both model specification formats.

Confidence Score: 5/5

This PR is safe to merge with no identified risks
The change is minimal, well-scoped, and follows existing patterns. It adds a commonly-used environment variable name to an existing array without breaking changes. The fix addresses a real user pain point where GOOGLE_API_KEY was not recognized for google/ prefixed models, while maintaining backward compatibility with existing environment variable names.
No files require special attention

Important Files Changed

File Analysis

Filename	Score	Overview
.changeset/silent-clubs-retire.md	5/5	Adds changeset for patch release documenting the fix to load `GOOGLE_API_KEY` from env
packages/core/lib/utils.ts	5/5	Adds `GOOGLE_API_KEY` to accepted environment variables for google provider, fixing API key loading for users who use this common env var name

Sequence Diagram

sequenceDiagram
    participant User
    participant V3
    participant LLMProvider
    participant GoogleClient
    participant loadApiKeyFromEnv
    participant providerEnvVarMap

    User->>V3: Initialize with model (e.g., google/gemini-2.5-flash)
    V3->>V3: Extract provider from modelName.split("/")[0]
    V3->>loadApiKeyFromEnv: loadApiKeyFromEnv(google, logger)
    loadApiKeyFromEnv->>providerEnvVarMap: Lookup google key
    providerEnvVarMap-->>loadApiKeyFromEnv: Array of 3 env var names
    loadApiKeyFromEnv->>loadApiKeyFromEnv: Check each env var in order
    loadApiKeyFromEnv-->>V3: Return first non-empty value
    V3->>V3: Set modelClientOptions with apiKey
    V3->>LLMProvider: getClient(modelName, clientOptions)
    LLMProvider->>GoogleClient: new GoogleClient(logger, modelName, clientOptions)
    GoogleClient->>GoogleClient: Check if clientOptions has apiKey
    GoogleClient-->>LLMProvider: Return client instance
    LLMProvider-->>V3: Return client
    V3-->>User: Stagehand initialized

    Note over V3,GoogleClient: For models without prefix (e.g., gemini-1.5-flash)
    User->>V3: Initialize with model gemini-1.5-flash
    V3->>V3: Extract provider: gemini-1.5-flash (no slash)
    V3->>loadApiKeyFromEnv: loadApiKeyFromEnv(gemini-1.5-flash, logger)
    loadApiKeyFromEnv->>providerEnvVarMap: Lookup gemini-1.5-flash key
    providerEnvVarMap-->>loadApiKeyFromEnv: undefined
    loadApiKeyFromEnv-->>V3: Return undefined
    V3->>LLMProvider: getClient(modelName, clientOptions with no apiKey)
    LLMProvider->>GoogleClient: new GoogleClient(logger, modelName, clientOptions)
    GoogleClient->>GoogleClient: clientOptions has no apiKey
    GoogleClient->>loadApiKeyFromEnv: loadApiKeyFromEnv(google_legacy, logger)
    loadApiKeyFromEnv->>providerEnvVarMap: Lookup google_legacy key
    providerEnvVarMap-->>loadApiKeyFromEnv: Single env var name
    loadApiKeyFromEnv-->>GoogleClient: Return env var value
    GoogleClient-->>LLMProvider: Return client instance
    LLMProvider-->>V3: Return client
    V3-->>User: Stagehand initialized

greptile-apps

_{1 file reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}

@tkattkat

load GOOGLE_API_KEY from env

e8e9862

cubic-dev-ai bot reviewed Dec 13, 2025

View reviewed changes

greptile-apps bot reviewed Dec 13, 2025

View reviewed changes

miguelg719 approved these changes Dec 13, 2025

View reviewed changes

add GOOGLE_API_KEY to google cua client

4984318

seanmcguire12 merged commit cd01f29 into main Dec 13, 2025
15 checks passed

This was referenced Dec 13, 2025

Version Packages #1414

Merged

Version Packages CloudEngineHub/stagehand#1

Open

seanmcguire12 mentioned this pull request Dec 20, 2025

fix: Add GOOGLE_API_KEY to google provider environment variables #1285

Closed

4 tasks

This was referenced Dec 20, 2025

Version Packages kenchikuliu/stagehand#1

Open

Version Packages shaneholloman/stagehand#1

Open

shrey150 mentioned this pull request Dec 23, 2025

modelApiKey is required when using BROWSERBASE environment #1274

Closed

2 tasks

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

fix: load `GOOGLE_API_KEY` from env #1412

fix: load `GOOGLE_API_KEY` from env #1412

Uh oh!

seanmcguire12 commented Dec 13, 2025 •

edited by cubic-dev-ai bot

Loading

Uh oh!

changeset-bot bot commented Dec 13, 2025 •

edited

Loading

Uh oh!

cubic-dev-ai bot left a comment

Uh oh!

greptile-apps bot commented Dec 13, 2025

Uh oh!

greptile-apps bot left a comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

fix: load GOOGLE_API_KEY from env #1412

fix: load GOOGLE_API_KEY from env #1412

Uh oh!

Conversation

seanmcguire12 commented Dec 13, 2025 • edited by cubic-dev-ai bot Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

why

what changed

Summary by cubic

Uh oh!

changeset-bot bot commented Dec 13, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

🦋 Changeset detected

Uh oh!

cubic-dev-ai bot left a comment

Choose a reason for hiding this comment

Uh oh!

greptile-apps bot commented Dec 13, 2025

Greptile Overview

Greptile Summary

Changes

Architecture Context

Confidence Score: 5/5

Important Files Changed

Sequence Diagram

Uh oh!

greptile-apps bot left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

fix: load `GOOGLE_API_KEY` from env #1412

fix: load `GOOGLE_API_KEY` from env #1412

seanmcguire12 commented Dec 13, 2025 •

edited by cubic-dev-ai bot

Loading

changeset-bot bot commented Dec 13, 2025 •

edited

Loading