AI governance

AI governance extends the data governance capabilities of Unity Catalog to AI resources, applying the same access control, lineage, and audit model that protects your data assets to your AI assets and AI traffic.

AI asset governance with Unity Catalog

Unity Catalog manages AI assets as securable objects. You can grant and revoke access to the following AI assets using standard Unity Catalog privileges:

• Models: Registered ML models in Unity Catalog. See Manage model lifecycle.
• Functions: Unity Catalog functions used as agent tools or for data transformations. See Create AI agent tools using Unity Catalog functions.
• Connections: Unity Catalog HTTP connections used to access external APIs and MCP servers. See HTTP connections.

AI traffic governance with AI Gateway

Beta

This feature is in Beta. Account admins can control access to this feature from the account console Previews page. See Manage Databricks previews.

AI Gateway is the enterprise control plane for governing AI traffic across your organization. Use AI Gateway to manage and monitor LLM endpoints and MCP servers from a central location:

• LLMs: Control access to hosted and external LLM endpoints, enforce rate limits, and track usage and costs across providers.
• MCPs: Manage access to managed, external, and custom MCP servers alongside your LLM endpoints.

See AI Gateway.