Databricks MCP Server

🚀 Databricks MCP 服务器

Databricks MCP 服务器是用于与 Databricks 进行交互的模型上下文协议 (MCP) 服务器，能助力用户高效操作 Databricks 工作区。

🚀 快速开始

要使用 Databricks MCP 服务器，请按照以下步骤操作：

1. 下载并安装适合你系统的版本。
2. 启动服务器。
3. 使用提供的工具与 Databricks 工作区交互。

📦 安装指南

你可以从 Releases 页面下载适合你平台的最新版本。

VS Code

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

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

# 对于 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 服务器提供了与 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 (数字，可选)：返回的最大行数
• 返回：执行结果的 JSON 对象

支持平台

• Linux（amd64）
• Windows（amd64）
• macOS（amd64）

💻 使用示例

基础用法

列出目录

# 示例命令
curl -X POST http://localhost:8080/mcp/list_catalogs

列出模式

# 示例命令
curl -X POST http://localhost:8080/mcp/list_schemas?catalog=my_catalog

列出表

# 示例命令
curl -X POST http://localhost:8080/mcp/list_tables?catalog=my_catalog&schema=my_schema

⚠️ 注意事项

⚠️ 重要提示

• 确保服务器运行在安全的环境中。
• 配置适当的防火墙规则以限制访问。

通过这些工具，你可以方便地与 Databricks 工作区交互并执行各种操作。