AiGatewayModelProviderService
laktory.models.resources.databricks.AiGatewayModelProviderService
¤
Bases: AiGatewayModelProviderServiceBase
Databricks AI Gateway model provider service
Unity Catalog governed connection to an external model provider (OpenAI, Azure OpenAI, Anthropic, Amazon Bedrock, Gemini, custom, etc.) and the provider models that model services can route to.
Examples:
import io
from laktory import models
service_yaml = '''
parent: schemas/dev.ai
model_provider_service_id: openai_prod
comment: OpenAI production account
config:
provider_type: EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI
targets:
- model: gpt-5
native_api_types:
- openai/v1/chat/completions
openai:
direct:
api_key:
secret_reference:
name: secrets/dev.ai.openai_api_key
'''
service = (
models.resources.databricks.AiGatewayModelProviderService.model_validate_yaml(
io.StringIO(service_yaml)
)
)
| BASE | DESCRIPTION |
|---|---|
comment
|
User-provided description
TYPE:
|
config
|
Provider authentication, exposed models, request-forwarding controls, rate limits, and payload logging. Required on Create. On Update, it is required only when
TYPE:
|
model_provider_service_id
|
Name for the model provider service, e.g. 'openai_prod'
TYPE:
|
parent
|
Parent Unity Catalog schema where the inference table is created, in the form
TYPE:
|
provider_config
|
Configure the provider for management through account provider
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfig
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
allow_all_targets
|
When true, accepts any model exposed by the upstream provider;
TYPE:
|
amazon_bedrock
|
TYPE:
|
anthropic
|
TYPE:
|
azure_openai
|
TYPE:
|
custom
|
TYPE:
|
forward_headers
|
Whether to forward incoming HTTP headers to the upstream provider. Defaults to false and is configured for the entire provider service, not per request. Upstream authentication is configured separately in the provider-specific configuration
TYPE:
|
forward_query_parameters
|
Whether to forward incoming query parameters to the upstream provider. Defaults to false and is configured for the entire provider service, not per request
TYPE:
|
forward_unmanaged_paths
|
Whether to proxy paths that AI Gateway does not recognize as configured provider-native API types. Defaults to false. When true, these paths are forwarded unchanged to the upstream provider. When false, only recognized API paths are served. Enabling this broadens the upstream API surface exposed through the provider service *
TYPE:
|
gemini_enterprise
|
TYPE:
|
inference_table
|
Payload logging configuration for requests sent directly to this provider service. Requests routed through a model service are captured by that model service's inference table instead *
TYPE:
|
microsoft_foundry
|
TYPE:
|
openai
|
TYPE:
|
provider_type
|
External model provider. Required on Create and immutable thereafter. Set the matching provider-specific configuration, such as
TYPE:
|
rate_limits
|
Rate limits for requests sent directly to this provider service. Requests routed through a model service use that model service's rate limits instead
TYPE:
|
targets
|
Models and provider-native API types exposed by this provider service. Each entry must include at least one
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAmazonBedrock
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
direct
|
OpenAI configuration with an API key supplied in the request
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAmazonBedrockDirect
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
aws_access_key
|
AWS access-key-pair authentication. Set
TYPE:
|
region
|
GCP region of the Gemini Enterprise endpoint (e.g.,
TYPE:
|
service_credential
|
Reference to a Unity Catalog service credential authorizing Microsoft Foundry requests. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAmazonBedrockDirectAwsAccessKey
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
access_key_id
|
AWS access key ID. Required on Create when using access-key auth. Treated as username-equivalent (not a secret value): round-trips on reads and is scrubbed from audit logs
TYPE:
|
secret_access_key
|
AWS secret access key paired with
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAmazonBedrockDirectAwsAccessKeySecretAccessKey
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAmazonBedrockDirectAwsAccessKeySecretAccessKeySecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAmazonBedrockDirectServiceCredential
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAnthropic
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
direct
|
OpenAI configuration with an API key supplied in the request
TYPE:
|
relayed
|
Relayed authentication. Each inference request supplies the caller's OAuth token, which is forwarded to Anthropic. No Anthropic credential is stored. Mutually exclusive with
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAnthropicDirect
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
api_key
|
OpenAI API key. Required when creating the service. Supply the value in
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAnthropicDirectApiKey
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAnthropicDirectApiKeySecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAnthropicRelayed
¤
Bases: BaseModel
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAzureOpenai
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
direct
|
OpenAI configuration with an API key supplied in the request
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAzureOpenaiDirect
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
api_key
|
OpenAI API key. Required when creating the service. Supply the value in
TYPE:
|
base_url
|
Optional custom base URL. Defaults to
TYPE:
|
entra_service_principal
|
Entra ID service-principal authentication. Set
TYPE:
|
service_credential
|
Reference to a Unity Catalog service credential authorizing Microsoft Foundry requests. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAzureOpenaiDirectApiKey
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAzureOpenaiDirectApiKeySecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAzureOpenaiDirectEntraServicePrincipal
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
client_id
|
Entra ID client (application) ID. Required on Create
TYPE:
|
client_secret
|
Entra ID client secret. Supply the value in
TYPE:
|
tenant_id
|
Entra ID (Azure AD) tenant ID. Required on Create
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAzureOpenaiDirectEntraServicePrincipalClientSecret
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAzureOpenaiDirectEntraServicePrincipalClientSecretSecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigAzureOpenaiDirectServiceCredential
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigCustom
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
direct
|
OpenAI configuration with an API key supplied in the request
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigCustomDirect
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
api_key
|
OpenAI API key. Required when creating the service. Supply the value in
TYPE:
|
base_url
|
Optional custom base URL. Defaults to
TYPE:
|
header_auth
|
Header-based API-key auth: the secret is forwarded on outbound requests under a caller-chosen HTTP header rather than as an
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigCustomDirectApiKey
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigCustomDirectApiKeySecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigCustomDirectHeaderAuth
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
api_key_name
|
HTTP header name that carries the API key on outbound requests (e.g.,
TYPE:
|
api_key_value
|
Secret value forwarded under the
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigCustomDirectHeaderAuthApiKeyValue
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigCustomDirectHeaderAuthApiKeyValueSecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigGeminiEnterprise
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
direct
|
OpenAI configuration with an API key supplied in the request
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigGeminiEnterpriseDirect
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
api_key
|
OpenAI API key. Required when creating the service. Supply the value in
TYPE:
|
project_id
|
GCP project ID hosting the Gemini Enterprise endpoint. Required on Create
TYPE:
|
region
|
GCP region of the Gemini Enterprise endpoint (e.g.,
TYPE:
|
service_credential
|
Reference to a Unity Catalog service credential authorizing Microsoft Foundry requests. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigGeminiEnterpriseDirectApiKey
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigGeminiEnterpriseDirectApiKeySecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigGeminiEnterpriseDirectServiceCredential
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigInferenceTable
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
parent
|
Parent Unity Catalog schema where the inference table is created, in the form
TYPE:
|
table_name_prefix
|
Prefix used to form the inference table's registered name. AI Gateway appends
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigMicrosoftFoundry
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
direct
|
OpenAI configuration with an API key supplied in the request
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigMicrosoftFoundryDirect
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
api_key
|
OpenAI API key. Required when creating the service. Supply the value in
TYPE:
|
base_url
|
Optional custom base URL. Defaults to
TYPE:
|
entra_service_principal
|
Entra ID service-principal authentication. Set
TYPE:
|
service_credential
|
Reference to a Unity Catalog service credential authorizing Microsoft Foundry requests. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigMicrosoftFoundryDirectApiKey
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigMicrosoftFoundryDirectApiKeySecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigMicrosoftFoundryDirectEntraServicePrincipal
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
client_id
|
Entra ID client (application) ID. Required on Create
TYPE:
|
client_secret
|
Entra ID client secret. Supply the value in
TYPE:
|
tenant_id
|
Entra ID (Azure AD) tenant ID. Required on Create
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigMicrosoftFoundryDirectEntraServicePrincipalClientSecret
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigMicrosoftFoundryDirectEntraServicePrincipalClientSecretSecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigMicrosoftFoundryDirectServiceCredential
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigOpenai
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
direct
|
OpenAI configuration with an API key supplied in the request
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigOpenaiDirect
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
api_key
|
OpenAI API key. Required when creating the service. Supply the value in
TYPE:
|
base_url
|
Optional custom base URL. Defaults to
TYPE:
|
organization
|
Optional OpenAI organization ID. When set, the platform forwards it as the
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigOpenaiDirectApiKey
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
plaintext
|
Inline plaintext credential. INPUT_ONLY: the value never round-trips on reads. Get and List responses omit
TYPE:
|
secret_reference
|
Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service owner's access and is never copied onto the model provider service, so rotating the UC Secret takes effect with no change to the model provider service. On Create, supply
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigOpenaiDirectApiKeySecretReference
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
name
|
Resource name of the provider service. Format:
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigRateLimits
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
key
|
Scope of the rate limit. Depending on this value, the limit applies to a principal, the service as a whole, or each user by default. Possible values are:
TYPE:
|
principal
|
Principal this limit applies to: user email, group name, or service principal application ID. Required when
TYPE:
|
renewal_period
|
Renewal period. Possible values are:
TYPE:
|
requests
|
Maximum requests allowed in one renewal period. Leave unset for no request limit. Set to
TYPE:
|
tokens
|
Maximum tokens allowed in one renewal period. Leave unset for no token limit. Set to
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceConfigTargets
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
model
|
Provider-side model identifier, such as
TYPE:
|
native_api_types
|
Provider-native API types supported by this model, such as
TYPE:
|
laktory.models.resources.databricks.aigatewaymodelproviderservice.AiGatewayModelProviderServiceProviderConfig
¤
Bases: BaseModel
| PARAMETER | DESCRIPTION |
|---|---|
workspace_id
|
Workspace ID which the resource belongs to. This workspace must be part of the account which the provider is configured with
TYPE:
|