Skip to main content
Glama

ami-mcp v0.2.1

Actions Status Documentation Status

PyPI version PyPI platforms

GitHub Discussion

Coverage

ATLAS AMI(ATLAS Metadata Interface)와 PMG 단면적 데이터베이스를 래핑하여 LLM을 위한 도구로 노출하는 MCP 서버입니다. MC 샘플을 탐색하고, 단면적 및 필터 효율을 조회하며, PMG 해시태그 분류를 검증해야 하는 ATLAS 물리학자를 위해 설계되었습니다.

주요 기능

ami-mcp를 사용하면 Claude(또는 MCP 호환 LLM)가 ATLAS 메타데이터를 직접 쿼리할 수 있습니다:

  • 샘플 탐색: PMG 해시태그 분류(WeakBoson/Vjets/Baseline), 이름 패턴 또는 임의의 AMI 쿼리를 통해 MC 데이터셋 검색

  • 메타데이터 조회: 단면적, 필터 효율, k-팩터, 데이터셋 출처 및 AMI 처리 태그 정보 검색

  • 단면적 DB 쿼리: PMG xsec 데이터베이스 파일(PMGxsecDB_mc16.txt 등)에서 DSID 항목 조회

  • 샘플 검증: 해시태그 분류 확인 및 PMG 단면적 데이터베이스와 메타데이터 비교

  • 일반 쿼리: ami://query-language 리소스를 가이드로 사용하여 LLM이 작성한 임의의 AMI 명령 문자열 실행

Related MCP server: rucio-mcp

설치

pip install ami-mcp

또는 pixi 사용 (ATLAS 시설 권장):

pixi add ami-mcp

요구 사항

  • Python 3.10 또는 3.11 (pyAMI는 <3.12 필요)

  • 유효한 VOMS 프록시 (voms-proxy-init -voms atlas)

  • Grid CA 인증서 (ATLAS 사이트의 CVMFS에서 사용 가능)

빠른 시작

1. 인증 설정

voms-proxy-init -voms atlas

pip을 통해 설치한 경우(pixi/conda-forge가 아님), X509_CERT_DIR도 설정해야 합니다. CVMFS 기반 시설(예: UChicago Analysis Facility, CERN lxplus)에서는 다음과 같이 합니다:

export X509_CERT_DIR=/cvmfs/atlas.cern.ch/repo/ATLASLocalRootBase/etc/grid-security-emi/certificates

pixi 또는 conda-forge를 통해 설치하면 ca-policy-lcg가 포함되어 X509_CERT_DIR을 자동으로 설정하므로 추가 단계가 필요하지 않습니다.

2. 서버 테스트

ami-mcp serve

서버는 stdio를 통해 MCP로 통신합니다. MCP 클라이언트가 이를 실행하도록 구성하십시오.

3. Claude Code 구성

pixi 사용 (권장 — X509_CERT_DIR이 자동으로 설정됨):

{
  "mcpServers": {
    "ami": {
      "type": "stdio",
      "command": "pixi",
      "args": ["run", "ami-mcp", "serve"],
      "env": {
        "ATLAS_PMGXSEC_PATH": "/cvmfs/atlas.cern.ch/repo/sw/database/GroupData/dev/PMGTools"
      }
    }
  }
}

pip 사용 (X509_CERT_DIR을 수동으로 설정해야 함):

{
  "mcpServers": {
    "ami": {
      "command": "ami-mcp",
      "args": ["serve"],
      "env": {
        "X509_CERT_DIR": "/cvmfs/atlas.cern.ch/repo/ATLASLocalRootBase/etc/grid-security-emi/certificates",
        "ATLAS_PMGXSEC_PATH": "/cvmfs/atlas.cern.ch/repo/sw/database/GroupData/dev/PMGTools"
      }
    }
  }
}

4. Claude Desktop 구성

~/Library/Application Support/Claude/claude_desktop_config.json(macOS) 또는 %APPDATA%\Claude\claude_desktop_config.json(Windows)에 추가하십시오:

{
  "mcpServers": {
    "ami": {
      "command": "ami-mcp",
      "args": ["serve"],
      "env": {
        "X509_CERT_DIR": "/path/to/ca-certificates",
        "ATLAS_PMGXSEC_PATH": "/path/to/PMGTools"
      }
    }
  }
}

사용 가능한 도구

AMI 쿼리

도구

설명

ami_execute

모든 AMI 명령 문자열 실행 (주요 강력 도구)

ami_get_dataset_info

데이터셋에 대한 메타데이터 가져오기 (nFiles, nEvents, 상태 등)

ami_get_dataset_prov

계보 요약이 포함된 출처 체인 가져오기; 데이터 유형별 필터링

ami_list_datasets

이름 패턴으로 데이터셋 검색 (% 와일드카드 지원)

PMG 해시태그

도구

설명

ami_search_by_hashtags

해시태그 조합으로 데이터셋 찾기 (예: WeakBoson/Vjets/Baseline)

ami_get_dataset_hashtags

데이터셋에 대한 PMGL1–PMGL4 분류 조회

물리 메타데이터

도구

설명

ami_get_physics_params

AMI에서 단면적(nb→pb 변환), 필터 효율, k-팩터 가져오기

ami_get_ami_tag

AMI 처리 태그 정보 가져오기; e8351_s3681_r13144와 같은 태그 체인 허용

단면적 데이터베이스

도구

설명

ami_list_xsec_databases

사용 가능한 PMGxsecDB_*.txt 파일 나열

ami_lookup_xsec

xsec DB에서 DSID 단면적, 필터 효율, k-팩터 조회

검증

도구

설명

ami_validate_sample

해시태그 분류 확인 및 xsec DB와 메타데이터 비교

예시 프롬프트

구성이 완료되면 Claude에게 다음과 같이 질문할 수 있습니다:

  • "mc21_13TeV에서 모든 Baseline WeakBoson/Vjets 샘플을 찾아줘"

  • "DSID 700320의 단면적과 필터 효율은 얼마인가요?"

  • "이 EVNT 데이터셋의 해시태그 분류를 조회해줘"

  • "mc21 단면적 데이터베이스를 기준으로 이 샘플들을 검증해줘"

  • "이 DAOD 데이터셋의 출처 체인을 보여줘"

  • "AMI 태그 e8351은 무엇에 해당하나요? 어떤 생성기 버전인가요?"

Related MCP Connectors

Related MCP Servers

  • A
    license
    Not graded
    quality
    B
    maintenance
    An MCP server that exposes Rucio distributed data management operations as tools for LLMs. Designed for ATLAS physicists working with grid data on analysis facilities, but usable with any Rucio instance.
    42 PyPI
    6
    Apache 2.0