图像生成服务

基于MCP协议的图像生成服务,支持通过文本提示生成图像,并可配置分辨率、宽高比等参数。

  • by mikeyny
  • TypeScript
  • 本地部署
  • 120 星标
  • 更新于
在 GitHub 查看 mikeyny/ai-image-gen-mcp

使用说明

图像生成MCP服务器

一 MCP(模型上下文协议) 使用Replicate生成图像的服务器实现 black-forest-labs/flux-schnell 模型。

理想情况下,它与Cursor的MCP功能一起使用,但也可以与任何MCP客户端一起使用。

特性

  • 根据文本提示生成图像
  • 可配置的图像参数(分辨率、纵横比、质量)
  • 将生成的图像保存到指定目录
  • 完全符合MCP协议
  • 错误处理和验证

先决条件

  • Node.js 16+
  • 复制API令牌
  • MCP的TypeScript SDK

设置

  1. 克隆存储库

  2. 安装依赖项:

    npm install
    
  3. 在的代码中直接添加Replicate API令牌 src/imageService.ts 通过更新 apiToken 常数:

    // No environment variables are used since they can't be easily set in cursor
    const apiToken = "your-replicate-api-token-here";
    

    注: 如果与Claude一起使用,您可以创建 .env 文件,并在其中设置API令牌:

    REPLICATE_API_TOKEN=your-replicate-api-token-here
    

    然后构建项目:

    npm run build
    

用法

要与光标一起使用,请执行以下操作:

  1. 转到“设置”
  2. 选择特征
  3. 向下滚动到“MCP服务器”
  4. 点击“添加新的MCP服务器”
  5. 将类型设置为“命令”
  6. 将命令设置为: node ./path/to/dist/server.js

API参数

参数 类型 必填 默认 说明
prompt string 是 - 图像生成的文本提示
output_dir string 是 - 保存生成图像的服务器目录路径
go_fast boolean 否 false 启用更快的生成模式
megapixels string 否 “1” 分辨率质量(“1”、“2”、“4”)
num_outputs number 否 1 要生成的图像数量(1-4)
aspect_ratio string 否 “1:1” 纵横比(“1:1”、“4:3”、“16:9”)
output_format string 否 “webp” 图像格式(“webp”、“png”、“jpeg”)
output_quality number No 80 压缩质量(1-100)
num_inference_steps number No 4 去噪步骤数(4-20)

示例请求

{
  "prompt": "black forest gateau cake spelling out 'FLUX SCHNELL'",
  "output_dir": "/var/output/images",
  "filename": "black_forest_cake",
  "output_format": "webp"
  "go_fast": true,
  "megapixels": "1",
  "num_outputs": 2,
  "aspect_ratio": "1:1"
}

示例响应

{
  "image_paths": [
    "/var/output/images/output_0.webp",
    "/var/output/images/output_1.webp"
  ],
  "metadata": {
    "model": "black-forest-labs/flux-schnell",
    "inference_time_ms": 2847
  }
}

错误处理

服务器处理以下错误类型:

  • 验证错误(无效参数)
  • API错误(复制API问题)
  • 服务器错误(文件系统、权限)
  • 未知错误(意外问题)

每个错误响应包括:

  • 错误代码
  • 人类可读信息
  • 详细错误信息

许可证

国际协调委员会