Databricks管理服务器

Databricks MCP Server是一个提供与Databricks交互工具的服务,支持集群、作业、笔记本、DBFS和SQL管理。

  • by AndresGarciaSobrado91
  • Python
  • 本地部署
  • 更新于
在 GitHub 查看 andresgarciasobrado91/databricks-mcp-server

使用说明

_CHMCP服务器

这是一个服务器,它提供了与ViewModel交互的工具。

特性

  • 集群管理
  • 作业管理
  • 笔记本管理
  • DBFS管理
  • SQL管理

安装

所需Python版本:3.11

# Create and activate virtual environment
uv venv

# On Windows
.\.venv\Scripts\activate

# On Linux/Mac
source .venv/bin/activate

# Install dependencies
uv pip install pip
pip install -r requirements.txt

配置

可以使用环境变量或.env文件配置服务器。

环境变量

  • DATABRICKS_HOST:Rancher主机URL(例如。https://adb-123456789012345.12.azuredatabricks.net)
  • DATABRICKS_TOKEN:Databricks API令牌
  • TRANSPORT:使用的交通工具(如sse)。选项:[stdio、sse、可流式传输http]。默认值:sse
  • SERVER_HOST:将服务器绑定到的主机(例如0.0.0.0)
  • SERVER_PORT:将服务器绑定到的端口(例如8000)
  • DEBUG:是否在调试模式下运行服务器(例如True)
  • LOG_LEVEL:日志级别(例如INFO)

.env文件

在项目的根目录中创建一个.env文件,其中包含以下变量:

DATABRICKS_HOST=https://adb-123456789012345.12.azuredatabricks.net
DATABRICKS_TOKEN=your_databricks_api_token
TRANSPORT=sse
SERVER_HOST=0.0.0.0
SERVER_PORT=8000
DEBUG=True
LOG_LEVEL=INFO

获取copula凭据

  1. 主机:您的copula实例URL(例如,您的instance.cloud.Databricks.com)
  2. Token:在ViewModel中创建个人访问令牌:
    • 转到用户设置(单击右上角的用户名)
    • 选择“开发人员”选项卡
    • 点击“访问令牌”下的“管理”
    • 生成一个新的令牌,并保存它(它将不再显示)

以独立模式运行服务器

python -m main

配置克劳德桌面/光标/Windsurf

将以下内容添加到您的Claude Desktop/Cursor配置文件中:

{
  "mcpServers": {
    "databricks": {
      "command": "uv",
      "args": [
        "--directory",
        "/absolute/path/to/this/project",
        "run",
        "python",
        "-m",
        "main"
      ],
      "env": {
        "DATABRICKS_HOST": "https://your-host.cloud.databricks.com",
        "DATABRICKS_TOKEN": "your-databricks-api-token",
        "TRANSPORT": "stdio"
      }
    }
  }
}

如果您正在使用Claude Desktop,请重新启动它以使更改生效。