AI导哈

图像生成工具

一个基于ComfyUI工作流的图像生成工具，支持base64输出，适用于AI助手的图像生成需求。

  • by idans320
  • TypeScript
  • 本地部署
  • 更新于
在 GitHub 查看 idans320/comfyicu-mcp-server

使用说明

ComfyICU MCP工具

一种模型上下文协议（MCP）工具，用于通过具有base64输出支持的ComfyUI工作流生成图像。

概述

此MCP工具与ComfyICU服务无缝集成，使AI助手能够直接从文本提示中生成图像。该工具处理作业轮询、base64转换和优雅的错误处理。

特性

  • 直接积分法：可与任何兼容MCP的AI助手配合使用
  • Base64输出：将图像作为base64数据返回以进行内联显示
  • 迭代精化：支持根据以前的尝试迅速改进
  • 稳健的错误处理：全面的状态报告和超时管理
  • 尺寸优化：自动检查大小，并回退到URL传递

安装

npm install mcp-framework zod

配置

确保您的ComfyICU服务正在运行且可访问。该工具需要一个 ComfyICURunner 具有以下接口的服务：

interface ComfyICURunner {
  startImageGeneration(prompt: string): Promise<{id: string}>;
  getJobStatus(id: string): Promise<{status: string, output?: string}>;
}

用法

基础世代

await generateImage({
  prompt: "a minimalist logo design with geometric shapes"
});

带有精炼说明

await generateImage({
  prompt: "a minimalist logo design with geometric shapes",
  refinement_notes: "increase contrast, use blue and white color scheme"
});

api参考

generateImage(input: GenerateImageInput)

参数：

  • prompt （字符串，必填）：图像生成的文本描述
  • refinement_notes （字符串，可选）：精炼输出的附加指导

退货：

  • 成功：图像为具有mime类型的base64数据
  • 回退：如果base64转换失败，则返回图像URL
  • 错误：带有状态的详细错误消息

技术细节

超时处理

  • 默认超时：6分钟
  • 轮询间隔：5秒
  • 状态监测：持续到完成或超时

图像处理

  • 大小限制：1MB用于base64转换（与Claude Desktop兼容）
  • 回退策略：超大图像的URL传递
  • 格式支持：自动mime类型检测

错误状态

  • COMPLETED：成功生成图像数据
  • ERROR：带有诊断信息的发电故障
  • TIMEOUT：进程超过了最大持续时间

依赖项

  • mcp-framework：核心MCP功能
  • zod：架构验证
  • ComfyICURunner：服务集成层

许可证

MIT许可证-请参阅 许可证 文件以获取详细信息。

贡献

快速开始

  1. 分叉回购
  2. 创建要素分支： git checkout -b feature/your-thing
  3. 进行更改
  4. 本地测试
  5. 提交PR

指南

  • 代码的风格：遵循现有模式，正确使用TypeScript
  • 承诺：清晰、描述性的信息
  • 测试：在提交之前测试您的更改
  • 问题：使用GitHub issues处理bug和功能请求

我们接受什么

  • 错误修正
  • 性能改进
  • 文档更新
  • 新功能（先在问题中讨论）

我们不接受的

  • 不经讨论就打破变化
  • 与现有样式不匹配的代码
  • 没有明确用例的功能

保持简单，测试你的代码，写清晰的提交消息。就这样

相关 MCP