Add minimal server and client

Author: tendant

examples/minimal_client/main.go

@@ -0,0 +1,71 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/mark3labs/mcp-go/client/transport"
+)
+
+func main() {
+	// Create a new Streamable HTTP transport with a longer timeout
+	trans, err := transport.NewStreamableHTTP("http://localhost:8080/mcp",
+		transport.WithHTTPTimeout(30*time.Second))
+	if err != nil {
+		fmt.Printf("Failed to create transport: %v\n", err)
+		os.Exit(1)
+	}
+	defer trans.Close()
+
+	// Create a context with timeout
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	// Initialize the connection
+	fmt.Println("Initializing connection...")
+	initRequest := transport.JSONRPCRequest{
+		JSONRPC: "2.0",
+		ID:      1,
+		Method:  "initialize",
+	}
+
+	initResponse, err := trans.SendRequest(ctx, initRequest)
+	if err != nil {
+		fmt.Printf("Failed to initialize: %v\n", err)
+		os.Exit(1)
+	}
+
+	// Print the initialization response
+	initResponseJSON, _ := json.MarshalIndent(initResponse, "", "  ")
+	fmt.Printf("Initialization response: %s\n", initResponseJSON)
+	fmt.Printf("Session ID: %s\n", trans.GetSessionId())
+
+	// Call the echo tool
+	fmt.Println("\nCalling echo tool...")
+	echoRequest := transport.JSONRPCRequest{
+		JSONRPC: "2.0",
+		ID:      2,
+		Method:  "tools/call",
+		Params: map[string]interface{}{
+			"name": "echo",
+			"arguments": map[string]interface{}{
+				"message": "Hello from minimal client!",
+			},
+		},
+	}
+
+	echoResponse, err := trans.SendRequest(ctx, echoRequest)
+	if err != nil {
+		fmt.Printf("Failed to call echo tool: %v\n", err)
+		os.Exit(1)
+	}
+
+	// Print the echo response
+	echoResponseJSON, _ := json.MarshalIndent(echoResponse, "", "  ")
+	fmt.Printf("Echo response: %s\n", echoResponseJSON)
+
+	fmt.Println("\nTest completed successfully!")
+}

examples/minimal_server/main.go

@@ -0,0 +1,83 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/mark3labs/mcp-go/server"
+)
+
+func main() {
+	// Create a new MCP server
+	mcpServer := server.NewMCPServer("minimal-server", "1.0.0")
+
+	// Add a simple echo tool
+	mcpServer.AddTool(
+		mcp.Tool{
+			Name:        "echo",
+			Description: "Echoes back the input",
+			InputSchema: mcp.ToolInputSchema{
+				Type: "object",
+				Properties: map[string]interface{}{
+					"message": map[string]interface{}{
+						"type":        "string",
+						"description": "The message to echo",
+					},
+				},
+				Required: []string{"message"},
+			},
+		},
+		func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+			// Extract the message from the request
+			message, ok := request.Params.Arguments["message"].(string)
+			if !ok {
+				return nil, fmt.Errorf("message must be a string")
+			}
+
+			// Create the result
+			result := &mcp.CallToolResult{
+				Result: mcp.Result{},
+				Content: []mcp.Content{
+					mcp.TextContent{
+						Type: "text",
+						Text: fmt.Sprintf("Echo: %s", message),
+					},
+				},
+			}
+
+			return result, nil
+		},
+	)
+
+	// Create a new Streamable HTTP server with direct JSON responses
+	streamableServer := server.NewStreamableHTTPServer(mcpServer,
+		server.WithEnableJSONResponse(true),
+	)
+
+	// Start the server in a goroutine
+	go func() {
+		log.Println("Starting Minimal Streamable HTTP server on :8080...")
+		if err := streamableServer.Start(":8080"); err != nil {
+			log.Fatalf("Failed to start server: %v", err)
+		}
+	}()
+
+	// Wait for interrupt signal to gracefully shutdown the server
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+
+	log.Println("Shutting down server...")
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	if err := streamableServer.Shutdown(ctx); err != nil {
+		log.Fatalf("Server shutdown failed: %v", err)
+	}
+	log.Println("Server exited properly")
+}