MCP Server Mlflow

🚀 MLflow 提示注册表 MCP 服务器

本项目是为 MLflow 提示注册表 实现的 Model Context Protocol (MCP) 服务器，它能让用户访问在 MLflow 中管理的提示模板。通过实现 MCP Prompts specification，该服务器可用于发现和使用从 MLflow 中托管的提示模板。其主要应用场景是将提示模板加载到 Claude Desktop 中，方便用户指示 Claude 处理重复任务或常见工作流。

🚀 快速开始

安装配置

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

✨ 主要特性

工具

list-prompts

• 功能：列出可用提示
• 输入参数：
  • cursor（可选字符串）：用于分页的游标
  • filter（可选字符串）：提示筛选条件
• 返回结果：提示对象列表

get-prompt

• 功能：获取并编译特定提示
• 输入参数：
  • name（字符串）：要获取的提示名称
  • arguments（可选对象）：包含提示变量的 JSON 对象
• 返回结果：已编译的提示对象