imagegen-go: 图像生成MCP服务器

项目背景与介绍

图像生成技术在现代科技中发挥着至关重要的作用，它不仅可以用来创建艺术作品，还能用于增强现实、游戏开发、市场营销以及设计等众多领域。imagegen-go项目通过实现MCP(Model Control Protocol)协议，使用户能够高效地控制图像生成过程。这一项目的核心是利用OpenAI的创新技术，与DALL-E的图像生成能力相结合，提供一种先进的解决方案来生成视觉内容。

环境要求

为了顺利运行imagegen-go项目，您需满足以下环境要求：

• Go语言：版本要求为1.23.5或更高，这提供了一个高效的构建和执行环境。
• OpenAI API密钥：这是连接OpenAI服务的必需项，确保您已申请并配置有效的API密钥。

配置与编译

在开始使用图像生成服务器之前，您需要进行一些配置。请设置您的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

日志管理

MCP协议简介

MCP协议是一个基于JSON-RPC 2.0的协议，用于实现模型与工具之间的通信。imagegen-go项目现已成功实现如下MCP方法：

• initialize：用于服务器初始化配置。
• tools/list：获取当前可用工具列表。
• tools/call：调用工具以生成所需图像。
• resources/list：检索可用资源列表。
• prompts/list：获取系统可处理的提示词列表。

生成图像工具概述

• 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/                # 编译后的二进制文件
│   └── imagegen-go     # 主程序
├── main/               # 主程序源代码
│   ├── main.go         # 主程序入口
│   ├── types.go        # 数据类型定义
│   ├── utils.go        # 工具函数
│   └── pretty-json.go  # JSON 格式化
├── pkg/                # 依赖包
│   └── openai/         # OpenAI API客户端
│       └── client.go   # OpenAI API交互
├── logs/               # 日志目录
├── test_mcp.sh         # 基本测试脚本
├── debug_mcp.sh        # 详细调试脚本
├── interactive_test.sh # 交互式测试脚本
├── test_single_request.sh # 单独请求测试脚本
└── README.md           # 文档说明

注意事项与建议

• 请确保您的OpenAI API密钥有效，并且拥有足够的使用额度，以免影响图像生成过程。
• 默认情况下，生成的图像会保存在用户的Downloads目录中，提供的目标路径若无效则会回退至默认路径。
• 本服务器通过标准输入/输出进行与客户端沟通，而非传统的HTTP服务器。请根据通信机制进行配置和使用。

应用场景与价值

通过强大的图像生成能力，imagegen-go项目可广泛应用于商业广告创作、个性化定制产品设计以及跨领域的创意开发。其高效的图像生成性能和灵活的配置能力，使其在艺术创作、教育培训以及企业展示等场景中皆可服务，为用户带来独特的视觉体验和价值提升。

总之，imagegen-go项目不仅展示了MCP协议与OpenAI技术结合的巨大潜力，也为图像生成领域带来了堪称革命性的解决方案。期待您的参与和更加深入的探索。