MLflow Prompt Registry MCP 服务器

MLflow Prompt Registry MCP 服务器

用于访问存储在 MLflow Prompt Registry 中的提示模板的 MCP 服务器。

快速入门

1. 安装 MLflow 并启动 Prompt Registry

安装并运行 MLflow 服务:

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

2. 创建 MLflow 提示模板

按照 官方指南 创建提示模板。

3. 构建 MCP 服务器

执行以下命令构建项目:

npm install
npm run build

4. 配置 Claude Desktop

编辑 claude_desktop_config.json 文件,添加以下内容:

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

MLFLOW_TRACKING_URI 替换为实际的 MLflow 地址。