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 密鑰。

編譯項目

運行以下命令進行編譯：

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 目錄中，方便查看和分析。可使用以下命令查看日誌：

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

調試步驟

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 目錄。