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补丁自动执行以下操作：
- 启用MCP服务器：集合
mcp_server=True在所有Gradio发射上 - 启用API：集合
show_api=True公开Gradio API端点 - 添加工具：注册其他特定于赛道的MCP工具
- 保持兼容性：无需更改现有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连接问题
- 检查网址：确保正确
/gradio_api/mcp/sse端点 - 认证：为私有空间/数据集添加Bearer令牌
- 网络：验证防火墙/代理设置
- 依赖项：确保
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贡献
- 分叉存储库
- 安装开发依赖项：
pip install -e .[dev] - 进行更改
- 运行测试：
pytest - 提交拉取请求
许可证
MIT许可证-请参阅 许可证 文件。
致谢
______________________________________________________________________
为人工智能研究界精心打造