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 pipeline、管理 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 使用的特定工作流程。