Skip to main content
Glama
kkruglik

MLflow MCP Server

by kkruglik

register_model

Registers a machine learning model into the MLflow model registry, creating the registered model if it does not exist.

Instructions

Register a model into the model registry. Creates the registered model if it doesn't exist.

Args: model_name: Name for the registered model. model_uri: URI of the model to register. Supports: - LoggedModel: 'models:/m-abc123' - Run artifact: 'runs:/run_id/artifact_path' tags: Optional dict of tags to set on the model version.

Examples: register_model("btc-classifier", "models:/m-abc123") register_model("btc-classifier", "runs:/abc123/model", tags={"framework": "lightgbm"})

Input Schema

TableJSON Schema
NameRequiredDescriptionDefault
model_nameYes
model_uriYes
tagsNo
Behavior4/5

Does the description disclose side effects, auth requirements, rate limits, or destructive behavior?

Annotations declare the tool is not read-only (readOnlyHint=false), consistent with the description's 'register' and 'creates' verbs. The description adds behavioral context by specifying URI formats (LoggedModel and Run artifact) and the optional tags, but does not clarify behavior if the model already exists.

Agents need to know what a tool does to the world before calling it. Descriptions should go beyond structured annotations to explain consequences.

Conciseness5/5

Is the description appropriately sized, front-loaded, and free of redundancy?

The description is well-structured: a concise opening sentence, clear Args section, and practical Examples. Every sentence adds value, and it is front-loaded with the main action.

Shorter descriptions cost fewer tokens and are easier for agents to parse. Every sentence should earn its place.

Completeness4/5

Given the tool's complexity, does the description cover enough for an agent to succeed on first attempt?

Given three parameters, no output schema, and sparse annotations, the description covers the core functionality and parameter details with examples. However, it omits potential error conditions or behavior on re-registration, which would strengthen completeness.

Complex tools with many parameters or behaviors need more documentation. Simple tools need less. This dimension scales expectations accordingly.

Parameters5/5

Does the description clarify parameter syntax, constraints, interactions, or defaults beyond what the schema provides?

With 0% schema description coverage, the description fully compensates. It explains model_name, model_uri with examples (including two URI schemes), and tags as an optional dict. This adds significant meaning beyond the schema's minimal type info.

Input schemas describe structure but not intent. Descriptions should explain non-obvious parameter relationships and valid value ranges.

Purpose5/5

Does the description clearly state what the tool does and how it differs from similar tools?

The description clearly states the tool registers a model into the registry and creates the registered model if it doesn't exist. This specific verb-resource combination distinguishes it from sibling tools like get_registered_model or delete_registered_model.

Agents choose between tools based on descriptions. A clear purpose with a specific verb and resource helps agents select the right tool.

Usage Guidelines3/5

Does the description explain when to use this tool, when not to, or what alternatives exist?

The description provides examples of usage with different URI types but does not explicitly state when to use this tool versus alternatives like copy_model_version or transition_model_version_stage. It implies usage when registering a model but lacks explicit context.

Agents often have multiple tools that could apply. Explicit usage guidance like "use X instead of Y when Z" prevents misuse.

Install Server

Other Tools

Latest Blog Posts

MCP directory API

We provide all the information about MCP servers via our MCP API.

curl -X GET 'https://glama.ai/api/mcp/v1/servers/kkruglik/mlflow-mcp'

If you have feedback or need assistance with the MCP directory API, please join our Discord server