图像生成MCP服务器

这是一个使用Replicate的black-forest-labs/flux-schnell模型来生成图像的MCP（模型上下文协议）服务器实现。

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

功能

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

预备条件

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

设置

1. 克隆仓库

2. 安装依赖项：

   npm install
   

3. 在代码中的src/imageService.ts文件中更新apiToken常量，直接添加您的Replicate API令牌：

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

   注意： 如果与Claude一起使用，您可以在根目录创建一个.env文件，并在那里设置API令牌：

   REPLICATE_API_TOKEN=your-replicate-api-token-here
   

   然后构建项目：

   npm run build
   

使用方法

要与cursor一起使用：

1. 前往设置
2. 选择功能
3. 向下滚动至“MCP服务器”
4. 点击“添加新MCP服务器”
5. 将类型设为“命令”
6. 将命令设为：node ./path/to/dist/server.js

API参数

参数	类型	必需	默认值	描述
prompt	字符串	是	-	图像生成的文本提示
output_dir	字符串	是	-	用于保存生成图像的服务器目录路径
go_fast	布尔值	否	false	启用快速生成模式
megapixels	字符串	否	"1"	分辨率质量 ("1", "2", "4")
num_outputs	数字	否	1	要生成的图像数量 (1-4)
aspect_ratio	字符串	否	"1:1"	宽高比 ("1:1", "4:3", "16:9")
output_format	字符串	否	"webp"	图像格式 ("webp", "png", "jpeg")
output_quality	数字	否	80	压缩质量 (1-100)
num_inference_steps	数字	否	4	去噪步骤的数量 (4-20)

示例请求

{
  "prompt": "黑森林蛋糕拼出'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错误（Replicate API问题）
• 服务器错误（文件系统、权限）
• 未知错误（意外问题）

每个错误响应包括：

• 错误代码
• 易于理解的消息
• 详细的错误信息

许可证

ISC