MCP Server for ZenML
Сервер для интеграции ZenML MLOps и LLMOps пайплайнов с клиентами MCP вроде Claude Desktop или Cursor.
Описание
Это MCP-сервер на Python для взаимодействия с API ZenML, позволяющий LLM получать доступ к информации о пользователях, стеках, пайплайнах, артефактах и логах. Поддерживает запуск новых пайплайнов. Идеален для AI-инструментов в MLOps: мониторинг экспериментов, анализ данных без ручного доступа. Использует UV для установки, Docker для контейнеризации, интегрируется с Anthropic MCP Registry. Бета-версия с автоматизированными тестами.
Возможности
Доступ к сущностям ZenML
Получение информации о пользователях, стеках, пайплайнах, шагах, сервисах, артефактах и коннекторах.
Мониторинг пайплайнов
Просмотр запусков, расписаний, шаблонов, шагов, логов и кода шагов (для облачных стеков).
Запуск пайплайнов
Триггеринг новых запусков пайплайнов при наличии шаблонов.
Интеграция с MCP
Стандартизированное подключение к LLM-клиентам для безопасного доступа к локальным и удаленным данным ZenML.
Установка
Клонируйте репозиторий: git clone https://github.com/zenml-io/mcp-zenml.git. Установите UV (рекомендуется через brew на Mac). Создайте MCP-конфиг (mcp.json):
{
"mcpServers": {
"zenml": {
"command": "/usr/local/bin/uv",
"args": ["run", "path/to/server/zenml_server.py"],
"env": {
"ZENML_STORE_URL": "https://your-zenml-server.com",
"ZENML_STORE_API_KEY": "your-api-key"
}
}
}
}
Замените пути и ключи. Для Claude Desktop: перетащите mcp-zenml.mcpb в настройки. Для Cursor: создайте .cursor/mcp.json в репозитории и включите в настройках. Docker: docker pull zenmldocker/mcp-zenml:latest и запустите с env-переменными.