MLflow Prompt Registry MCP Server

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

此服务器实现了MCP Prompts 规范，用于发现并使用来自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来配置Desktop版Claude：

{
  "mcpServers": {
    "mlflow": {
      "command": "node",
      "args": ["<绝对路径到此仓库>/dist/index.js"],
      "env": {
        "MLFLOW_TRACKING_URI": "http://localhost:5000"
      }
    }
  }
}

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