Databricks MCP 服务器

Databricks MCP 服务器

一个用于与 Databricks 服务交互的模型完成协议 (MCP) 服务器。

快速入门

  1. 安装 uv 包管理器:

    • MacOS/Linux: curl -LsSf https://astral.sh/uv/install.sh | sh
    • Windows: 在 PowerShell 中运行 irm https://astral.sh/uv/install.ps1 | iex
  2. 克隆仓库并进入目录:

    git clone https://github.com/JustTryAI/databricks-mcp-server.git
    cd databricks-mcp-server
    
  3. 设置虚拟环境并安装依赖:

    uv venv
    source .venv/bin/activate  # Linux/Mac
    .\.venv\Scripts\activate  # Windows
    uv pip install -e .
    uv pip install -e ".[dev]"
    
  4. 配置环境变量:

    export DATABRICKS_HOST=https://your-databricks-instance.azuredatabricks.net
    export DATABRICKS_TOKEN=your-personal-access-token
    

    或使用 .env 文件。

  5. 启动服务器:

    ./start_mcp_server.sh  # Linux/Mac
    .\start_mcp_server.ps1  # Windows
    
  6. 测试功能:

    uv run scripts/show_clusters.py
    uv run scripts/show_notebooks.py