MCP Go Sdk

🚀 MCP Go SDK Translation

This SDK provides a Go implementation of the Model Context Protocol (MCP), enabling bidirectional communication between clients and servers for tool execution, resource access, and prompt handling.

🚀 Quick Start

Here is a simple example of creating an MCP server:

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() {
    // Create a new server
    srv := server.NewServer("Example Server")

    // Add a tool
    srv.AddTool("greet", func(name string) string {
        return "Hello, " + name + "!"
    }, "Greet a person")

    // Create a session
    session := server.NewSession(context.Background(), srv)

    // Create and start the transport
    t := transport.NewStdioTransport(session)
    if err := t.Start(); err != nil {
        log.Fatal(err)
    }
}

✨ Features

Transport Layer

• Multiple transport options (Standard I/O, SSE, WebSocket)
• Bidirectional communication
• Configurable endpoints and settings

Server Implementation

• Tool registration and execution
• Resource pattern matching and access
• Prompt template rendering
• Session management
• Reflection-based handler invocation

Core Protocol Types

• JSON-RPC message handling
• MCP-specific types (tools, resources, prompts)
• Functionality and initialization configuration

📦 Installation

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

💻 Usage Examples

Basic Usage

Create a Server

// Create a new server with a name
srv := server.NewServer("My Server")

// Optionally configure server capabilities
srv.WithCapabilities(protocol.ServerCapabilities{
    SupportsAsync: true,
})

Add Tools

// Add a synchronous tool
srv.AddTool("myTool", func(arg1 string, arg2 int) (string, error) {
    return fmt.Sprintf("Processed %s with %d", arg1, arg2), nil
}, "Tool description")

// Add an asynchronous tool
srv.AddAsyncTool("longRunningTool", func(params string) error {
    // Long-running operation
    return nil
}, "Async tool description")

Add Resources

// Add a resource with pattern matching
srv.AddResource("files/{path}", func(path string) ([]byte, error) {
    return ioutil.ReadFile(path)
}, "Access files")

// Resource patterns support multiple parameters
srv.AddResource("api/{version}/{endpoint}", func(version, endpoint string) (interface{}, error) {
    return callAPI(version, endpoint)
}, "API access")

Add Prompts

// Add a simple text prompt
srv.AddPrompt("confirm", func(action string) string {
    return fmt.Sprintf("Are you sure you want to %s?", action)
}, "Confirmation prompt")

// Add a multi-message prompt
srv.AddPrompt("chat", func(context string) []protocol.PromptMessage {
    return []protocol.PromptMessage{
        {
            Role: protocol.RoleAssistant,
            Content: protocol.MessageContent{
                Text: []string{context},
            },
        },
    }
}, "Chat prompt")

Transport Configuration

// Configure standard I/O transport
t := transport.NewStdioTransport(session)
if err := t.Start(); err != nil {
    log.Fatal(err)
}

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

📄 License

This SDK is released under the MIT License. Please refer to the file LICENSE for details.

Contribution

If you want to contribute, please refer to the contribution guidelines in the project repository.