Comfy MCP 服务器

一个使用 FastMCP 框架，通过远程 Comfy 服务器，基于提示词生成图像的服务器。

概述

此脚本使用 FastMCP 框架设置一个服务器，该服务器使用指定的工作流程，基于提示词生成图像。 它与远程 Comfy 服务器交互，以提交提示词并检索生成的图像。

前提条件

• 用于 Python 的 uv 包和项目管理器。
• 从 Comfy UI 导出的工作流程文件。 此代码包含一个示例 Flux-Dev-ComfyUI-Workflow.json，仅在此处用作参考。 您需要从您的工作流程导出，并相应地设置环境变量。

您可以安装本地开发所需的软件包：

uvx mcp[cli]

配置

设置以下环境变量：

• COMFY_URL 指向您的 Comfy 服务器 URL。
• COMFY_WORKFLOW_JSON_FILE 指向 comfyui 工作流程的 API 导出 json 文件的绝对路径。
• PROMPT_NODE_ID 指向文本提示节点 ID。
• OUTPUT_NODE_ID 指向包含最终图像的输出节点 ID。
• OUTPUT_MODE 设置为 url 或 file 以选择所需的输出。

（可选）如果您运行了 Ollama 服务器，您可以连接到它以生成提示词。

• OLLAMA_API_BASE 指向 ollama 运行的 URL。
• PROMPT_LLM 指向 ollama 上托管的用于生成提示词的模型名称。

示例：

export COMFY_URL=http://your-comfy-server-url:port
export COMFY_WORKFLOW_JSON_FILE=/path/to/the/comfyui_workflow_export.json
export PROMPT_NODE_ID=6 # 在这里使用正确的节点 ID
export OUTPUT_NODE_ID=9 # 在这里使用正确的节点 ID
export OUTPUT_MODE=file

用法

可以使用以下命令启动 Comfy MCP 服务器：

uvx comfy-mcp-server

Claude Desktop 配置示例

{
  "mcpServers": {
    "Comfy MCP Server": {
      "command": "/path/to/uvx",
      "args": [
        "comfy-mcp-server"
      ],
      "env": {
        "COMFY_URL": "http://your-comfy-server-url:port",
        "COMFY_WORKFLOW_JSON_FILE": "/path/to/the/comfyui_workflow_export.json",
        "PROMPT_NODE_ID": "6",
        "OUTPUT_NODE_ID": "9",
        "OUTPUT_MODE": "file",
      }
    }
  }
}

功能

generate_image(prompt: str, ctx: Context) -> Image | str

此函数使用指定的提示词生成图像。 它遵循以下步骤：

1. 检查是否设置了所有环境变量。
2. 从 JSON 文件加载提示词模板。
3. 将提示词提交到 Comfy 服务器。
4. 轮询服务器以获取提示词处理的状态。
5. 检索并返回生成的图像（一旦准备就绪）。

generate_prompt(topic: str, ctx: Context) -> str

此函数从指定的主题生成一个全面的图像生成提示词。

依赖项

• mcp: 用于设置 FastMCP 服务器。
• json: 用于处理 JSON 数据。
• urllib: 用于发出 HTTP 请求。
• time: 用于在轮询中添加延迟。
• os: 用于访问环境变量。
• langchain: 用于创建简单的 LLM Prompt 链，以从主题生成图像生成提示词。
• langchain-ollama: 用于 LangChain 的 ollama 特定模块。

许可证

此项目已获得 MIT 许可证的许可 - 有关详细信息，请参阅 LICENSE 文件。