MLflow 提示注册表 MCP 服务器

Model Context Protocol (MCP) 服务器为 MLflow Prompt Registry 实现，使用户能够访问在 MLflow 中管理的提示模板。

此服务器实现了 MCP Prompts 规范，用于发现和使用从 MLflow 提示注册表中托管的提示模板。主要用例是在 Claude Desktop 中加载 MLflow 中的提示模板，使用户能够方便地 instruct Claude 执行重复任务或常见工作流。

![./assets/mcp-mlflow-prompt-demo.png]

工具

• list-prompts
  • 列出可用的提示
  • 输入:
    • cursor (可选字符串): 分页光标
    • filter (可选字符串): 提示过滤器
  • 返回: 提示对象列表
• get-prompt
  • 获取并编译特定提示
  • 输入:
    • name (字符串): 要检索的提示名称
    • arguments (可选对象): 包含提示变量的 JSON 对象
  • 返回: 编译后的提示对象

设置

1: 安装 MLflow 并启动提示注册表

如果尚未安装，安装并启动一个 MLflow 服务器以托管提示注册表：

pip install mlflow>=2.21.1
mlflow server --port 5000

2: 在 MLflow 中创建提示模板

如果您还没有，请按照此指南在 MLflow 中创建一个提示模板。

3: 构建 MCP 服务器

npm install
npm run build

4: 将服务器添加到 Claude Desktop

通过编辑 claude_desktop_config.json 配置 Claude for Desktop：

{
  "mcpServers": {
    "mlflow": {
      "command": "node",
      "args": ["<absolute-path-to-this-repository>/dist/index.js"],
      "env": {
        "MLFLOW_TRACKING_URI": "http://localhost:5000"
      }
    }
  }
}

请确保将 MLFLOW_TRACKING_URI 替换为您的实际 MLflow 服务器地址。