MLflow Prompt Registry MCP 服务器

用于 MLflow Prompt Registry 的模型上下文协议 (MCP) 服务器，支持访问在 MLflow 中管理的 prompt 模板。

该服务器实现了 MCP Prompts 规范，用于发现和使用来自 MLflow Prompt Registry 的 prompt 模板。 主要用例是从 MLflow 加载 prompt 模板到 Claude Desktop，允许用户方便地指示 Claude 执行重复性任务或常见工作流程。

工具

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

设置

1: 安装 MLflow 并启动 Prompt Registry

如果尚未安装并启动 MLflow 服务器来托管 Prompt Registry，请执行以下操作：

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

2: 在 MLflow 中创建 prompt 模板

如果尚未创建，请按照 本指南 在 MLflow 中创建一个 prompt 模板。

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