MLflow Prompt Registry MCP Server

MLflow Prompt Registry MCP 服务

用于 MLflow Prompt Registry 的模型上下文协议（MCP）服务，可访问由 MLflow 管理的提示词模板。

此服务实现了 MCP 提示词规范，用于发现和使用来自 MLflow Prompt Registry 的提示词模板。其主要用途是在 Claude Desktop 中加载 MLflow 的提示词模板，让用户可以方便地指示 Claude 执行重复性任务或常见工作流。

工具

• list-prompts
  • 列出可用的提示词
  • 输入：
    • cursor（可选字符串）：用于分页的游标
    • filter（可选字符串）：提示词筛选条件
  • 返回：提示词对象列表
• get-prompt
  • 检索并编译指定的提示词
  • 输入：
    • name（字符串）：要检索的提示词名称
    • arguments（可选对象）：包含提示词变量的 JSON 对象
  • 返回：编译后的提示词对象

设置

1：安装 MLflow 并启动 Prompt Registry

如果尚未安装并启动 MLflow 服务，请先安装并启动，以托管 Prompt Registry：

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 服务地址。