MLflow 提示注册表 MCP 服务器

Model Context Protocol (MCP) 服务器为 MLflow Prompt Registry，实现对 MLflow 中管理的提示模板的访问。

此服务器实现了 MCP Prompts 规范 以发现和使用 MLflow 中托管的提示模板。主要用例是将提示模板从 MLflow 加载到 Claude Desktop，使用户能够方便地指示 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 服务器地址。