Genie MCP Server

Genie MCP 服务

此项目提供一个异步服务，通过 MCP StreamableHTTP Protocol 与 Databricks Genie 交互。它支持用户向 Genie 提问并获取答案或数据，同时利用 Databricks 身份验证并进行稳健的错误处理。

功能

• 使用 Starlette 和 Uvicorn 构建的异步 API 服务
• 集成 Databricks Genie，支持对话式查询
• 自动管理 Databricks OAuth 令牌

项目结构

• mcp_server.py：服务的主入口，通过 MCP StreamableHTTP Protocol 暴露工具。
• genie_room.py：处理 Genie API 交互和对话逻辑。
• token_minter.py：负责生成和刷新 Databricks OAuth 令牌。
• requirements.txt：Python 依赖项。
• app.yaml：部署配置示例。

部署到 Databricks Apps

按照以下步骤，将 Genie MCP Server 部署为 Databricks 应用：

1. 将代码仓库克隆到工作区

   在 Databricks 工作区中，导航到要部署应用的目录（例如，/Workspace/Users/[email protected]/genie_mcp_server）。然后克隆代码仓库：

   git clone https://github.com/your-org/genie_mcp_server.git
   

2. 配置 Genie Space ID 和其他环境变量

   打开克隆代码仓库根目录中的 app.yaml 文件。将 SPACE_ID 的值更新为与你的 Genie 空间匹配的值。

   app.yaml 示例：

   command:
   - "python"
   - "mcp_server.py"
   
   env:
   - name: "SPACE_ID"
     value: "your_space_id"
   

3. 在 Databricks 中创建并部署应用

   • 前往 Databricks Apps 界面。
   • 创建一个新应用，并指定克隆代码仓库的目录路径。
   • 完成应用创建和部署流程。 databricks apps deploy genie-mcp-server --source-code-path /Workspace/Users/[email protected]/genie_mcp_server

4. Access the API

   Once deployed, your Genie MCP Server app will be running and accessible via the configured port and endpoint (e.g., https://app-url.aws.databricksapps.com/api/mcp/). You can now send MCP requests to your deployed app from within your Databricks workspace or from external clients, depending on your network configuration.