Imagegen Go

🚀 图像生成服务器（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 协议，实现模型与工具间的通信。
• 利用 OpenAI 的 DALL - E API 生成图像。
• 提供多个测试脚本，方便调试和使用。

📦 安装指南

环境准备

确保你已经安装了 Go 1.23.5 或更高版本，并获取了有效的 OpenAI API 密钥。

环境变量设置

export OPENAI_API_KEY="你的OpenAI API密钥"

项目编译

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

💻 使用示例

测试脚本使用

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

基本测试

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

./test_mcp.sh

详细调试

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

./debug_mcp.sh

交互式测试

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

./interactive_test.sh

单独请求测试

测试单个请求：

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

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

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

调试步骤

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

📚 详细文档

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 目录

🔧 技术细节

项目结构

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

📄 许可证

文档未提及相关内容，故跳过该章节。

⚠️ 重要提示

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

💡 使用建议

在使用测试脚本时，可根据不同的需求选择合适的脚本进行调试，若需要自定义提示词，可使用交互式测试脚本。