Databricks MCP Server

用于与Databricks交互的模型上下文协议（MCP）服务器。

安装

你可以从发布页面下载适用于你的平台的最新版本。

VS Code

通过点击以下链接在VS Code中安装Databricks MCP Server扩展：

<img src="https://img.shields.io/badge/VS_Code-VS_Code?style=flat-square&label=Install%20Server&color=0098FF" alt="在VS Code中安装">

或者，你可以通过运行以下命令手动安装该扩展：

# 对于VS Code
code --add-mcp '{"name":"databricks","command":"npx","args":["databricks-mcp-server@latest"]}'
# 对于VS Code Insiders
code-insiders --add-mcp '{"name":"databricks","command":"npx","args":["databricks-mcp-server@latest"]}'

工具

Databricks MCP Server提供了一个与Databricks工作区交互的模型上下文协议（MCP）接口。它提供了以下功能：

列出目录

列出Databricks工作区中的所有目录。

工具名称： list_catalogs

参数： 无

返回值： 目录对象的JSON数组

列出模式

列出指定Databricks目录中的所有模式。

工具名称： list_schemas

参数：

• catalog (字符串，必需)：要列出其模式的目录名称

返回值： 模式对象的JSON数组

列出表

列出指定Databricks模式中的所有表，并可选地进行过滤。

工具名称： list_tables

参数：

• catalog (字符串，必需)：包含模式的目录名称
• schema (字符串，必需)：要列出其表的模式名称
• filter_pattern (字符串，可选，默认：".*")：用于过滤表名的正则表达式模式

返回值： 表对象的JSON数组

执行SQL

在Databricks SQL仓库上执行SQL语句并返回结果。

工具名称： execute_sql

参数：

• statement (字符串，必需)：要执行的SQL语句
• timeout_seconds (数字，可选，默认：60)：语句执行的超时时间（秒）
• row_limit (数字，可选，默认：100)：结果中返回的最大行数

返回值： 包含查询结果列和行的JSON对象，以及用于执行该语句的SQL仓库的信息。

列出SQL仓库

列出Databricks工作区中的所有SQL仓库。

工具名称： list_warehouses

参数： 无

返回值： SQL仓库对象的JSON数组

支持的平台

• Linux (amd64)
• Windows (amd64)
• macOS (Intel/amd64)
• macOS (Apple Silicon/arm64)

使用方法

认证

应用程序使用Databricks统一认证。有关如何配置认证的详细信息，请参阅Databricks认证文档。

运行服务器

启动MCP服务器：

./databricks-mcp-server

服务器将启动并监听标准输入/输出上的MCP协议命令。

开发

先决条件

• Go 1.24或更高版本