舒适的MCP服务器

使用FastMCP框架通过远程Comfy服务器根据提示生成图像的服务器。

概述

此脚本使用FastMCP框架设置一个服务器，该服务器根据提示并通过指定的工作流程与远程Comfy服务器交互来生成图像。

先决条件

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

您可以使用以下命令安装本地开发所需的包：

uvx mcp[cli]

配置

设置以下环境变量：

• 将COMFY_URL设置为指向您的Comfy服务器URL。
• 将COMFY_WORKFLOW_JSON_FILE设置为Comfy UI工作流导出的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提示链，以从主题生成图像生成提示。
• langchain-ollama：用于与Ollama集成。

许可证

此项目受MIT许可证的保护。