Trackio MCP

🚀 trackio-mcp

trackio実験追跡のためのMCP (Model Context Protocol) サーバーサポート

このパッケージは、AIエージェントがModel Context Protocol (MCP) を通じて trackio の実験を観察し、相互作用することを可能にします。trackio の前に trackio_mcp をインポートするだけで、自動的にMCPサーバー機能が有効になります。

🚀 クイックスタート

基本的な使用法

trackio をインポートする前に、単に trackio_mcp をインポートします。

import trackio_mcp  # これによりMCPサーバー機能が有効になります
import trackio as wandb

# 既存のtrackioコードは変更なしで動作します
wandb.init(project="my-experiment")
wandb.log({"loss": 0.1, "accuracy": 0.95})
wandb.finish()

MCPサーバーは以下の場所で自動的に利用可能になります。

• ローカル: http://localhost:7860/gradio_api/mcp/sse
• Spaces: https://your-space.hf.space/gradio_api/mcp/sse

MCPを使用してHugging Face Spacesにデプロイする

import trackio_mcp
import trackio as wandb

# MCPが自動的に有効になった状態でSpacesにデプロイします
wandb.init(
    project="my-experiment", 
    space_id="username/my-trackio-space"
)

wandb.log({"loss": 0.1})
wandb.finish()

スタンドアロンのMCPサーバーを起動する

from trackio_mcp.tools import launch_trackio_mcp_server

# ポート7861でスタンドアロンのMCPサーバーを起動します
launch_trackio_mcp_server(port=7861, share=False)

✨ 主な機能

• ゼロコードでの統合: trackio の前に trackio_mcp をインポートするだけです。
• 自動MCPサーバー: すべてのtrackioデプロイメント（ローカルとSpaces）でMCPサーバーを有効にします。
• 豊富なツールセット: trackioの機能をAIエージェント用のMCPツールとして公開します。
• Spaces互換: Hugging Face Spacesデプロイメントとシームレスに動作します。
• ドロップイン置換: 既存のtrackioコードに変更は必要ありません。

📦 インストール

pip install trackio-mcp

または、開発用の依存関係を含める場合は、以下のコマンドを使用します。

pip install trackio-mcp[dev]

💻 使用例

基本的な使用法

import trackio_mcp  # これによりMCPサーバー機能が有効になります
import trackio as wandb

# 既存のtrackioコードは変更なしで動作します
wandb.init(project="my-experiment")
wandb.log({"loss": 0.1, "accuracy": 0.95})
wandb.finish()

高度な使用法

# MCPを使用してHugging Face Spacesにデプロイする
import trackio_mcp
import trackio as wandb

# MCPが自動的に有効になった状態でSpacesにデプロイします
wandb.init(
    project="my-experiment", 
    space_id="username/my-trackio-space"
)

wandb.log({"loss": 0.1})
wandb.finish()

利用可能なMCPツール

接続後、AIエージェントは以下のtrackioツールを使用できます。

コアツール (Gradio API経由)

• log: trackioの実行にメトリクスを記録します。
• upload_db_to_space: ローカルデータベースをSpaceにアップロードします。

拡張ツール (trackio-mcp経由)

• get_projects: すべてのtrackioプロジェクトをリストします。
• get_runs: 特定のプロジェクトの実行を取得します。
• filter_runs: 名前パターンで実行をフィルタリングします。
• get_run_metrics: 特定の実行のメトリクスデータを取得します。
• get_available_metrics: プロジェクトのすべての利用可能なメトリクス名を取得します。
• load_run_data: 実行データをロードし、必要に応じて平滑化して処理します。
• get_project_summary: 包括的なプロジェクト統計を取得します。

エージェントの対話例

Human: "私の 'image-classification' プロジェクトの最新の結果を見せてください"

Agent: あなたのtrackioプロジェクトを確認し、最新の結果を取得します。

[Tool: get_projects] → "image-classification" プロジェクトを見つけます
[Tool: get_runs] → "image-classification" の実行を取得します
[Tool: get_run_metrics] → 最新の実行のメトリクスを取得します
[Tool: get_available_metrics] → メトリクス名を取得します

Agent: あなたの最新の画像分類実行は、最終損失0.18で94.2%の精度を達成しました。モデルは50エポックでトレーニングされ、エポック45で最高の検証精度94.7%を達成しました。

MCPクライアントの設定

{
  "mcpServers": {
    "trackio": {
      "url": "https://your-space.hf.space/gradio_api/mcp/sse"
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "url": "https://your-private-space.hf.space/gradio_api/mcp/sse",
      "headers": {
        "Authorization": "Bearer YOUR_HF_TOKEN"
      }
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "url": "http://localhost:7860/gradio_api/mcp/sse"
    }
  }
}

claude mcp add --transport sse trackio https://your-space.hf.space/gradio_api/mcp/sse

claude mcp add --transport sse --header "Authorization: Bearer YOUR_HF_TOKEN" trackio https://your-private-space.hf.space/gradio_api/mcp/sse

{
  "mcpServers": {
    "trackio": {
      "type": "sse",
      "url": "http://localhost:7860/gradio_api/mcp/sse"
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "url": "https://your-space.hf.space/gradio_api/mcp/sse"
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "url": "https://your-private-space.hf.space/gradio_api/mcp/sse",
      "headers": {
        "Authorization": "Bearer YOUR_HF_TOKEN"
      }
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "url": "http://localhost:7860/gradio_api/mcp/sse"
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "serverUrl": "https://your-space.hf.space/gradio_api/mcp/sse"
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "serverUrl": "https://your-private-space.hf.space/gradio_api/mcp/sse",
      "headers": {
        "Authorization": "Bearer YOUR_HF_TOKEN"
      }
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "serverUrl": "http://localhost:7860/gradio_api/mcp/sse"
    }
  }
}

{
  "mcp": {
    "servers": {
      "trackio": {
        "type": "http",
        "url": "https://your-space.hf.space/gradio_api/mcp/sse"
      }
    }
  }
}

{
  "mcp": {
    "servers": {
      "trackio": {
        "type": "http", 
        "url": "https://your-private-space.hf.space/gradio_api/mcp/sse",
        "headers": {
          "Authorization": "Bearer YOUR_HF_TOKEN"
        }
      }
    }
  }
}

{
  "mcp": {
    "servers": {
      "trackio": {
        "type": "http",
        "url": "http://localhost:7860/gradio_api/mcp/sse"
      }
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "command": "npx",
      "args": ["mcp-remote", "https://your-space.hf.space/gradio_api/mcp/sse"]
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "command": "npx", 
      "args": ["mcp-remote", "https://your-private-space.hf.space/gradio_api/mcp/sse"],
      "env": {
        "HF_TOKEN": "YOUR_HF_TOKEN"
      }
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "command": "npx",
      "args": ["mcp-remote", "http://localhost:7860/gradio_api/mcp/sse"]
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "url": "https://your-space.hf.space/gradio_api/mcp/sse"
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "url": "https://your-private-space.hf.space/gradio_api/mcp/sse",
      "headers": {
        "Authorization": "Bearer YOUR_HF_TOKEN"
      }
    }
  }
}

{
  "mcpServers": {
    "trackio": {
      "url": "http://localhost:7860/gradio_api/mcp/sse"
    }
  }
}

設定

環境変数

• TRACKIO_DISABLE_MCP: MCP機能を無効にするには "true" に設定します（デフォルト: MCP有効）

プログラムによる制御

import os
os.environ["TRACKIO_DISABLE_MCP"] = "true"  # MCPを無効にする
import trackio_mcp  # MCPは有効になりません
import trackio

動作原理

trackio-mcp はモンキーパッチを使用して自動的に以下のことを行います。

1. MCPサーバーを有効にする: すべてのGradio起動で mcp_server=True を設定します。
2. APIを有効にする: show_api=True を設定して、Gradio APIエンドポイントを公開します。
3. ツールを追加する: 追加のtrackio固有のMCPツールを登録します。
4. 互換性を維持する: 既存のtrackioコードに変更は必要ありません。

このパッケージは以下をパッチします。

• gradio.Blocks.launch() - コアGradio起動メソッド
• trackio.ui.demo.launch() - trackioダッシュボードの起動
• /gradio_api/mcp/sse に新しいMCPエンドポイントを追加します。

デプロイメント例

ローカル開発

import trackio_mcp
import trackio

# MCPを有効にしてローカルトラッキングを開始します
trackio.show()  # ダッシュボード + MCPサーバーが http://localhost:7860 で起動します

パブリックSpacesデプロイメント

import trackio_mcp
import trackio as wandb

# MCPサポート付きでパブリックSpacesにデプロイします
wandb.init(
    project="public-model",
    space_id="username/model-tracking"
)

wandb.log({"epoch": 1, "loss": 0.5})
wandb.finish()

プライベートSpaces/データセットデプロイメント

import trackio_mcp
import trackio as wandb

# プライベートデータセット付きでプライベートSpacesにデプロイします
wandb.init(
    project="private-model",
    space_id="organization/private-model-tracking",  # プライベートSpace
    dataset_id="organization/private-model-metrics"  # プライベートデータセット
)

wandb.log({"epoch": 1, "loss": 0.5})
wandb.finish()

CLIインターフェイス

# スタンドアロンのMCPサーバーを起動します
trackio-mcp server --port 7861

# ステータスと設定を確認します
trackio-mcp status

# MCPサーバーの機能をテストします
trackio-mcp test --url http://localhost:7860

セキュリティに関する考慮事項

• プライベートSpaces: プライベートSpaces/データセットとの認証にはHFトークンを使用します。
• アクセス制御: MCPサーバーはtrackioのアクセス制御を引き継ぎます。
• ネットワークセキュリティ: 本番デプロイメントではファイアウォールルールを考慮します。
• トークン管理: HFトークンを安全に保管し、環境変数を使用します。

トラブルシューティング

MCPサーバーが利用できない場合

import trackio_mcp
import trackio

# MCPが無効になっているか確認します
import os
print("MCP Disabled:", os.getenv("TRACKIO_DISABLE_MCP"))

# 手動で確認します
trackio.show()  # 出力でMCPサーバーのURLを探します

接続問題

1. URLを確認する: 正しい /gradio_api/mcp/sse エンドポイントを確認します。
2. 認証: プライベートSpaces/データセットにはベアラートークンを追加します。
3. ネットワーク: ファイアウォール/プロキシ設定を確認します。
4. 依存関係: gradio[mcp] がインストールされていることを確認します。

ツール検出問題

# ツールを手動でテストします
from trackio_mcp.tools import register_trackio_tools

tools = register_trackio_tools()
tools.launch(mcp_server=True)  # ツールインターフェイスをテストします

コントリビューション

1. リポジトリをフォークします。
2. 開発用の依存関係をインストールします: pip install -e .[dev]
3. 変更を加えます。
4. テストを実行します: pytest
5. プルリクエストを送信します。

📄 ライセンス

MITライセンス - LICENSE ファイルを参照してください。

謝辞

• trackio - 優れた実験追跡ライブラリ
• Gradio - 組み込みのMCPサーバーサポート
• Model Context Protocol - 標準化されたAIツールプロトコル

AI研究コミュニティのために誠意を持って作成されました