MCP Databricks Server

🚀 Databricks MCP サーバー

このサーバーは、モデルコンテキストプロトコル（MCP）を使用して、Databricks API に接続し、大規模言語モデルが 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

サーバーのテスト

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

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 ジョブの詳細情報を取得します

LLM との組み合わせでの使用

MCP プロトコルをサポートする LLM と組み合わせて使用すると、このサーバーを通じて自然言語で Databricks 環境と対話できます：

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

📚 ドキュメント

トラブルシューティング

接続問題

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

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

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