Skip to main content
Glama

ami-mcp v0.2.1

Actions Status Documentation Status

PyPI version PyPI platforms

GitHub Discussion

Coverage

MCP-сервер, который оборачивает ATLAS AMI (интерфейс метаданных ATLAS) и базу данных сечений PMG, предоставляя их в качестве инструментов для LLM. Разработан для физиков ATLAS, которым необходимо находить образцы MC, искать сечения и эффективность фильтрации, а также проверять классификацию хэштегов PMG.

Что он делает

ami-mcp позволяет Claude (или любой другой LLM, совместимой с MCP) запрашивать метаданные ATLAS напрямую:

  • Поиск образцов: поиск наборов данных MC по классификации хэштегов PMG (WeakBoson/Vjets/Baseline), по шаблону имени или с помощью произвольного запроса AMI

  • Поиск метаданных: получение сечений, эффективности фильтрации, k-факторов, происхождения наборов данных и информации о тегах обработки AMI

  • Запрос базы данных сечений: поиск записей DSID в файлах базы данных сечений PMG (PMGxsecDB_mc16.txt и т. д.)

  • Проверка образцов: проверка классификации хэштегов и сравнение метаданных с базой данных сечений PMG

  • Общие запросы: выполнение произвольных строк команд AMI, сформулированных LLM с использованием ресурса ami://query-language в качестве руководства

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 (доступны на CVMFS на объектах ATLAS)

Быстрый старт

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

Сервер общается по протоколу MCP через stdio. Настройте свой 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

Получение сечения (конвертировано из nb в pb), эффективности фильтрации, k-фактора из AMI

ami_get_ami_tag

Получение информации о теге обработки AMI; принимает цепочки тегов, такие как e8351_s3681_r13144

База данных сечений

Инструмент

Описание

ami_list_xsec_databases

Список доступных файлов PMGxsecDB_*.txt

ami_lookup_xsec

Поиск сечения DSID, эффективности фильтрации, k-фактора в БД сечений

Проверка

Инструмент

Описание

ami_validate_sample

Проверка классификации хэштегов и сравнение метаданных с БД сечений

Примеры запросов

После настройки вы можете спрашивать Claude о следующем:

  • "Найди все образцы Baseline WeakBoson/Vjets в mc21_13TeV"

  • "Каковы сечение и эффективность фильтрации для 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