us-all avatar

Mlflow

us-all/mlflow-mcp-server

Connects Claude to MLflow tracking servers (OSS or Databricks) with 82 tools spanning experiments, runs, model registry, MLflow 3 logged models, traces, assessments, webhooks, and prompt optimization jobs. The big win is aggregation tools like summarize-experiment that fold metric stats and top runs into one response instead of five round trips, plus extractFields projection to cut token overhead. Ships with four workflow prompts (debug failed traces, promote best run, compare top runs) and six resource URIs for inline fetching. Use MLFLOW_TOOLS to scope down to just the categories you need. Defaults to read only mode until you flip MLFLOW_ALLOW_WRITE. If you're doing MLflow 3 workflows or need Databricks trace attachments, this is the only third party MCP that ships them.

GitHub stars
1
tools
80
auth
Secret needed
transport
STDIO
env vars
8
last updated
Jun 19, 2026

Tools

80 tools

Public tool metadata for what this MCP can expose to an agent.

create-experiment

No parameter schema in public metadata yet.

search-experiments

No parameter schema in public metadata yet.

get-experiment

No parameter schema in public metadata yet.

get-experiment-by-name

No parameter schema in public metadata yet.

update-experiment

No parameter schema in public metadata yet.

delete-experiment

No parameter schema in public metadata yet.

restore-experiment

No parameter schema in public metadata yet.

set-experiment-tag

No parameter schema in public metadata yet.

delete-experiment-tag

No parameter schema in public metadata yet.

create-run

No parameter schema in public metadata yet.

get-run

No parameter schema in public metadata yet.

search-runs

No parameter schema in public metadata yet.

update-run

No parameter schema in public metadata yet.

delete-run

No parameter schema in public metadata yet.

restore-run

No parameter schema in public metadata yet.

log-metric

No parameter schema in public metadata yet.

log-param

No parameter schema in public metadata yet.

log-batch

No parameter schema in public metadata yet.

log-inputs

No parameter schema in public metadata yet.

get-metric-history

No parameter schema in public metadata yet.

set-run-tag

No parameter schema in public metadata yet.

delete-run-tag

No parameter schema in public metadata yet.

list-artifacts

No parameter schema in public metadata yet.

get-best-run

No parameter schema in public metadata yet.

compare-runs

No parameter schema in public metadata yet.

search-runs-by-tags

No parameter schema in public metadata yet.

summarize-runAggregation tool that folds several round-trips into one structured response.

Aggregation tool that folds several round-trips into one structured response.

No parameter schema in public metadata yet.

create-registered-model

No parameter schema in public metadata yet.

get-registered-model

No parameter schema in public metadata yet.

search-registered-models

No parameter schema in public metadata yet.

rename-registered-model

No parameter schema in public metadata yet.

update-registered-model

No parameter schema in public metadata yet.

delete-registered-model

No parameter schema in public metadata yet.

get-latest-model-versions

No parameter schema in public metadata yet.

set-registered-model-tag

No parameter schema in public metadata yet.

delete-registered-model-tag

No parameter schema in public metadata yet.

set-registered-model-alias

No parameter schema in public metadata yet.

delete-registered-model-alias

No parameter schema in public metadata yet.

get-model-version-by-alias

No parameter schema in public metadata yet.

create-model-version

No parameter schema in public metadata yet.

get-model-version

No parameter schema in public metadata yet.

search-model-versions

No parameter schema in public metadata yet.

update-model-version

No parameter schema in public metadata yet.

delete-model-version

No parameter schema in public metadata yet.

transition-model-version-stage

No parameter schema in public metadata yet.

get-model-version-download-uri

No parameter schema in public metadata yet.

set-model-version-tag

No parameter schema in public metadata yet.

delete-model-version-tag

No parameter schema in public metadata yet.

create-logged-model

No parameter schema in public metadata yet.

search-logged-models

No parameter schema in public metadata yet.

get-logged-model

No parameter schema in public metadata yet.

finalize-logged-model

No parameter schema in public metadata yet.

delete-logged-model

No parameter schema in public metadata yet.

set-logged-model-tags

No parameter schema in public metadata yet.

delete-logged-model-tag

No parameter schema in public metadata yet.

log-logged-model-params

No parameter schema in public metadata yet.

search-traces

No parameter schema in public metadata yet.

get-trace

No parameter schema in public metadata yet.

get-trace-info

No parameter schema in public metadata yet.

delete-traces

No parameter schema in public metadata yet.

set-trace-tag

No parameter schema in public metadata yet.

delete-trace-tag

No parameter schema in public metadata yet.

list-trace-attachments

No parameter schema in public metadata yet.

get-trace-attachment

No parameter schema in public metadata yet.

log-feedback

No parameter schema in public metadata yet.

log-expectation

No parameter schema in public metadata yet.

get-assessment

No parameter schema in public metadata yet.

update-assessment

No parameter schema in public metadata yet.

delete-assessment

No parameter schema in public metadata yet.

create-webhook

No parameter schema in public metadata yet.

list-webhooks

No parameter schema in public metadata yet.

get-webhook

No parameter schema in public metadata yet.

update-webhook

No parameter schema in public metadata yet.

delete-webhook

No parameter schema in public metadata yet.

test-webhook

No parameter schema in public metadata yet.

create-prompt-optimization-job

No parameter schema in public metadata yet.

get-prompt-optimization-job

No parameter schema in public metadata yet.

search-prompt-optimization-jobs

No parameter schema in public metadata yet.

cancel-prompt-optimization-job

No parameter schema in public metadata yet.

delete-prompt-optimization-job

No parameter schema in public metadata yet.

Sponsors

Configuration

MLFLOW_TRACKING_URI*

MLflow tracking server URI (e.g. http://localhost:5050, https://mlflow.example.com).

MLFLOW_TRACKING_TOKENsecret

Bearer token for authenticated MLflow servers (e.g. Databricks).

MLFLOW_TRACKING_USERNAME

Basic-auth username (alternative to token).

MLFLOW_TRACKING_PASSWORDsecret

Basic-auth password (alternative to token).

MLFLOW_EXPERIMENT_ID

Default experiment ID for run-creation tools.

MLFLOW_TOOLS

Comma-separated category allowlist. Default: all categories enabled.

MLFLOW_DISABLE

Comma-separated category disablelist.

MLFLOW_ALLOW_WRITEdefault: false

Set to 'true' to enable write/destructive tools. Default read-only.

Readme

MLflow MCP Server

The widest-coverage MLflow MCP — including MLflow 3 traces, prompt-optimization, webhooks, and Databricks trace attachments that no other MCP exposes.

82 tools across experiments, runs, registry, logged models, traces, assessments, webhooks, prompt-optimization. Aggregation tools (summarize-experiment, summarize-run) fold 3–5 round-trips into one structured response with already-fetched metric stats.

npm downloads tools @us-all standard Glama MCP server

What it does that others don't


View the full README on GitHub

Details

Repository
github.com/us-all/mlflow-mcp-server
Package
@us-all/mlflow-mcp
Transport
STDIO
Auth
Required
Registry
MCP registry, active

More from us-all