Genie MCP Server

Genie MCP サーバー

このプロジェクトは、MCP StreamableHTTP Protocol を介して Databricks Genie とやり取りするための非同期サーバーを提供します。Databricks 認証と堅牢なエラー処理を活用し、ユーザーが Genie にクエリを送信して回答やデータを受け取れるようにします。

特長

• 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 space に合わせて更新します。

   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.