Databricks AI Dev Kit
🔒 Proactive Dependency Security
As part of our commitment to supply chain integrity, we continually monitor our dependency tree against known vulnerabilities and industry advisories. In response to a recently disclosed supply chain incident affecting litellm versions 1.82.7–1.82.8, we have audited our packages and removed the litellm dependency for most usage. It is solely used in the test directory for skills evaluation and optimization, and has been pinned to a safe version.
For full third-party attribution, see NOTICE.txt.
Overview
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, Antigravity, Windsurf, etc.) the trusted sources it needs to build faster and smarter on Databricks.
What Can I Build?
- 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
Choose Your Own Adventure
| Adventure | Best For | Start Here |
|---|---|---|
| :star: 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 skills into your workspace for Genie Code (--install-to-genie) | Genie Code skills (install) |
| MCP Tools Only | Just executable actions (no guidance) | Register MCP server |
Quick Start
Prerequisites
- uv - Python package manager
- Databricks CLI - Command line interface for Databricks
- AI coding environment (one or more):
Install in existing project
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, .gemini, or .agents
Mac / Linux
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 --force
Specify profile and force reinstall
bash <(curl -sL https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.sh) --profile DEFAULT --force
Install for specific tools only
bash <(curl -sL https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.sh) --tools cursor,gemini,antigravity
Next steps: Respond to interactive prompts and follow the on-screen instructions.
- Note: Cursor and Copilot require updating settings manually after install.
Windows (PowerShell)
Basic installation (uses DEFAULT profile, project scope)
irm https://raw.githubusercontent.com/databricks-solutions/ai-dev-kit/main/install.ps1 | iex
Advanced Options (click to expand)
Download script first