Zenml MCP server

MCP Server for ZenML

This project implements a Model Context Protocol (MCP) server for interacting with the ZenML API.

What is MCP?

The Model Context Protocol (MCP) is an open protocol that standardizes how applications provide context to Large Language Models (LLMs). It acts like a "USB-C port for AI applications" - providing a standardized way to connect AI models to different data sources and tools.

MCP follows a client-server architecture where:

• MCP Hosts: Programs like Claude Desktop or IDEs that want to access data through MCP
• MCP Clients: Protocol clients that maintain 1:1 connections with servers
• MCP Servers: Lightweight programs that expose specific capabilities through the standardized protocol
• Local Data Sources: Your computer's files, databases, and services that MCP servers can securely access
• Remote Services: External systems available over the internet that MCP servers can connect to

What is ZenML?

ZenML is an open-source platform for building and managing ML and AI pipelines. It provides a unified interface for managing data, models, and experiments.

For more information, see the ZenML website and our documentation.

Features

The server provides MCP tools to access core read functionality from the ZenML server, providing a way to get live information about:

Core Entities

• Users - user accounts and permissions
• Stacks - infrastructure configurations
• Stack Components - individual stack building blocks
• Flavors - available component types
• Service Connectors - cloud authentication

Pipeline Execution

• Pipelines - pipeline definitions
• Pipeline Runs - execution history and status
• Pipeline Steps - individual step details, code, and logs
• Schedules - automated run schedules
• Artifacts - metadata about data artifacts (not the data itself)

Deployment & Serving

• Snapshots - frozen pipeline configurations (the "what to run/serve" artifact)
• Deployments - runtime serving instances with status, URL, and logs
• Services - model serving endpoints

Organization & Discovery

• Projects - organizational containers for ZenML resources
• Tags - cross-cutting metadata labels for discovery
• Builds - pipeline build artifacts with image and code info

Models

• Models - ML model registry entries
• Model Versions - versioned model artifacts

Compatibility APIs (migration recommended)

• Pipeline run templates remain available in ZenML 0.96.4, while Snapshots are preferred for new workflows (see Migration Guide)

The server also allows you to trigger new pipeline runs using snapshots (preferred) or the deprecated template-based trigger parameter.

Note: We're continuously improving this integration based on user feedback. Please join our Slack community to share your experience and help us make it even better!

Tool profiles and write policy

The default compact profile advertises 16 tools. Seven generic tools cover the resource catalog, reads, ordinary mutations, and finite lifecycle actions:

Shortened. The full README is on GitHub.