Loading integration
Enterprise access
Agent Serve is available to teams by invitation. Tell us what you want to build and we will set up your workspace with you.
Agent Serve · Databases integration
Run SQL queries and manage jobs on Databricks
Databricks is a Databases integration for Agent Serve, the AI workspace where teams build and deploy AI agents. Agent Serve's Databricks integration provides 12 Databricks tools that AI agents can use inside Agent Serve's visual workflow builder. Databricks connects with an API key. Explore deployment options with the Agent Serve team.
Last updated
Connect to Databricks to execute SQL queries against SQL warehouses, trigger and monitor job runs, manage clusters, and retrieve run outputs. Requires a Personal Access Token and workspace host URL.
Agent Serve accounts are created by invitation. Request access for your team, or accept an invitation from your workspace admin, then sign in.
Open your workspace, drag a Databricks block onto the workflow builder, and paste in your Databricks API key.
Pick the tool you need, wire in an AI agent for reasoning or data transformation, and run. Your Databricks automation is live.
12 Databricks tools available in Agent Serve.
Execute a SQL statement against a Databricks SQL warehouse and return results inline. Supports parameterized queries and Unity Catalog.
Poll a SQL statement by its ID to retrieve status and results. Use this after Execute SQL when a query runs longer than the wait timeout.
List all SQL warehouses in a Databricks workspace including their size, state, and type. Use this to discover the warehouse ID needed for Execute SQL.
List all jobs in a Databricks workspace with optional filtering by name.
Get the full definition and settings of a single Databricks job by its job ID.
Trigger an existing Databricks job to run immediately with optional job-level or notebook parameters.
Get the status, timing, and details of a Databricks job run by its run ID.
List job runs in a Databricks workspace with optional filtering by job, status, and time range.
Cancel a running or pending Databricks job run. Cancellation is asynchronous; poll the run status to confirm termination.
Get the output of a completed Databricks job run, including notebook results, error messages, and logs. For multi-task jobs, use the task run ID (not the parent run ID).
List all clusters in a Databricks workspace including their state, configuration, and resource details.
Get the state, configuration, and resource details of a single Databricks cluster by its cluster ID.
Ready-to-use templates featuring Databricks. Click any to build it instantly.
Agent Serve's Databricks integration adds 12 Databricks tools to the AI agents you build in Agent Serve's visual workflow builder — you build it all visually. Run SQL queries and manage jobs on Databricks. Teams often pair Databricks with Affinity and incident.io in the same agent.
Agent Serve is available to teams by invitation. Request access and the team will map out the Databricks agents, connections, and team access your process needs.