Mage.ai MCP Integration

Mage.ai MCP 統合

Mage.ai 向けの包括的な Model Context Protocol（MCP）統合です。Anthropic の Claude などの AI アシスタントが、Mage のデータパイプラインを効率よく操作できます。

機能

• パイプラインの参照：Mage.ai のすべてのパイプラインとそのコンポーネントを確認できます
• コードの表示と編集：ブロックのコードを読み取り、変更できます
• パイプラインの実行：パイプライン全体または個別のブロックを実行できます
• 検索機能：名前、タイプ、内容でブロックを検索できます
• パイプラインの作成：新しいパイプラインとブロックを一から作成できます

インストール

# Clone the repository
git clone https://github.com/yourusername/mage-mcp.git
cd mage-mcp

# Install dependencies
pip install -r requirements.txt

設定

サーバーは環境変数またはコマンドライン引数を使用して設定できます。

• MAGE_API_URL：Mage.ai API の URL（デフォルト：http://localhost:6789/api/）
• MAGE_API_KEY：認証用の API キー（必要な場合）

設定を永続化するには、.env ファイルを作成します。

MAGE_API_URL=http://localhost:6789/api/
MAGE_API_KEY=your_api_key_if_needed

使用方法

サーバーの起動

付属のスクリプトを使用して MCP サーバーを起動します。

# Using stdio transport (default)
python run_server.py

# Using SSE transport
python run_server.py --transport sse --host localhost --port 3000

# Additional options
python run_server.py --api-url http://your-mage-instance:6789/api/ --api-key your_key --debug

Claude Desktop へのインストール

最適な環境で利用するには、このサーバーを Claude Desktop に直接インストールしてください。

# Using mcp CLI
mcp install run_server.py --name "Mage.ai Manager"

# With environment variables
mcp install run_server.py -v MAGE_API_URL=http://localhost:6789/api/ -v MAGE_API_KEY=your_key

開発モード

テストおよび開発用です。

mcp dev run_server.py

MCP リソースとツール

リソース

• mage://pipelines - すべてのパイプラインを一覧表示
• mage://pipelines/{pipeline_uuid} - パイプラインの詳細を取得
• mage://pipelines/{pipeline_uuid}/blocks - パイプライン内のブロックを一覧表示
• mage://pipelines/{pipeline_uuid}/blocks/{block_uuid} - ブロックの詳細を取得
• mage://pipelines/{pipeline_uuid}/blocks/{block_uuid}/content - ブロックのコードを取得
• mage://pipelines/{pipeline_uuid}/content - パイプライン全体の内容を取得
• mage://pipelines/{pipeline_uuid}/execution - パイプラインの実行詳細を取得
• mage://blocks/search/{query} - 名前、タイプ、内容でブロックを検索

ツール

• list_all_pipelines() - 詳細情報付きですべてのパイプラインを一覧表示
• get_pipeline_details(pipeline_uuid) - パイプラインの情報を取得
• list_pipeline_blocks(pipeline_uuid) - パイプライン内のブロックを一覧表示
• get_block_content(pipeline_uuid, block_uuid) - ブロックのコードを取得
• create_pipeline(name, description, pipeline_type) - 新しいパイプラインを作成
• create_block(pipeline_uuid, name, block_type, language, content, upstream_blocks) - ブロックを作成
• update_block_content(pipeline_uuid, block_uuid, content) - ブロックのコードを更新
• execute_pipeline(pipeline_uuid) - パイプラインを実行
• execute_block(pipeline_uuid, block_uuid) - 指定したブロックを実行
• get_pipeline_code(pipeline_uuid) - パイプラインのコード全体を取得

インタラクション例

User: Show me all the pipelines in Mage.ai.
Claude: Let me check what pipelines are available in your Mage.ai instance.

[Lists all pipelines with descriptions and types]

User: Can you show me the blocks in the data_ingestion pipeline?
Claude: Here are all the blocks in the data_ingestion pipeline:

[Lists all blocks with their types, languages, and status]

User: I need to modify the 'transform_data' block to fix a bug.
Claude: Let me help you with that. First, let's look at the current code:

[Shows the current block code]

Here's how we can fix the bug:
[Explains the issue and suggests changes]

Would you like me to update the block with these changes?

User: Yes, please update it.
Claude: I've updated the block. Here's the new content:

[Shows the updated code]

The changes have been saved successfully.

コントリビューション

コントリビューションを歓迎します。お気軽にプルリクエストを送信してください。

ライセンス

このプロジェクトは MIT License のもとでライセンスされています。詳細は LICENSE ファイルをご覧ください。