图像生成服务

基于Together AI和Replicate的模型上下文协议(MCP)服务器,用于根据文本提示生成图像,支持本地运行或SSE端点部署。

  • by GMKR
  • TypeScript
  • 本地部署
  • 4 星标
  • 更新于
在 GitHub 查看 GMKR/mcp-imagegen

使用说明

MCP图像生成器

模型上下文协议(MCP)服务器,用于使用Together AI的图像生成模型生成图像。此MCP服务器可以在本地运行,也可以使用SSE端点运行。 MCP映像生成器需要提供程序,目前仅支持“复制”和“一起”。您需要设置 TOGETHER_API_KEY 或 REPLICATE_API_TOKEN 环境变量。并设置 PROVIDER 环境变量“复制”或“一起”/

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 可以是“复制”或“一起”。

使用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 (数字,可选):要生成的图像的宽度(默认值:512)
  • height (数字,可选):要生成的图像的高度(默认值:512)
  • numberOfImages (number,可选):要生成的图像数量(默认值:1)

环境变量

  • PROVIDER:用于图像生成的提供程序(默认值:“复制”)
  • REPLICATE_API_TOKEN:您的Replicate API令牌
  • TOGETHER_API_KEY:您的Together AI API密钥
  • MODEL_NAME:用于图像生成的模型(默认值:“黑森林实验室/flux schnell”)

许可证

麻省理工学院