MCP Go Sdk

🚀 MCP Go SDK 翻譯

此 SDK 提供了 Go 語言實現的模型上下文協議（MCP），支持客戶端與服務器之間的雙向通信，可用於工具執行、資源訪問和提示處理。

🚀 快速開始

以下是一個創建 MCP 服務器的簡單示例：

package main

import (
    "context"
    "log"
    "github.com/SetiabudiResearch/mcp-go-sdk/pkg/mcp/server"
    "github.com/SetiabudiResearch/mcp-go-sdk/pkg/mcp/transport"
)

func main() {
    // 創建一個新的服務器
    srv := server.NewServer("Example Server")

    // 添加一個工具
    srv.AddTool("greet", func(name string) string {
        return "Hello, " + name + "!"
    }, "Greet a person")

    // 創建一個會話
    session := server.NewSession(context.Background(), srv)

    // 創建並啟動傳輸
    t := transport.NewStdioTransport(session)
    if err := t.Start(); err != nil {
        log.Fatal(err)
    }
}

✨ 主要特性

• 傳輸層

  • 多種傳輸選項（標準 I/O、SSE、WebSocket）
  • 雙向通信
  • 可配置的端點和設置

• 服務器實現

  • 工具註冊與執行
  • 資源模式匹配與訪問
  • 提示模板渲染
  • 會話管理
  • 基於反射的處理程序調用

• 核心協議類型

  • JSON-RPC 消息處理
  • MCP 特定類型（工具、資源、提示）
  • 功能和初始化配置

📦 安裝指南

go get github.com/SetiabudiResearch/mcp-go-sdk

💻 使用示例

基礎用法

創建服務器

// 創建一個具有名稱的新服務器
srv := server.NewServer("My Server")

// 可選配置服務器功能
srv.WithCapabilities(protocol.ServerCapabilities{
    SupportsAsync: true,
})

添加工具

// 添加同步工具
srv.AddTool("myTool", func(arg1 string, arg2 int) (string, error) {
    return fmt.Sprintf("Processed %s with %d", arg1, arg2), nil
}, "Tool description")

// 添加異步工具
srv.AddAsyncTool("longRunningTool", func(params string) error {
    // 長時間運行的操作
    return nil
}, "Async tool description")

添加資源

// 添加帶模式匹配的資源
srv.AddResource("files/{path}", func(path string) ([]byte, error) {
    return ioutil.ReadFile(path)
}, "Access files")

// 資源模式支持多個參數
srv.AddResource("api/{version}/{endpoint}", func(version, endpoint string) (interface{}, error) {
    return callAPI(version, endpoint)
}, "API access")

添加提示

// 添加簡單的文本提示
srv.AddPrompt("confirm", func(action string) string {
    return fmt.Sprintf("Are you sure you want to %s?", action)
}, "Confirmation prompt")

// 添加多消息提示
srv.AddPrompt("chat", func(context string) []protocol.PromptMessage {
    return []protocol.PromptMessage{
        {
            Role: protocol.RoleAssistant,
            Content: protocol.MessageContent{
                Text: []string{context},
            },
        },
    }
}, "Chat prompt")

傳輸配置

// 配置標準 I/O 傳輸
t := transport.NewStdioTransport(session)
if err := t.Start(); err != nil {
    log.Fatal(err)
}

// 配置 WebSocket 傳輸
addr := "localhost:8080"
ws := transport.NewWebSocketTransport(addr, session)
if err := ws.Start(); err != nil {
    log.Fatal(err)
}

📄 許可證

此 SDK 以 MIT 許可證發佈。請參閱文件 LICENSE 瞭解詳細信息。