Trackio MCP

trackio mcp

](https://badge.fury.io/py/trackio-mcp)  

MCP（模型上下文协议）服务器支持轨迹实验跟踪

该软件包使AI代理能够观察和交互 轨道 通过模型上下文协议（MCP）进行实验。只需导入 trackio_mcp 之前 trackio 以自动启用MCP服务器功能。

特性

• 零代码集成：只需导入 trackio_mcp 之前 trackio
• 自动MCP服务器：在所有trackio部署（本地和空间）上启用MCP服务器
• 丰富的工具集：将trackio功能作为人工智能代理的MCP工具公开
• 空间兼容：与Hugging Face Spaces部署无缝协作
• 简易替代者：无需更改现有trackio代码

安装

pip install trackio-mcp

或者具有开发依赖性：

pip install trackio-mcp[dev]

快速开始

基本用法

只需导入 trackio_mcp 进口前 trackio:

import trackio_mcp  # This enables MCP server functionality
import trackio as wandb

# Your existing trackio code works unchanged
wandb.init(project="my-experiment")
wandb.log({"loss": 0.1, "accuracy": 0.95})
wandb.finish()

MCP服务器将在以下位置自动可用：

• 本地: http://localhost:7860/gradio_api/mcp/sse
• 空间: https://your-space.hf.space/gradio_api/mcp/sse

使用MCP部署到拥抱面部空间

import trackio_mcp
import trackio as wandb

# Deploy to Spaces with MCP enabled automatically
wandb.init(
    project="my-experiment", 
    space_id="username/my-trackio-space"
)

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

独立MCP服务器

为trackio工具启动专用MCP服务器：

from trackio_mcp.tools import launch_trackio_mcp_server

# Launch standalone MCP server on port 7861
launch_trackio_mcp_server(port=7861, share=False)

可用的MCP工具

一旦连接，AI代理就可以使用这些trackio工具：

核心工具（通过Gradio API）

• 日志：将指标记录到trackio运行中
• upload_db_to_space：将本地数据库上载到空间

扩展工具（通过trackio mcp）

• get_项目：列出所有trackio项目
• get_runs：获取特定项目的跑步记录
• 过滤器_运行：筛选器按名称模式运行
• get_run_metrics：获取特定运行的指标数据
• 获取可用信息：获取项目的所有可用度量名称
• load_run_data：使用可选平滑功能加载和处理运行数据
• get_project_summary：获取全面的项目统计数据

代理交互示例

Human: "Show me the latest results from my 'image-classification' project"

Agent: I'll check your trackio projects and get the latest results.

[Tool: get_projects] → finds "image-classification" project
[Tool: get_runs] → gets runs for "image-classification" 
[Tool: get_run_metrics] → gets metrics for latest run
[Tool: get_available_metrics] → gets metric names

Agent: Your latest image-classification run achieved 94.2% accuracy with a final loss of 0.18. The model trained for 50 epochs with best validation accuracy of 94.7% at epoch 45.

MCP客户端配置

Claude Desktop

增添 ~/Library/Application Support/Claude/claude_desktop_config.json （macOS）或同等级：

公共空间：

{
  "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 Code

看 克劳德代码MCP文档 了解更多信息。

公共空间：

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"
    }
  }
}

Cursor

添加到光标 ~/.cursor/mcp.json 文件或创建 .cursor/mcp.json 在您的项目文件夹中。看 光标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"
    }
  }
}

Windsurf

添加到您的Windsurf MCP配置文件中。看 Windsurf MCP文件 了解更多信息。

公共空间：

{
  "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"
    }
  }
}

VS Code

增添 .vscode/mcp.json。参见 VS代码MCP文档 了解更多信息。

公共空间：

{
  "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"
      }
    }
  }
}

Gemini CLI

增添 mcp.json 在您的项目目录中。看 Gemini CLI配置 了解详情。

公共空间：

{
  "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"]
    }
  }
}

Cline

创建 .cursor/mcp.json （或IDE的等效工具）：

公共空间：

{
  "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：设置为 "true" 禁用MCP功能（默认：启用MCP）

程序控制

import os
os.environ["TRACKIO_DISABLE_MCP"] = "true"  # Disable MCP
import trackio_mcp  # MCP won't be enabled
import trackio

运作原理

trackio-mcp 使用monkey补丁自动执行以下操作：

1. 启用MCP服务器：集合 mcp_server=True 在所有Gradio发射上
2. 启用API：集合 show_api=True 公开Gradio API端点
3. 添加工具：注册其他特定于赛道的MCP工具
4. 保持兼容性：无需更改现有trackio代码

该软件包补丁：

• gradio.Blocks.launch() -核心Gradio发射方法
• trackio.ui.demo.launch() -Trackio仪表板启动
• 在以下位置添加新的MCP端点 /gradio_api/mcp/sse

部署示例

地方发展

import trackio_mcp
import trackio

# Start local tracking with MCP enabled
trackio.show()  # Dashboard + MCP server at http://localhost:7860

公共空间部署

import trackio_mcp
import trackio as wandb

# Deploy to public Spaces with MCP support
wandb.init(
    project="public-model",
    space_id="username/model-tracking"
)

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

私有空间/数据集部署

import trackio_mcp
import trackio as wandb

# Deploy to private Spaces with private dataset
wandb.init(
    project="private-model",
    space_id="organization/private-model-tracking",  # Private space
    dataset_id="organization/private-model-metrics"  # Private dataset
)

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

CLI接口

# Launch standalone MCP server
trackio-mcp server --port 7861

# Check status and configuration
trackio-mcp status

# Test MCP server functionality
trackio-mcp test --url http://localhost:7860

安全考虑

• 私人空间：使用HF令牌对私有空间/数据集进行身份验证
• 访问控制：MCP服务器继承trackio的访问控制
• 网络安全：考虑生产部署的防火墙规则
• 许可证管理：安全地存储HF令牌，使用环境变量

故障排除

MCP服务器不可用

import trackio_mcp
import trackio

# Check if MCP was disabled
import os
print("MCP Disabled:", os.getenv("TRACKIO_DISABLE_MCP"))

# Manual verification
trackio.show()  # Look for MCP server URL in output

连接问题

1. 检查网址：确保正确 /gradio_api/mcp/sse 端点
2. 认证：为私有空间/数据集添加Bearer令牌
3. 网络：验证防火墙/代理设置
4. 依赖项：确保 gradio[mcp] 已安装

工具发现问题

# Test tools manually
from trackio_mcp.tools import register_trackio_tools

tools = register_trackio_tools()
tools.launch(mcp_server=True)  # Test tools interface

贡献

1. 分叉存储库
2. 安装开发依赖项： pip install -e .[dev]
3. 进行更改
4. 运行测试： pytest
5. 提交拉取请求

许可证

MIT许可证-请参阅 许可证 文件。

致谢

• 轨道 -优秀的实验跟踪库
• 格拉迪奥 -用于内置MCP服务器支持
• 模型上下文协议 -用于标准化的AI工具协议

______________________________________________________________________

为人工智能研究界精心打造