AI-Driven Development (vibe coding) on Databricks just got a whole lot better. The AI Dev Kit gives your AI coding assistant (Claude Code, Cursor, Windsurf, etc.) the trusted sources it needs to build faster and smarter on Databricks.
- Spark Declarative Pipelines (streaming tables, CDC, SCD Type 2, Auto Loader)
- Databricks Jobs (scheduled workflows, multi-task DAGs)
- AI/BI Dashboards (visualizations, KPIs, analytics)
- Unity Catalog (tables, volumes, governance)
- Genie Spaces (natural language data exploration)
- Knowledge Assistants (RAG-based document Q&A)
- MLflow Experiments (evaluation, scoring, traces)
- Model Serving (deploy ML models and AI agents to endpoints)
- Databricks Apps (full-stack web applications with foundation model integration)
- ...and more
| Adventure | Best For | Start Here |
|---|---|---|
| ⭐ Install AI Dev Kit | Start here! Follow quick install instructions to add to your existing project folder | Quick Start (install) |
| Visual Builder App | Web-based UI for Databricks development | databricks-builder-app/ |
| Core Library | Building custom integrations (LangChain, OpenAI, etc.) | pip install |
| Skills Only | Provide Databricks patterns and best practices (without MCP functions) | Install skills |
| Genie Code Skills | Install Databricks skills for Genie Code to reference | Genie Code skills (install) |
| MCP Tools Only | Just executable actions (no guidance) | Register MCP server |
- uv - Python package manager
- Databricks CLI - Command line interface for Databricks
- AI coding environment (one or more):
By default this will install at a project level rather than a user level. This is often a good fit, but requires you to run your client from the exact directory that was used for the install. Note: Project configuration files can be re-used in other projects. You find these configs under .claude, .cursor, or .gemini
Basic installation (uses DEFAULT profile, project scope)
bash <(curl -sL https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.sh)Advanced Options (click to expand)
Global installation with force reinstall
bash <(curl -sL https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.sh) --global --forceSpecify profile and force reinstall
bash <(curl -sL https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.sh) --profile DEFAULT --forceInstall for specific tools only
bash <(curl -sL https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.sh) --tools cursor,geminiNext steps: Respond to interactive prompts and follow the on-screen instructions.
- Note: Cursor and Copilot require updating settings manually after install.
Basic installation (uses DEFAULT profile, project scope)
irm https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.ps1 | iexAdvanced Options (click to expand)
Download script first
irm https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.ps1 -OutFile install.ps1Global installation with force reinstall
.\install.ps1 -Global -ForceSpecify profile and force reinstall
.\install.ps1 -Profile DEFAULT -ForceInstall for specific tools only
.\install.ps1 -Tools cursor,geminiNext steps: Respond to interactive prompts and follow the on-screen instructions.
- Note: Cursor and Copilot require updating settings manually after install.
Full-stack web application with chat UI for Databricks development:
cd ai-dev-kit/databricks-builder-app
./scripts/setup.sh
# Follow instructions to start the appUse databricks-tools-core directly in your Python projects:
from databricks_tools_core.sql import execute_sql
results = execute_sql("SELECT * FROM my_catalog.schema.table LIMIT 10")Works with LangChain, OpenAI Agents SDK, or any Python framework. See databricks-tools-core/ for details.
Will install and deploy all available skills to your personal skills directory for all Genie Code sessions to reference while planning/building anything directly in the UI. No post-install steps as workspace is automatically configured during install process for Genie Code to use the skills.
Basic installation (uses DEFAULT profile)
cd ai-dev-kit/databricks-skills
./install_skills_to_genie_code.shAdvance installation (uses provided profile)
cd ai-dev-kit/databricks-skills
./install_skills_to_genie_code <profile_name>Skill modification or Custom Skill
After the script successfully installs the skills to your workspace, you may find the skills under /Workspace/Users/<your_user_name>/.assistant/skills.
This directory is customizable if you wish to only use certain skills or even create custom skills that are related to your organization to make Genie Code even better. You can modify/remove existing skills or create new skills folders that Genie Code will automatically use in any session.
| Component | Description |
|---|---|
databricks-tools-core/ |
Python library with high-level Databricks functions |
databricks-mcp-server/ |
MCP server exposing 50+ tools for AI assistants |
databricks-skills/ |
20 markdown skills teaching Databricks patterns |
databricks-builder-app/ |
Full-stack web app with Claude Code integration |
(c) 2026 Databricks, Inc. All rights reserved.
The source in this project is provided subject to the Databricks License. See LICENSE.md for details.
Third-Party Licenses
Acknowledgments
MCP Databricks Command Execution API from databricks-exec-code by Natyra Bajraktari and Henryk Borzymowski.