MCP를 지원하는 PyTorch HUD API

PyTorch HUD API와 상호 작용하고 CI/CD 데이터, 작업 로그 및 분석에 대한 액세스를 제공하는 Python 라이브러리 및 MCP 서버입니다.

개요

이 프로젝트는 다음을 포함한 PyTorch CI/CD 분석을 위한 도구를 제공합니다.

• 워크플로, 작업 및 테스트 실행에 대한 데이터 액세스

• 대규모 CI 로그에 대한 효율적인 로그 분석

• 분석을 위한 ClickHouse 쿼리 통합

• 리소스 활용도 지표

Related MCP server: Swarmia 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