ComfyUI MCP Server

ComfyUI MCP Server

site icon
2025.04.17 0
Python图像生成AI 工具集成内容生成
ComfyUI MCP Server 是一个为 ComfyUI 实现的完整 Model Context Protocol (MCP) 服务,旨在为 AI 代理提供标准化的 API 来生成和操作图像。它通过丰富的元数据和模式信息,使 AI 助手能够轻松地通过 ComfyUI 使用 Stable Diffusion 模型进行图像生成和操作。
View on GitHub

Overview

基本能力

产品定位

ComfyUI MCP Server 是一个为 ComfyUI 实现的 MCP 协议服务,旨在为 AI 代理提供标准化的 API 来生成和操作图像。

核心功能

  • 完整的 MCP 协议实现:包括工具发现、会话和上下文管理、实时通信等。
  • 丰富的图像生成能力:支持文本到图像(txt2img)、图像到图像转换(img2img)、修复(inpainting)、ControlNet 支持和 LoRA 模型集成。
  • 高级基础设施:包括优先级支持的作业队列、进度跟踪和实时更新、认证和速率限制、会话持久化以及作业历史的数据库后端。

适用场景

  • AI 代理通过标准化 API 生成和操作图像。
  • 需要实时图像生成和转换的应用场景。
  • 需要高级图像处理功能(如修复、ControlNet 等)的项目。

工具列表

  • generate_image:从文本生成图像。
  • img2img:基于文本转换图像。
  • get_job_status:检查作业状态。
  • list_models:列出可用模型。
  • get_tools_manifest:获取完整的工具清单。

常见问题解答

  • 连接问题:确保 ComfyUI 正在运行并可通过配置的 URL 访问。
  • 图像生成问题:检查 ComfyUI 是否安装了所需的模型,并更新工作流 JSON 以匹配 ComfyUI 安装。
  • 超时问题:增加 advanced_comfyui_client.py 中的超时值。

使用教程

使用依赖

  • Python 3.10+
  • 已安装并运行的 ComfyUI
  • 推荐 4GB+ RAM
  • 推荐 CUDA 兼容的 GPU(非必需)

安装教程

选项 1:克隆并安装

git clone https://github.com/yourusername/comfyui-mcp-server.git
cd comfyui-mcp-server
pip install -r requirements.txt
python install.py --create-venv

选项 2:手动设置

mkdir comfyui-mcp-server
cd comfyui-mcp-server
pip install websockets requests aiohttp aiohttp_cors aiohttp_session cryptography aiosqlite mcp

调试方式

  1. 启动 ComfyUI:
cd /path/to/ComfyUI
python main.py --port 8188
  1. 启动 MCP 服务器:
cd /path/to/comfyui-mcp-server
python mcp_integration.py
  1. 测试连接:
python mcp_client.py --action manifest
python mcp_client.py --action generate --prompt "a dog wearing sunglasses"

许可证

该项目遵循 Apache-2.0 开源许可条款,请参阅 Apache-2.0 了解完整条款。