Skip to main content

Configure BYOK for dbt Wizard in dbt platform

Use bring-your-own-key (BYOK) to connect dbt Wizard or dbt Copilot in dbt platform to your own AI provider account instead of using dbt Labs' managed infrastructure.

The following BYOK instructions apply to dbt platform only. For CLI BYOK setup, refer to Configure BYOK for dbt Wizard.

When you configure a provider with your own key, usage costs appear on your provider account instead of your dbt Labs account, and token costs are billed by whichever provider you choose.

Prerequisites

  • A dbt platform account on Starter, Enterprise, or Enterprise+ plans
  • dbt admin permissions to enable AI features and configure providers in Account settings
  • AI features enabled for your account — refer to Enable AI in dbt platform
  • An API key or credentials for your supported AI provider
 See the full list of supported AI providers

Supported AI providers

dbt Wizard

dbt Wizard supports different AI providers depending on where you use it.

Providerdbt Wizard in dbt platformdbt Wizard CLI
OpenAI✓ (managed or BYOK)✓ (OpenAI subscription or BYOK)
Anthropic✓ (BYOK)✓ (BYOK)
Azure AI Foundry / Azure OpenAI✓ (BYOK)✓ (BYOK)
AWS Bedrock-✓ (BYOK)
Google Gemini-✓ (BYOK)
Snowflake Cortex-✓ (BYOK)
Databricks Unity AI Gateway-✓ (BYOK)

Anthropic enterprise and subscription licenses (such as Claude Enterprise) aren't supported per Anthropic's terms of service. BYOK requires an Anthropic API key.

Refer to the following pages for more information:

dbt Copilot

dbt Copilot supports different AI providers, including bring your own key (BYOK) for Enterprise and Enterprise+ plans:

  • dbt Labs-managed OpenAI API key
  • BYOK OpenAI API key
  • BYOK Azure OpenAI API key

Snowflake Cortex, AWS Bedrock, Azure AI Foundry, and Anthropic aren't supported for dbt Copilot.

Configure AI provider

Once AI features have been enabled, Enterprise and Enterprise+ accounts can configure a custom AI provider. If you bring your own provider, you will incur API calls and associated charges from that provider.

* Managed (or Managed by dbt Labs): dbt Labs manages the AI provider connection; no user provider key is required. Refer to Billing for more information.

dbt Wizard

To configure your AI provider for dbt Wizard:

  1. Click on your account name and select Account settings in the side menu.
  2. Under Settings, click AI features.
  3. Under AI providers, click Edit to configure the AI integration.
  4. For each provider, select your Key management option from the dropdown.

Managed by dbt Labs (default, no setup required). Refer to Billing for more information.*

  1. Select the toggle for dbt Labs to use dbt Labs' managed* OpenAI key.
  2. Click Save.
Example of the dbt Labs integration pageExample of the dbt Labs integration page

Managed by you (Enterprise or Enterprise+ plans)

  1. Select OpenAI from the options.
  2. Enter your OpenAI API key.
  3. Click Save.
Example of the OpenAI integration pageExample of the OpenAI integration page
Data residency limitation

OpenAI projects with data residency controls enabled and configured for the United States (project region set to US) don't currently support BYOK. These projects can only use the API key in the dbt platform configuration. Specifying custom endpoints required for data residency isn’t yet supported, and we’re evaluating a solution for this.

To use BYOK, ensure your OpenAI project doesn’t have data residency controls enabled. Projects without project region settings will use the standard OpenAI endpoint (https://api.openai.com) and support BYOK.

  • For BYOK, enable the latest text generation models as well as the text-embedding-3-small model.
  • Ensure your project doesn't have data residency controls enabled. Projects without project region settings use the standard OpenAI endpoint (https://api.openai.com) and support BYOK.

dbt Copilot

To configure your AI provider for dbt Copilot:

  1. Click on your account name and select Account settings in the side menu.
  2. Under Settings, click Copilot.
  3. Under AI providers, click Edit to configure the AI integration.
  4. For each provider, select your Key management option from the dropdown.
  1. Select the toggle for dbt Labs to use dbt Labs' managed* OpenAI key.
  2. Click Save.
Example of the dbt Labs integration pageExample of the dbt Labs integration page

dbt Wizard is available in Studio IDE as a public preview feature, and as a standalone beta feature across the managed dbt platform. However, certain customers may have disabled experimental features, in which case, they can use Wizard CLI via terminal-access and will continue to have access to dbt Copilot until dbt Wizard is released as generally available to all customers. Contact dbt Support with any questions.

Please contact dbt Support with any questions

Was this page helpful?

This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.

0
Loading