FindMCP Logo FindMCP.info
PyTorch HUD API with MCP Support

PyTorch HUD API with MCP Support

site icon
2025.03.15 1
PythonCI/CD 分析日志分析测试结果分析开发效率
claude-pytorch-treehugger 是一个 Python 库和 MCP 服务器，用于与 PyTorch HUD API 交互，提供对 CI/CD 数据、作业日志和分析的访问。该项目为 PyTorch CI/CD 分析提供工具，包括工作流、作业和测试运行的数据访问，大型 CI 日志的高效日志分析，ClickHouse 查询集成以及资源利用率指标。
View on GitHub

Overview

基本能力

产品定位

claude-pytorch-treehugger 是一个专门为 PyTorch CI/CD 流程设计的分析工具，提供数据访问和日志分析功能。

核心功能

  1. 数据访问
  2. get_commit_summary: 获取基本提交信息（不含作业）
  3. get_job_summary: 获取聚合的作业状态计数
  4. get_filtered_jobs: 根据状态/工作流/名称过滤作业
  5. get_failure_details: 获取失败作业的详细失败信息

  6. get_recent_commit_status: 获取最近提交的状态及作业统计

  7. 日志分析

  8. download_log_to_file: 下载日志到本地存储
  9. extract_log_patterns: 查找错误、警告等
  10. extract_test_results: 解析测试执行结果
  11. filter_log_sections: 提取特定日志部分

  12. search_logs: 跨多个日志搜索

  13. 其他功能

  14. ClickHouse 查询集成
  15. 资源利用率指标

适用场景

  • PyTorch 项目的 CI/CD 流程监控和分析
  • 大型 CI 日志的高效分析
  • 测试结果和失败原因的快速定位

使用教程

使用依赖

# 安装依赖（如果不使用 pip 安装）
pip install -r requirements.txt

安装教程

# 从 GitHub 仓库安装
pip install git+https://github.com/izaitsevfb/claude-pytorch-treehugger.git

# 添加 MCP 服务器
claude mcp add hud pytorch-hud

调试方式

# 启动 MCP 服务器
python -m pytorch_hud

# 运行测试
python -m unittest discover test

# 类型检查
mypy -p pytorch_hud -p test

# 代码检查
ruff check pytorch_hud/ test/

许可证

该项目遵循 MIT 开源许可条款。