
Ai Image Gen Mcp
使用户能够使用 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
设置
-
克隆仓库
-
安装依赖项:
bashnpm install
-
在
src/imageService.ts
中直接更新apiToken
常量来添加你的 Replicate API 令牌:bash// 由于在 cursor 中无法轻松设置环境变量,因此不使用环境变量 const apiToken = "your-replicate-api-token-here";
注意: 如果与 Claude 一起使用,可以在根目录中创建一个
.env
文件并设置你的 API 令牌:bashREPLICATE_API_TOKEN=your-replicate-api-token-here
然后构建项目:
bashnpm run build
使用
与 cursor 一起使用:
- 进入设置
- 选择功能
- 向下滚动到“MCP 服务器”
- 点击“添加新的 MCP 服务器”
- 将类型设置为“命令”
- 将命令设置为:
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