Skip to main content
The Databricks toolkit lets agents browse Unity Catalog (catalogs, schemas, tables), run SQL on a warehouse, fetch statement results, list warehouses and workspace objects, export notebooks, inspect table lineage, and review query history. Authentication uses a saved connection with either a personal access token (PAT) or an OAuth M2M service principal (client ID + secret). Add it on the agent Toolkits tab and create or select a Databricks connection.

​
Prerequisites

  • A Databricks workspace and SQL warehouse the agent should use
  • Either a PAT, or a service principal client ID and secret, with access to the catalogs and warehouses you need
  • An agent you can edit

​
Setup

  1. Open the agent → Toolkits → add Databricks.
  2. Create a connection with:
    • Workspace host (required) — for example https://dbc-xxxx.cloud.databricks.com
    • SQL warehouse ID (required)
    • Either a personal access token, or client ID + client secret for a service principal
  3. Save the agent. @-mention Databricks tools in the personality prompt for catalog and SQL tasks.

​
Available tools

​
Unity Catalog & SQL

​
Workspace & API

​
Use cases

  • Discover catalogs and table schemas before writing a query
  • Run analytical SQL and summarize results in chat
  • Inspect lineage for a table before changing a pipeline
  • Export a notebook for review or troubleshooting

​
Tips

  • Access is limited to what the PAT or service principal can see (Unity Catalog grants and warehouse permissions).
  • Notebook access can differ from UC table grants — exporting a notebook needs workspace ACLs (CAN_READ or higher).
  • Prefer bounded SQL (LIMIT, filters) for interactive exploration.
  • Store tokens and secrets in a saved connection — do not put them in the personality prompt.

​
Troubleshooting