MCP를 지원하는 PyTorch HUD API
PyTorch HUD API와 상호 작용하고 CI/CD 데이터, 작업 로그 및 분석에 대한 액세스를 제공하는 Python 라이브러리 및 MCP 서버입니다.
개요
이 프로젝트는 다음을 포함한 PyTorch CI/CD 분석을 위한 도구를 제공합니다.
- 워크플로, 작업 및 테스트 실행에 대한 데이터 액세스
- 대규모 CI 로그에 대한 효율적인 로그 분석
- 분석을 위한 ClickHouse 쿼리 통합
- 리소스 활용도 지표
사용법(인간용)
지엑스피1
claude mcp add hud pytorch-hud
개발
# Install dependencies (if not installing with pip)
pip install -r requirements.txt
# Start MCP server
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
: 여러 로그에서 검색
개발
# Run tests
python -m unittest discover test
# Type checking
mypy -p pytorch_hud -p test
# Linting
ruff check pytorch_hud/ test/
선적 서류 비치
특허
MIT