🚀 PyTorch HUD API 带 MCP 支持的 Python 库
本项目是一个用于与 PyTorch HUD API 交互的 Python 库和 MCP 服务器，为 CI/CD 数据、作业日志和分析提供了全面支持，能有效助力 PyTorch 的持续集成与交付工作。
🚀 快速开始
面向用户的安装与使用
若你是普通用户，可按以下步骤操作：
# 从 GitHub 仓库安装
pip install git+https://github.com/izaitsevfb/claude-pytorch-treehugger.git
claude mcp add hud pytorch-hud
开发者的安装与启动
如果你是开发者，可按以下步骤进行开发环境的搭建与启动：
# 安装依赖项（如果未通过 pip 安装）
pip install -r requirements.txt
# 启动 MCP 服务器
python -m pytorch_hud
✨ 主要特性
数据访问
get_commit_summary：获取基本提交信息，不包含作业。get_job_summary：获取聚合的作业状态统计。get_filtered_jobs：根据状态、工作流程或名称筛选作业。get_failure_details：获取失败作业的详细失败信息。get_recent_commit_status：获取近期提交的状态及作业统计数据。
日志分析
download_log_to_file：将日志下载到本地存储。extract_log_patterns：查找错误、警告等信息。extract_test_results：解析测试执行结果。filter_log_sections：提取特定的日志部分。search_logs：在多个日志中进行搜索。
📦 安装指南
用户安装
pip install git+https://github.com/izaitsevfb/claude-pytorch-treehugger.git
开发者安装
pip install -r requirements.txt
💻 使用示例
基础用法
# 从 GitHub 仓库安装
pip install git+https://github.com/izaitsevfb/claude-pytorch-treehugger.git
# 添加 HUD
claude mcp add hud pytorch-hud
高级用法
# 开发者安装依赖
pip install -r requirements.txt
# 启动 MCP 服务器
python -m pytorch_hud
📚 详细文档
- CLAUDE.md：详细使用说明、代码风格和实现注意事项。
- mcp-guide.md：MCP 协议的通用信息。
🔧 技术细节
此项目提供了以下工具用于 PyTorch 的持续集成/交付（CI/CD）数据分析：
- 获取工作流程、作业和测试运行的数据。
- 高效分析大型 CI 日志。
- ClickHouse 查询集成用于分析。
- 资源利用率指标。
📄 许可证
本项目采用 MIT 许可证。