README
🚀 MCP 图像生成器
MCP 图像生成器是一个借助 Together AI 图像生成模型的 Model Context Protocol (MCP) 服务器。它既可以在本地运行,也能通过 SSE 端点运行。目前,该生成器仅支持 "Replicate" 和 "Together" 这两种图像生成服务提供者。使用时,需要设置 TOGETHER_API_KEY 或 REPLICATE_API_TOKEN 环境变量,并将 PROVIDER 环境变量设为 "replicate" 或 "together"。
🚀 快速开始
✨ SSE 端点(Docker 环境)
克隆仓库
git clone https://github.com/gmkr/mcp-imagegen.git
cd mcp-imagegen
使用 Dockerfile 构建并运行容器
docker build -f Dockerfile.server -t mcp-imagegen .
docker run -p 3000:3000 mcp-imagegen
配置 MCP 客户端
{
"mcpServers": {
"imagegenerator": {
"url": "http://localhost:3000/sse",
"env": {
"PROVIDER": "replicate",
"REPLICATE_API_TOKEN": "your-replicate-api-token"
}
}
}
}
您可以根据实际情况,将 url 调整为您使用的 MCP 服务器端点。provider 可以选择 "replicate" 或 "together"。
💻 在本地使用标准输入输出
先决条件
- Node.js
- Together AI API 密钥或 Replicate API 令牌
安装
- 克隆仓库:
git clone https://github.com/gmkr/mcp-imagegen.git
cd mcp-imagegen
- 安装依赖项:
pnpm install
配置
创建一个配置文件用于您的 MCP 客户端。以下是示例配置:
{
"mcpServers": {
"imagegenerator": {
"command": "pnpx",
"args": [
"-y",
"tsx",
"/path/to/mcp-imagegen/src/index.ts"
],
"env": {
"PROVIDER": "replicate",
"REPLICATE_API_TOKEN": "your-replicate-api-token"
}
}
}
}
请将 /path/to/mcp-imagegen 替换为您克隆仓库的实际路径,并将 your-replicate-api-token 替换为您的实际 Replicate API 令牌。
💻 使用示例
基础用法
MCP 图像生成器提供了一个名为 generate_image 的工具,用于根据文本提示生成图像。
参数:
prompt(字符串):要生成图像的文本提示width(数字,可选):生成图像的宽度(默认:512)height(数字,可选):生成图像的高度(默认:512)numberOfImages(数字,可选):要生成的图像数量(默认:1)
🔧 技术细节
环境变量
| 属性 | 详情 |
|------|------|
| PROVIDER | 用于图像生成的服务提供商(默认:"replicate") |
| REPLICATE_API_TOKEN | 您的 Replicate API 令牌 |
| TOGETHER_API_KEY | 您的 Together AI API 密钥 |
| MODEL_NAME | 要使用的图像生成模型(默认:"black-forest-labs/flux-schnell") |
📄 许可证
本项目采用 MIT 许可证。
微信扫一扫