Databricks MCP 服务端

这是一个模型上下文协议（MCP）服务器，用于通过 Statement Execution API 在 Databricks 上执行 SQL 查询。它可以使用 Databricks API 执行 SQL 请求来检索数据。在代理模式下使用时，它能够成功地迭代多个请求以执行复杂任务。与 Unity Catalog 元数据配合使用效果更佳。

功能

• 执行 Databricks 上的 SQL 查询
• 列出目录中的可用架构
• 列出架构中的表
• 描述表结构

设置

系统要求

• Python 3.10+ 版本
• 如果计划通过 uv 安装，请确保已 安装

安装

1. 安装所需的依赖项：

pip install -r requirements.txt

或使用 uv：

uv pip install -r requirements.txt

2. 设置环境变量：

   选项 1：使用 .env 文件（推荐）

   创建一个包含 Databricks 凭据的 .env 文件：

   DATABRICKS_HOST=your-databricks-instance.cloud.databricks.com
   DATABRICKS_TOKEN=your-databricks-access-token
   DATABRICKS_SQL_WAREHOUSE_ID=your-sql-warehouse-id
   

   选项 2：通过命令行设置

   export DATABRICKS_HOST="your-databricks-instance.cloud.databricks.com"
   export DATABRICKS_TOKEN="your-databricks-access-token"
   export DATABRICKS_SQL_WAREHOUSE_ID="your-sql-warehouse-id"
   

权限要求

• SQL 仓库权限：确保用户具有执行查询的权限。
• Catalog 和 Schema 访问：根据组织的安全策略配置对特定目录、架构和表的访问。
• 最佳实践：
  • 定期轮换访问令牌。
  • 避免将敏感信息硬编码到代码中。

运行服务器

独立运行

python dbapi.py

与 Cursor 集成

在 Agent Composer 或其他 MCP 客户端中，可以使用以下工具：

execute_sql_query("SELECT * FROM my_schema.my_table LIMIT 10")
list_schemas("my_catalog")
list_tables("my_catalog.my_schema")
describe_table("my_catalog.my_schema.my_table")

处理长时间运行的查询

服务器设计用于处理长时间运行的查询，通过轮询 Databricks API 直到查询完成或超时。默认超时时间为 10 分钟（60 次重试，间隔为 10 秒），可以在 dbapi.py 文件中根据需要调整。

依赖项

• httpx：用于向 Databricks API 发送 HTTP 请求
• python-dotenv：用于从 .env 文件加载环境变量
• mcp：模型上下文协议库
• asyncio：用于异步操作