Databricks MCP 服务器

Databricks MCP 服务器

一个用于通过 Databricks API 执行 SQL 查询的 Model Context Protocol (MCP) 服务器。

快速入门

系统要求

  • Python 3.10+

安装

  1. 克隆仓库:git clone https://github.com/RafaelCartenet/mcp-databricks-server.git
  2. 安装依赖:
    • 使用 pip:pip install -r requirements.txt
    • 或使用 uv:uv pip install -r requirements.txt

配置

创建 .env 文件并添加以下内容:

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

启动服务

运行以下命令启动服务器:

python main.py

示例工具调用

  • 执行 SQL 查询:execute_sql_query("SELECT * FROM my_schema.my_table LIMIT 10")
  • 列出模式:list_schemas("my_catalog")
  • 列出表:list_tables("my_schema")
  • 描述表结构:describe_table("my_table")