MCP 图像生成器

一个模型上下文协议 (MCP) 服务器，用于使用 Together AI 的图像生成模型生成图像。此 MCP 服务器可以在本地运行，也可以使用 SSE 端点运行。MCP 图像生成器需要一个提供商，目前仅支持“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

构建并运行 Docker 容器

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"。

在本地使用 stdio 运行

前提条件

• Node.js
• Together AI API 密钥或 Replicate API 令牌

安装

1. 克隆仓库:

   git clone https://github.com/gmkr/mcp-imagegen.git
   cd mcp-imagegen
   

2. 安装依赖:

   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 的工具，可用于根据文本提示生成图像。

工具: generate_image

根据提供的提示生成图像。

参数:

• prompt (string): 用于生成图像的文本提示
• width (number, optional): 要生成的图像的宽度 (默认: 512)
• height (number, optional): 要生成的图像的高度 (默认: 512)
• numberOfImages (number, optional): 要生成的图像的数量 (默认: 1)

环境变量

• PROVIDER: 用于图像生成的提供商 (默认: "replicate")
• REPLICATE_API_TOKEN: 您的 Replicate API 令牌
• TOGETHER_API_KEY: 您的 Together AI API 密钥
• MODEL_NAME: 用于图像生成的模型 (默认: "black-forest-labs/flux-schnell")

许可证

MIT