Claude Pytorch Treehugger

🚀 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 許可證。