ami-mcp v0.2.1

一个封装了 ATLAS AMI（ATLAS 元数据接口）和 PMG 横截面数据库的 MCP 服务器，将其作为工具暴露给 LLM。专为需要发现 MC 样本、查找横截面和过滤效率以及验证 PMG 标签分类的 ATLAS 物理学家而设计。

功能介绍

ami-mcp 让 Claude（或任何兼容 MCP 的 LLM）能够直接查询 ATLAS 元数据：

• 发现样本：通过 PMG 标签分类（WeakBoson/Vjets/Baseline）、名称模式或任意 AMI 查询来搜索 MC 数据集

• 查找元数据：检索横截面、过滤效率、k-因子、数据集来源和 AMI 处理标签信息

• 查询横截面数据库：在 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)

• 网格 CA 证书（在 ATLAS 站点的 CVMFS 上可用）

快速入门

1. 设置身份验证

voms-proxy-init -voms atlas

通过 pip 安装时（非 pixi/conda-forge），还需要设置 X509_CERT_DIR。在基于 CVMFS 的设施（例如 UChicago 分析设施、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 查询

PMG 标签

物理元数据

横截面数据库

验证

示例提示词

配置完成后，您可以询问 Claude 如下问题：

• "Find all Baseline WeakBoson/Vjets samples in mc21_13TeV"

• "What are the cross-section and filter efficiency for DSID 700320?"

• "Look up the hashtag classification for this EVNT dataset"

• "Validate these samples against the mc21 cross-section database"

• "Show me the provenance chain for this DAOD dataset"

• "What AMI tag e8351 corresponds to — which generator version?"