Mlflow Prompt Registry

🚀 MLflow 提示注册表 MCP 服务器

MLflow 提示注册表 MCP 服务器为用户提供了访问 MLflow 中管理的提示模板的途径，解决了用户在执行重复任务或常见工作流时，能够更便捷地使用提示模板的问题，提升了工作效率。

此服务器实现了 MCP Prompts 规范，可用于发现和使用从 MLflow 提示注册表中托管的提示模板。其主要用例是在 Claude Desktop 中加载 MLflow 中的提示模板，方便用户 instruct Claude 执行相关任务。

![./assets/mcp-mlflow-prompt-demo.png]

🚀 快速开始

✨ 主要特性

• 实现了 MCP Prompts 规范，可用于发现和使用 MLflow 提示注册表中的提示模板。
• 支持在 Claude Desktop 中加载 MLflow 中的提示模板，便于用户 instruct 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 对象
返回: 编译后的提示对象