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 對象
• 返回結果：已編譯的提示對象