ZenML MCP Server

Name: ZenML MCP Server
Author: ZenML

ZenMLOfficial

Get started Repository Website & docs

Connects MCP clients to a ZenML deployment to read pipeline, run and stack data and trigger new pipeline runs.

Setup

Choose your client, copy the config, then fill in any required environment variables.

Add to Cursor Add to VS Code

claude mcp add zenml --env ZENML_STORE_URL='<ZENML_STORE_URL>' --env ZENML_STORE_API_KEY='<ZENML_STORE_API_KEY>' --env LOGLEVEL='<LOGLEVEL>' --env NO_COLOR='<NO_COLOR>' --env PYTHONIOENCODING='<PYTHONIOENCODING>' --env PYTHONUNBUFFERED='<PYTHONUNBUFFERED>' -- docker run -i --rm -e ZENML_STORE_URL -e ZENML_STORE_API_KEY docker.io/zenmldocker/mcp-zenml:2.0.0

~/.codex/config.toml

[mcp_servers.zenml]
command = "docker"
args = ["run", "-i", "--rm", "-e", "ZENML_STORE_URL", "-e", "ZENML_STORE_API_KEY", "docker.io/zenmldocker/mcp-zenml:2.0.0"]
env = { ZENML_STORE_URL = "<ZENML_STORE_URL>", ZENML_STORE_API_KEY = "<ZENML_STORE_API_KEY>", LOGLEVEL = "<LOGLEVEL>", NO_COLOR = "<NO_COLOR>", PYTHONIOENCODING = "<PYTHONIOENCODING>", PYTHONUNBUFFERED = "<PYTHONUNBUFFERED>" }

# codex mcp add zenml --env ZENML_STORE_URL='<ZENML_STORE_URL>' --env ZENML_STORE_API_KEY='<ZENML_STORE_API_KEY>' --env LOGLEVEL='<LOGLEVEL>' --env NO_COLOR='<NO_COLOR>' --env PYTHONIOENCODING='<PYTHONIOENCODING>' --env PYTHONUNBUFFERED='<PYTHONUNBUFFERED>' -- docker run -i --rm -e ZENML_STORE_URL -e ZENML_STORE_API_KEY docker.io/zenmldocker/mcp-zenml:2.0.0

~/.cursor/mcp.json

{
  "mcpServers": {
    "zenml": {
      "command": "docker",
      "args": [
        "run",
        "-i",
        "--rm",
        "-e",
        "ZENML_STORE_URL",
        "-e",
        "ZENML_STORE_API_KEY",
        "docker.io/zenmldocker/mcp-zenml:2.0.0"
      ],
      "env": {
        "ZENML_STORE_URL": "<ZENML_STORE_URL>",
        "ZENML_STORE_API_KEY": "<ZENML_STORE_API_KEY>",
        "LOGLEVEL": "<LOGLEVEL>",
        "NO_COLOR": "<NO_COLOR>",
        "PYTHONIOENCODING": "<PYTHONIOENCODING>",
        "PYTHONUNBUFFERED": "<PYTHONUNBUFFERED>"
      }
    }
  }
}

.vscode/mcp.json

{
  "servers": {
    "zenml": {
      "type": "stdio",
      "command": "docker",
      "args": [
        "run",
        "-i",
        "--rm",
        "-e",
        "ZENML_STORE_URL",
        "-e",
        "ZENML_STORE_API_KEY",
        "docker.io/zenmldocker/mcp-zenml:2.0.0"
      ],
      "env": {
        "ZENML_STORE_URL": "<ZENML_STORE_URL>",
        "ZENML_STORE_API_KEY": "<ZENML_STORE_API_KEY>",
        "LOGLEVEL": "<LOGLEVEL>",
        "NO_COLOR": "<NO_COLOR>",
        "PYTHONIOENCODING": "<PYTHONIOENCODING>",
        "PYTHONUNBUFFERED": "<PYTHONUNBUFFERED>"
      }
    }
  }
}

~/.gemini/settings.json

{
  "mcpServers": {
    "zenml": {
      "command": "docker",
      "args": [
        "run",
        "-i",
        "--rm",
        "-e",
        "ZENML_STORE_URL",
        "-e",
        "ZENML_STORE_API_KEY",
        "docker.io/zenmldocker/mcp-zenml:2.0.0"
      ],
      "env": {
        "ZENML_STORE_URL": "<ZENML_STORE_URL>",
        "ZENML_STORE_API_KEY": "<ZENML_STORE_API_KEY>",
        "LOGLEVEL": "<LOGLEVEL>",
        "NO_COLOR": "<NO_COLOR>",
        "PYTHONIOENCODING": "<PYTHONIOENCODING>",
        "PYTHONUNBUFFERED": "<PYTHONUNBUFFERED>"
      }
    }
  }
}

claude_desktop_config.json

{
  "mcpServers": {
    "zenml": {
      "command": "docker",
      "args": [
        "run",
        "-i",
        "--rm",
        "-e",
        "ZENML_STORE_URL",
        "-e",
        "ZENML_STORE_API_KEY",
        "docker.io/zenmldocker/mcp-zenml:2.0.0"
      ],
      "env": {
        "ZENML_STORE_URL": "<ZENML_STORE_URL>",
        "ZENML_STORE_API_KEY": "<ZENML_STORE_API_KEY>",
        "LOGLEVEL": "<LOGLEVEL>",
        "NO_COLOR": "<NO_COLOR>",
        "PYTHONIOENCODING": "<PYTHONIOENCODING>",
        "PYTHONUNBUFFERED": "<PYTHONUNBUFFERED>"
      }
    }
  }
}

.trae/mcp.json

{
  "mcpServers": {
    "zenml": {
      "command": "docker",
      "args": [
        "run",
        "-i",
        "--rm",
        "-e",
        "ZENML_STORE_URL",
        "-e",
        "ZENML_STORE_API_KEY",
        "docker.io/zenmldocker/mcp-zenml:2.0.0"
      ],
      "env": {
        "ZENML_STORE_URL": "<ZENML_STORE_URL>",
        "ZENML_STORE_API_KEY": "<ZENML_STORE_API_KEY>",
        "LOGLEVEL": "<LOGLEVEL>",
        "NO_COLOR": "<NO_COLOR>",
        "PYTHONIOENCODING": "<PYTHONIOENCODING>",
        "PYTHONUNBUFFERED": "<PYTHONUNBUFFERED>"
      }
    }
  }
}

Environment variables

ZENML_STORE_URL
ZENML_STORE_API_KEY
LOGLEVEL
NO_COLOR
PYTHONIOENCODING
PYTHONUNBUFFERED

Replace the placeholders with your own credentials and keep them out of version control.

Setup

Environment variables

Related