mlflow for VS Code
io.github.us-all/mlflow
MLflow MCP — experiments, runs, registered models, versions, traces, assessments (MLflow 3)
client:VS Code
transport:stdio
runtime:npm
Install mlflow in VS Code
.vscode/mcp.json
{
"inputs": [
{
"type": "promptString",
"id": "mlflow-tracking-uri",
"description": "MLFLOW_TRACKING_URI",
"password": true
},
{
"type": "promptString",
"id": "mlflow-tracking-token",
"description": "MLFLOW_TRACKING_TOKEN",
"password": true
},
{
"type": "promptString",
"id": "mlflow-tracking-username",
"description": "MLFLOW_TRACKING_USERNAME",
"password": true
},
{
"type": "promptString",
"id": "mlflow-tracking-password",
"description": "MLFLOW_TRACKING_PASSWORD",
"password": true
},
{
"type": "promptString",
"id": "mlflow-experiment-id",
"description": "MLFLOW_EXPERIMENT_ID",
"password": true
},
{
"type": "promptString",
"id": "mlflow-tools",
"description": "MLFLOW_TOOLS",
"password": true
},
{
"type": "promptString",
"id": "mlflow-disable",
"description": "MLFLOW_DISABLE",
"password": true
},
{
"type": "promptString",
"id": "mlflow-allow-write",
"description": "MLFLOW_ALLOW_WRITE",
"password": true
}
],
"servers": {
"mlflow": {
"type": "stdio",
"command": "npx",
"args": [
"-y",
"@us-all/mlflow-mcp"
],
"env": {
"MLFLOW_TRACKING_URI": "${input:mlflow-tracking-uri}",
"MLFLOW_TRACKING_TOKEN": "${input:mlflow-tracking-token}",
"MLFLOW_TRACKING_USERNAME": "${input:mlflow-tracking-username}",
"MLFLOW_TRACKING_PASSWORD": "${input:mlflow-tracking-password}",
"MLFLOW_EXPERIMENT_ID": "${input:mlflow-experiment-id}",
"MLFLOW_TOOLS": "${input:mlflow-tools}",
"MLFLOW_DISABLE": "${input:mlflow-disable}",
"MLFLOW_ALLOW_WRITE": "${input:mlflow-allow-write}"
}
}
}
}VS Code prompts for each input the first time the server starts, so no secret is written to the file. VS Code docs