Skip to content

AiGatewayMcpService

laktory.models.resources.databricks.AiGatewayMcpService ¤

Bases: AiGatewayMcpServiceBase

Databricks AI Gateway MCP service

Unity Catalog governed access to an MCP server hosted through a Unity Catalog connection, with tool selection and rate limits.

Examples:

import io

from laktory import models

service_yaml = '''
parent: schemas/dev.ai
mcp_service_id: knowledge_tools
comment: Governed access to knowledge tools
config:
  source_connection:
    name: connections/dev.ai.mcp_connection
'''
service = models.resources.databricks.AiGatewayMcpService.model_validate_yaml(
    io.StringIO(service_yaml)
)
References
BASE DESCRIPTION
comment

User-provided description

TYPE: str | None | VariableType DEFAULT: None

config

Connection, tool selectors, and rate limits. Required on Create. On Update, provide this field when update_mask contains config or one of its subpaths

TYPE: AiGatewayMcpServiceConfig | None | VariableType DEFAULT: None

mcp_service_id

Name for the MCP service, e.g. 'my_mcp_service'

TYPE: str | VariableType

parent

Name of the parent schema. Format: schemas/{catalog}.{schema}. Each {...} component is capped at 255 characters individually

TYPE: str | VariableType

provider_config

Configure the provider for management through account provider

TYPE: AiGatewayMcpServiceProviderConfig | None | VariableType DEFAULT: None


laktory.models.resources.databricks.aigatewaymcpservice.AiGatewayMcpServiceConfig ¤

Bases: BaseModel

PARAMETER DESCRIPTION
include_tool_selectors

Tool names or prefix patterns to expose from the MCP server. Use exact tool names or prefix patterns such as read_*. An empty list exposes all tools. At most 1,024 selectors are allowed, and each selector can contain at most 256 characters

TYPE: list[str] | None | VariableType DEFAULT: None

rate_limits

Rate limits for tool invocations. Supported scopes are user, group, service principal, the service as a whole, and each user by default. Request and token limits are supported. Empty when no rate limit is configured

TYPE: list[AiGatewayMcpServiceConfigRateLimits] | None | VariableType DEFAULT: None

source_connection

Unity Catalog connection referencing the MCP server. Required on Create

TYPE: AiGatewayMcpServiceConfigSourceConnection | None | VariableType DEFAULT: None


laktory.models.resources.databricks.aigatewaymcpservice.AiGatewayMcpServiceConfigRateLimits ¤

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: RATE_LIMIT_KEY_SERVICE, RATE_LIMIT_KEY_SERVICE_PRINCIPAL, RATE_LIMIT_KEY_USER, RATE_LIMIT_KEY_USER_DEFAULT, RATE_LIMIT_KEY_USER_GROUP

TYPE: str | VariableType

principal

Principal this limit applies to: user email, group name, or service principal application ID. Required when key applies to a user, group, or service principal; otherwise it must be unset

TYPE: str | None | VariableType DEFAULT: None

renewal_period

Renewal period. Possible values are: RATE_LIMIT_RENEWAL_PERIOD_HOUR, RATE_LIMIT_RENEWAL_PERIOD_MINUTE

TYPE: str | VariableType

requests

Maximum requests allowed in one renewal period. Leave unset for no request limit. Set to 0 to deny all requests

TYPE: int | None | VariableType DEFAULT: None

tokens

Maximum tokens allowed in one renewal period. Leave unset for no token limit. Set to 0 to deny all requests

TYPE: int | None | VariableType DEFAULT: None


laktory.models.resources.databricks.aigatewaymcpservice.AiGatewayMcpServiceConfigSourceConnection ¤

Bases: BaseModel

PARAMETER DESCRIPTION
name

Resource name of the MCP service. Format: mcp-services/{catalog}.{schema}.{mcp_service}. Each {...} component is capped at 255 characters individually. Server-derived on Create from parent + mcp_service_id; required and immutable on Update/Get/Delete

TYPE: str | VariableType


laktory.models.resources.databricks.aigatewaymcpservice.AiGatewayMcpServiceProviderConfig ¤

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: str | None | VariableType DEFAULT: None