add auto-bedrock support based on bedrock/provider.model-name

[pirate]

why

Fixes: browserbase/stagehand-go#43

Adds automatic bedrock AI SDK client support when modelName ~= 'bedrock/...'

what changed

test plan

---

Summary by cubic

Adds automatic Amazon Bedrock support by routing models named bedrock/{provider}.{model} to the Bedrock client. This enables zero-API-key setup using your AWS credentials.

• New Features

  • Route bedrock/... model names to @ai-sdk/amazon-bedrock.
  • Treat Bedrock as no-API-key provider (uses AWS IAM/default credentials).
  • Example usage: bedrock/anthropic.claude-3.

• Dependencies

  • Added @ai-sdk/amazon-bedrock as an optional dependency.

^{Written for commit e1d7c4c. Summary will update on new commits. Review in cubic}

[changeset-bot]

⚠️ No Changeset found

Latest commit: e1d7c4c

Merging this PR will not cause a version bump for any packages. If these changes should not result in a new version, you're good to go. If these changes should result in a version bump, you need to add a changeset.

This PR includes no changesets

When changesets are added to this PR, you'll see the packages that this PR includes changesets for and the associated semver types

Click here to learn what changesets are, and how to add one.

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

[cubic-dev-ai]

No issues found across 4 files

Confidence score: 5/5

• Automated review surfaced no issues in the provided summaries.
• No files require special attention.

Architecture diagram

sequenceDiagram
    participant User as Client Code
    participant Factory as LLMProvider.ts
    participant Utils as utils.ts
    participant SDK as @ai-sdk/amazon-bedrock
    participant AWS as AWS Cloud

    Note over User, AWS: NEW: Bedrock Client Initialization

    User->>Factory: init("bedrock/anthropic.claude-3")
    Factory->>Factory: Parse provider string ("bedrock")

    Factory->>Utils: checkApiKey("bedrock")
    
    alt NEW: "bedrock" in providersWithoutApiKey
        Utils->>Utils: Skip Env Var validation
        Note right of Utils: Auth relies on standard AWS environment<br/>(AWS_PROFILE or AWS_ACCESS_KEY_ID)
        Utils-->>Factory: Validated (No Key Req)
    else Other Provider (e.g. openai)
        Utils->>Utils: Check specific Env Var (OPENAI_API_KEY)
        Utils-->>Factory: Validated (Key Found)
    end

    Factory->>Factory: Lookup provider in AISDKProviders
    Factory->>SDK: NEW: createAmazonBedrock() / bedrock object
    SDK-->>Factory: Provider Instance
    Factory-->>User: LLMClient Wrapper

    Note over User, AWS: Runtime Interaction

    User->>Factory: generate(...)
    Factory->>SDK: streamText / generateText
    SDK->>AWS: Invoke Model (via AWS SDK Creds)
    AWS-->>SDK: AI Response
    SDK-->>User: Result

Loading

[greptile-apps]

Greptile Overview

Greptile Summary

Added automatic AWS Bedrock support for model names following the bedrock/provider.model-name pattern, mirroring the existing implementation for Ollama.

Key changes:

• Imported bedrock and createAmazonBedrock from @ai-sdk/amazon-bedrock in LLMProvider.ts:21
• Registered bedrock provider in both AISDKProviders (line 38) and AISDKProvidersWithAPIKey (line 54) dictionaries
• Added bedrock to providersWithoutApiKey set in utils.ts:698 to skip API key validation warnings
• Added @ai-sdk/amazon-bedrock v4.0.31 as an optional dependency

The implementation follows the established pattern used for Ollama in PR #1504, enabling users to specify models like bedrock/amazon.nova-pro-latest without manually creating a custom client. Bedrock uses AWS credentials from the environment rather than requiring an explicit API key, similar to Ollama's local-only nature.

Confidence Score: 5/5

• Safe to merge - straightforward provider integration following established patterns
• The changes mirror the proven Ollama implementation from PR Fix ollama support with AI SDK #1504, use optional dependencies correctly, and require no breaking changes. All modifications are additive and follow the existing architectural patterns for AI SDK provider registration.
• No files require special attention

Important Files Changed

Filename	Overview
packages/core/lib/v3/llm/LLMProvider.ts	Added bedrock provider imports and registered it in AISDKProviders and AISDKProvidersWithAPIKey dictionaries
packages/core/lib/utils.ts	Added bedrock to providersWithoutApiKey set to skip API key requirement validation
packages/core/package.json	Added @ai-sdk/amazon-bedrock as optional dependency

Sequence Diagram

sequenceDiagram
    participant User
    participant Stagehand
    participant LLMProvider
    participant getAISDKLanguageModel
    participant BedrockSDK as @ai-sdk/amazon-bedrock

    User->>Stagehand: new Stagehand({ model: "bedrock/nova-pro" })
    Stagehand->>LLMProvider: getClient("bedrock/nova-pro")
    LLMProvider->>LLMProvider: Parse modelName ("bedrock/nova-pro")
    LLMProvider->>LLMProvider: Extract subProvider="bedrock", subModelName="nova-pro"
    LLMProvider->>getAISDKLanguageModel: getAISDKLanguageModel("bedrock", "nova-pro", clientOptions)
    
    alt clientOptions has valid values
        getAISDKLanguageModel->>getAISDKLanguageModel: Use AISDKProvidersWithAPIKey["bedrock"]
        getAISDKLanguageModel->>BedrockSDK: createAmazonBedrock(clientOptions)
        BedrockSDK-->>getAISDKLanguageModel: customProvider
        getAISDKLanguageModel->>getAISDKLanguageModel: customProvider("nova-pro")
    else no clientOptions or all undefined/null
        getAISDKLanguageModel->>getAISDKLanguageModel: Use AISDKProviders["bedrock"]
        getAISDKLanguageModel->>BedrockSDK: bedrock("nova-pro")
    end
    
    getAISDKLanguageModel-->>LLMProvider: languageModel
    LLMProvider->>LLMProvider: new AISdkClient({ model: languageModel })
    LLMProvider-->>Stagehand: LLMClient instance
    Stagehand-->>User: Initialized with Bedrock support

Loading

[greptile-apps]

_{No files reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}

[github-actions]

✱ Stainless preview builds

This PR will update the stagehand SDKs with the following commit message.

feat: add auto-bedrock support based on bedrock/provider.model-name

⚠️ stagehand-csharp studio · code

There was a regression in your SDK.
generate ⚠️ → lint ❗ → test ✅

✅ stagehand-go studio · code

Your SDK built successfully.
generate ✅ → lint ✅ → test ✅

go get github.com/stainless-sdks/stagehand-go@1bd5e501ba590c75662b3abd0065151ad707b108

⚡ stagehand-python studio · conflict

There was a conflict between your custom code and your generated changes.
You don't need to resolve this conflict right now, but you will need to resolve it for your changes to be released to your users. Read more about why this happened here.

✅ stagehand-openapi studio · code

Your SDK built successfully.
generate ✅ → lint ⏳ → test ⏳

⚠️ stagehand-typescript studio · code

There was a regression in your SDK.
generate ✅ → build ✅ → lint ✅ → test ❗

npm install https://pkg.stainless.com/s/stagehand-typescript/ac6b84fb7a76aff648b6873797071a6b36dd1f3b/dist.tar.gz

❗ stagehand-kotlin studio

Code was not generated because there was a fatal error.

⚡ stagehand-java studio · conflict

There was a conflict between your custom code and your generated changes.
You don't need to resolve this conflict right now, but you will need to resolve it for your changes to be released to your users. Read more about why this happened here.

✅ stagehand-php studio · code

Your SDK built successfully.
generate ✅ → lint ✅ → test ✅

✅ stagehand-ruby studio · code

Your SDK built successfully.
generate ✅ → lint ✅ → test ✅

⏳ These are partial results; builds are still running.

---

This comment is auto-generated by GitHub Actions and is automatically kept up to date as you push.
If you push custom code to the preview branch, re-run this workflow to update the comment.
Last updated: 2026-01-27 17:51:04 UTC