图像生成服务器（imagegen-go） MCP 服务

这是一个基于 MCP (模型控制协议) 协议的图像生成服务器，使用 OpenAI 的 DALL-E API 生成图像。

环境要求

• Go 1.23.5 或更高版本
• OpenAI API 密钥

设置环境变量

在使用前，请设置 OpenAI API 密钥：

export OPENAI_API_KEY="你的OpenAI API密钥"

编译

如果需要重新编译项目：

go build -o bin/imagegen-go main/*.go

测试脚本

本项目提供了多个测试脚本，用于调试 MCP 服务器：

1. 基本测试

运行基本测试，包括初始化、获取工具列表和生成图像：

./test_mcp.sh

2. 详细调试

运行详细调试，将日志和响应保存到文件：

./debug_mcp.sh

3. 交互式测试

运行交互式测试，允许输入自定义的提示词：

./interactive_test.sh

4. 单独请求测试

测试单个请求：

# 初始化请求
./test_single_request.sh init

# 获取工具列表
./test_single_request.sh list

# 生成图像
./test_single_request.sh generate "一只可爱的猫咪" 1024 1024

日志

所有测试脚本都会将日志保存到 logs 目录中，方便查看和分析。

MCP 协议

MCP 是一种基于 JSON-RPC 2.0 的协议，用于模型与工具之间的通信。本服务器实现了以下 MCP 方法：

• initialize - 初始化服务器
• tools/list - 获取可用工具列表
• tools/call - 调用工具（生成图像）
• resources/list - 获取可用资源列表
• prompts/list - 获取可用提示列表

生成图像工具

本服务器提供了一个名为 generate-image 的工具，用于生成图像。参数如下：

• prompt (必需) - 图像描述
• width (可选) - 图像宽度，默认为 1792
• height (可选) - 图像高度，默认为 1024
• destination (可选) - 保存图像的路径，默认为用户的 Downloads 目录

调试步骤

1. 设置环境变量：

   export OPENAI_API_KEY="你的OpenAI API密钥"
   

2. 运行初始化请求：

   ./test_single_request.sh init
   

3. 获取工具列表：

   ./test_single_request.sh list
   

4. 生成图像：

   ./test_single_request.sh generate "你的提示词" [宽度] [高度]
   

5. 查看日志：

   cat logs/single_request_*.log
   

项目结构

imagegen-go/
├── bin/                # 编译后的二进制文件
├── main/              # 主程序目录
├── test/             # 测试脚本目录
└── logs/             # 日志文件目录

注意事项

• 请确保 OpenAI API 密钥的有效性和安全性。
• 图像生成可能会受到 API 的限制，如次数和大小。
• 生成的图像将保存在指定的 destination 路径中，默认为用户的 Downloads 目录。