小贴士:按下Ctrl+D 或 ⌘+D,一键收藏本站,方便下次快速访问!
70
最近更新:26天前

使用户能够使用 Replicate 的模型从文本提示生成图像,支持配置参数并完全符合 MCP 协议。

图像生成 MCP 服务器

一个 MCP (Model Context Protocol) 服务器实现,使用 Replicate 的 black-forest-labs/flux-schnell 模型生成图像。

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

特性

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

前提条件

  • Node.js 16+
  • Replicate API 令牌
  • TypeScript SDK for MCP

设置

  1. 克隆仓库

  2. 安装依赖项:

    bash 复制代码
    npm install
  3. src/imageService.ts 中直接更新 apiToken 常量来添加你的 Replicate API 令牌:

    bash 复制代码
    // 由于在 cursor 中无法轻松设置环境变量,因此不使用环境变量
    const apiToken = "your-replicate-api-token-here";

    注意: 如果与 Claude 一起使用,可以在根目录中创建一个 .env 文件并设置你的 API 令牌:

    bash 复制代码
    REPLICATE_API_TOKEN=your-replicate-api-token-here

    然后构建项目:

    bash 复制代码
    npm run build

使用

与 cursor 一起使用:

  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 80 压缩质量 (1-100)
num_inference_steps number 4 去噪步骤数 (4-20)

示例请求

json 复制代码
{
  "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"
}

示例响应

json 复制代码
{
  "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 错误(Replicate API 问题)
  • 服务器错误(文件系统、权限)
  • 未知错误(意外问题)

每个错误响应包括:

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

许可

ISC