us-all/mlflow-mcp-server
🔌 MCP Serverus-all
将 AI Agent 连接至 MLflow,实现实验追踪、模型管理及 GenAI 链路分析的无缝集成。
mlflow-mcp-server 是一个全面的集成层,旨在将 MLflow 强大的 MLOps 能力直接引入 AI Agent 的上下文环境中。通过实现模型上下文协议 (MCP),它允许基于 LLM 的 Agent 在不离开原生环境的情况下,以编程方式查询和操作 MLflow 实体。主要功能包括对实验和运行记录的完整 CRUD 支持、与注册模型的深度集成,以及对 MLflow 3 GenAI 链路追踪的高级支持。该服务器内置了 5 个工作流提示词,如“分析失败链路”、“提升最优运行”和“对比顶级运行”等,可实现复杂 MLOps 任务的自动化。对于希望构建能够监控训练流水线、评估模型性能并通过自然语言指令管理部署生命周期的自主 Agent 的团队来说,该工具至关重要。
💡核心亮点
- ├─提供 82 种 MLflow 管理工具
- ├─全面支持 MLflow 3 GenAI 链路追踪
- └─内置自动化工作流提示词
🎯适用人群
- ├─MLOps 工程师
- └─AI Agent 开发者