databricks-solutions/ai-dev-kit

Databricks Toolkit for Coding Agents provided by Field Engineering

What it solves

它弥合了一般用途 AI 编码助手（如 Cursor、Claude Code 或 GitHub Copilot）与 Databricks 平台特有复杂性之间的差距。它提供这些助手在构建 Spark 流水线、管理 Unity Catalog 治理、部署 ML 模型时所需的上下文、模式和可执行工具，让用户不必每次都手动解释 Databricks 特定的 API 和最佳实践。

How it works

该套件由四个可组合的组件组成，可单独或一起使用：

  • MCP Server：实现 Model Context Protocol，向 AI 助手公开超过 50 个可执行工具，允许它们直接在 Databricks 上执行操作。
  • Skills：一系列基于 markdown 的指南，教会 AI 助手特定的 Databricks 模式和最佳实践。
  • Core Library：Python 库 (databricks-tools-core) 提供执行 SQL 等常见任务的高级函数。
  • Builder App：具备聊天 UI 的全栈网页应用，将这些工具整合为可视化开发体验。

Who it’s for

已经使用自己偏好的 AI 驱动 IDE 或 CLI 工具（如 Cursor、Windsurf 或 Gemini CLI）的开发者和数据工程师，想要在这些工作流中集成深度的 Databricks 专业知识和自动化。

Highlights

  • Broad Tool Support：兼容多种 AI 助手，包括 Claude Code、Cursor、Copilot 和 Gemini CLI。
  • Extensive Toolset：提供 75+ 个可执行工具，可构建从 AI/BI 仪表盘到基于 RAG 的知识助理等各种应用。
  • Flexible Deployment：可以作为项目级配置、全局设置，或部署为独立的网页应用。
  • Genie Code Integration：包含将 Skills 直接安装到 Databricks 工作区供 Genie Code 使用的特定工作流。