Go MCP

🚀 go-mcp

go-mcp 是一個用於構建 MCP（模型上下文協議）服務器的 Go 語言 SDK，它具備類型安全的特性，且提供直觀易用的接口，能有效幫助開發者高效搭建服務器。

🚀 快速開始

目錄結構示例

項目目錄結構如下：

your-project/
  cmd/temperature/
    main.go
    handler.go
  cmd/mcpgen/
    main.go

示例代碼

步驟一：定義 MCP 服務器

運行 mcpgen 工具生成代碼：

go run ./cmd/mcpgen

步驟二：實現 MCP 服務器邏輯

在 handler.go 中實現邏輯，例如溫度轉換工具：

package main

import (
	"context"
	"fmt"
	"math"
)

type toolHandler struct{}

func NewHandler(h *toolHandler) mcp.Handler {
	return mcp.HandlerFunc(func(ctx context.Context, req *mcp.Request) (*mcp.Response, error) {
		if req.IsPing() {
			return &mcp.Response{Id: req.Id}, nil
		}
		if req.IsCallTool() {
			content := req.GetCallTool().Content
			tool := content[0].GetTool()
			switch tool.ID {
			case "temperature-converter":
				fromUnit := string(tool.Params["from_unit"])
				toUnit := string(tool.Params["to_unit"])
				temp, err := tool.Params["temperature"].(float64)
				if err != nil {
					return nil, fmt.Errorf("invalid temperature value")
				}
				result := convertTemperature(temp, fromUnit, toUnit)
				return &mcp.Response{
					Id:    req.Id,
					Data:  mcp.NewData(map[string]interface{}{
						"result": result,
					}),
				}, nil
			default:
				return nil, fmt.Errorf("tool not found")
			}
		}
		return nil, nil
	})
}

func convertTemperature(value float64, fromUnit string, toUnit string) float64 {
	switch fromUnit {
	case "celsius":
		switch toUnit {
		case "fahrenheit":
			return (value * 9.0 / 5.0) + 32
		default:
			return value
		}
	case "fahrenheit":
		switch toUnit {
		case "celsius":
			return (value - 32) * 5.0 / 9.0
		default:
			return value
		}
	default:
		return value
	}
}

func main() {
	handler := NewHandler(new(toolHandler))
	server := mcp.NewServer(":8080", handler)
	if err := server.Start(); err != nil {
		fmt.Printf("Failed to start server: %v\n", err)
	}
}

在終端運行以下命令啟動服務器：

go run ./cmd/temperature/main.go

✨ 主要特性

• 類型安全：通過代碼生成確保工具和提示參數靜態類型化，在編譯時捕獲錯誤。
• 簡單直觀的 API：提供自然、符合 Go 風格的接口，降低學習曲線。
• 開發者友好：注重 API 的易用性。

💻 使用示例

基礎用法

發送 Ping 請求

使用 curl 命令發送 Ping 請求：

curl -X POST http://localhost:8080/ping

調用溫度轉換工具

調用預定義的 "temperature-converter" 工具：

curl -X POST http://localhost:8080/call_tool \
  --data '{"tool_id":"temperature-converter","params":{"from_unit":"celsius","to_unit":"fahrenheit","temperature":32}}'

查看工具列表

列出所有註冊的工具：

curl -X GET http://localhost:8080/tools

📚 詳細文檔

已支持功能

• Ping：驗證服務器連通性。
• Tools：管理工具註冊與調用。
• Prompts：處理用戶提示請求。

更多功能請參考項目文檔。

📄 許可證

本項目使用 MIT 許可證，詳情請見 LICENSE 文件。