Image Generation (Flux Schnell)

🚀 图像生成 MCP 服务器

本项目是一个 MCP (Model Context Protocol) 服务器实现，可借助 Replicate 的 black-forest-labs/flux-schnell 模型生成图像。它既适合与 Cursor 的 MCP 功能搭配使用，也能和其他 MCP 客户端兼容。

🚀 快速开始

先决条件

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

设置

1. 克隆仓库。

2. 安装依赖：

   npm install
   

3. 将你的 Replicate API 令牌直接在 src/imageService.ts 中更新 apiToken 常量：

   // 在这里直接添加您的 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. 设置类型为 "Command"。
6. 设置命令为：node ./path/to/dist/server.js。

✨ 主要特性

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

📚 详细文档

API 参数

示例请求

{
  "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"
}

示例响应

{
  "status": "success",
  "message": "图像已生成并保存到指定目录。",
  "data": {
    "filename": "black_forest_cake.png",
    "size": "2.5 MB",
    "dimensions": "1920x1080"
  }
}

错误处理

• 输入错误：如果提示或参数无效，将返回验证错误。
• API 错误：如果调用 Replicate API 失败，将返回相关错误信息。
• 保存错误：如果无法将图像保存到指定目录，将返回存储错误。

📄 许可证

该代码受 MIT License 保护。