Databricks MCP Server

🚀 Databricks MCP サーバー

このサーバーは、モデルコンテキストプロトコル（MCP）を使用して Databricks API に接続し、大規模言語モデル（LLMs）が SQL クエリの実行、ジョブの一覧表示、ジョブの状態取得を可能にします。

✨ 主な機能

• Databricks SQL ウェアハウス上で SQL クエリを実行します。
• すべての Databricks ジョブを一覧表示します。
• 特定の Databricks ジョブの状態を取得します。
• 特定の Databricks ジョブに関する詳細情報を取得します。

📦 インストール

前提条件

• Python 3.7 以上
• 以下の設定がされた Databricks ワークスペース：
  • 個人用アクセストークン
  • SQL ウェアハウスエンドポイント
  • クエリの実行とジョブへのアクセス権限

セットアップ手順

1. このリポジトリをクローンします。
2. 仮想環境を作成してアクティブ化します（推奨）：

   python -m venv .venv
   source .venv/bin/activate  # Windowsの場合：.venv\Scripts\activate
   

3. 依存関係をインストールします：

   pip install -r requirements.txt
   

4. ルートディレクトリに .env ファイルを作成し、以下の変数を含めます：

   DATABRICKS_HOST=your-databricks-instance.cloud.databricks.com
   DATABRICKS_TOKEN=your-personal-access-token
   DATABRICKS_HTTP_PATH=/sql/1.0/warehouses/your-warehouse-id
   

5. 接続をテストします（オプションですが推奨）：

   python test_connection.py
   

Databricks の資格情報の取得

1. ホスト：Databricks インスタンスの URL（例：your-instance.cloud.databricks.com）
2. トークン：Databricks で個人用アクセストークンを作成します：
   • ユーザー設定に移動します（右上のユーザー名をクリック）
   • 「開発者」タブを選択します
   • 「アクセストークン」の下で「管理」をクリックします
   • 新しいトークンを生成し、すぐに保存します
3. HTTP パス：SQL ウェアハウスの場合：
   • Databricks の SQL ウェアハウスに移動します
   • ウェアハウスを選択します
   • 接続詳細を見つけ、HTTP パスをコピーします

💻 使用例

サーバーの起動

MCP サーバーを起動します：

python main.py

この MCP サーバーを inspector でテストするには、以下を実行します：

npx @modelcontextprotocol/inspector python3 main.py

利用可能な MCP ツール

以下の MCP ツールが利用可能です：

1. run_sql_query(sql: str) - Databricks SQL ウェアハウス上で SQL クエリを実行します。
2. list_jobs() - ワークスペース内のすべての Databricks ジョブを一覧表示します。
3. get_job_status(job_id: int) - ジョブ ID を指定して特定の Databricks ジョブの状態を取得します。
4. get_job_details(job_id: int) - 特定の Databricks ジョブの詳細情報を取得します。

大規模言語モデルとの組み合わせの使用例

MCP プロトコルをサポートする大規模言語モデルと一緒に使用すると、このサーバーを使って自然言語で Databricks 環境と対話できます：

• 「データベース内のすべてのテーブルを表示」
• 「顧客テーブル内のレコード数を確認するクエリを実行」
• 「すべての Databricks ジョブを一覧表示」
• 「ジョブ #123 の状態を確認」
• 「ジョブ #456 の詳細を表示」

📚 詳細ドキュメント

トラブルシューティング

接続問題

• Databricks ホストが正しく、https:// プレフィックスを含まないことを確認します。
• SQL ウェアハウスが実行中でアクセス可能であることを確認します。
• 個人用アクセストークンが必要な権限を持っていることを検証します。
• 含まれているテストスクリプトを実行します：python test_connection.py

セキュリティに関する注意事項

• Databricks の個人用アクセストークンはワークスペースへの直接アクセスを提供します。
• .env ファイルが安全で、バージョン管理にコミットされないようにします。
• 適切な権限範囲を持つ Databricks トークンの使用を検討します。
• このサーバーを安全な環境で実行します。