MCP를 지원하는 PyTorch HUD API
PyTorch HUD API와 상호 작용하고 CI/CD 데이터, 작업 로그 및 분석에 대한 액세스를 제공하는 Python 라이브러리 및 MCP 서버입니다.
개요
이 프로젝트는 다음을 포함한 PyTorch CI/CD 분석을 위한 도구를 제공합니다.
워크플로, 작업 및 테스트 실행에 대한 데이터 액세스
대규모 CI 로그에 대한 효율적인 로그 분석
분석을 위한 ClickHouse 쿼리 통합
리소스 활용도 지표
Related MCP server: PlainSignal MCP Server
사용법(인간용)
지엑스피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/
선적 서류 비치
CLAUDE.md : 자세한 사용법, 코드 스타일 및 구현 참고 사항
mcp-guide.md : 일반 MCP 프로토콜 정보
특허
MIT