Skip to main content
stateset-agents mcp exposes the improvement loop as MCP tools, so an agent can run it directly instead of shelling out to the CLI. Every tool is a thin wrapper over the same module functions the CLI calls — grading, curation and preset resolution are not reimplemented in the server, so output is comparable either way.

Install and run

The server is an optional extra rather than a core dependency:
Without the extra, the command exits with the install hint rather than a traceback. stdio is the only transport in v1.
Or as client configuration directly:

The tools

No tool in v1 starts real training. dry_run_finetune only ever runs the preset builders in-process: no model download, GPU, or training launch. For actual execution, an authorized agent can separately use the fine-tuning JSON CLI to prepare a private snapshot and launch an exact reviewed plan. MCP tool access does not grant that authority.
improve_run calls the same orchestration function as stateset-agents improve run, so a curated set produced through MCP is byte-for-byte comparable with one produced from the shell. That matters when an agent curates a set and a person needs to check it.

A loop through the tools

What an agent driving this actually does, in order:
The MCP workflow stops at preview. Continue through a separately authorized CLI launch or hand the prepared run to a person in the terminal studio. Both use the same retained run/checksum rather than rebuilding the request after a disconnect.

Verifying it works

Use an MCP client to complete initialization and call tools/list, then read list_rewards or list_model_presets. A visible catalog proves discovery, not successful curation or training. Inspect returned artifacts and controlled errors from the actual tool. The reviewed source registers seven tools; discover the deployed catalog instead of treating that count as a capability guarantee. This source check was performed October 20, 2026 against stateset_agents/mcp_server.py.

Next steps

Agents quickstart

The same loop from the command line, including the training step MCP will not run.

CLI reference

What each wrapped command does natively.

MCP servers

The other servers StateSet publishes.

Agent Gate

Requiring an authorization proof before any MCP tool call takes effect.
Last modified on October 20, 2026