update demo

Author: leavez

examples/custom_context/main.go

@@ -122,10 +122,9 @@ func NewMCPServer() *MCPServer {
 	}
 }
 
-func (s *MCPServer) ServeSSE(addr string) *server.SSEServer {
-	return server.NewSSEServer(s.server,
-		server.WithBaseURL(fmt.Sprintf("http://%s", addr)),
-		server.WithSSEContextFunc(authFromRequest),
+func (s *MCPServer) ServeHTTP() *server.StreamableHTTPServer {
+	return server.NewStreamableHTTPServer(s.server,
+		server.WithHTTPContextFunc(authFromRequest),
 	)
 }
 
@@ -135,12 +134,12 @@ func (s *MCPServer) ServeStdio() error {
 
 func main() {
 	var transport string
-	flag.StringVar(&transport, "t", "stdio", "Transport type (stdio or sse)")
+	flag.StringVar(&transport, "t", "stdio", "Transport type (stdio or http)")
 	flag.StringVar(
 		&transport,
 		"transport",
 		"stdio",
-		"Transport type (stdio or sse)",
+		"Transport type (stdio or http)",
 	)
 	flag.Parse()
 
@@ -151,15 +150,15 @@ func main() {
 		if err := s.ServeStdio(); err != nil {
 			log.Fatalf("Server error: %v", err)
 		}
-	case "sse":
-		sseServer := s.ServeSSE("localhost:8080")
-		log.Printf("SSE server listening on :8080")
-		if err := sseServer.Start(":8080"); err != nil {
+	case "http":
+		httpServer := s.ServeHTTP()
+		log.Printf("HTTP server listening on :8080")
+		if err := httpServer.Start(":8080"); err != nil {
 			log.Fatalf("Server error: %v", err)
 		}
 	default:
 		log.Fatalf(
-			"Invalid transport type: %s. Must be 'stdio' or 'sse'",
+			"Invalid transport type: %s. Must be 'stdio' or 'http'",
 			transport,
 		)
 	}

examples/everything/main.go

@@ -472,17 +472,17 @@ func handleNotification(
 
 func main() {
 	var transport string
-	flag.StringVar(&transport, "t", "stdio", "Transport type (stdio or sse)")
-	flag.StringVar(&transport, "transport", "stdio", "Transport type (stdio or sse)")
+	flag.StringVar(&transport, "t", "stdio", "Transport type (stdio or http)")
+	flag.StringVar(&transport, "transport", "stdio", "Transport type (stdio or http)")
 	flag.Parse()
 
 	mcpServer := NewMCPServer()
 
-	// Only check for "sse" since stdio is the default
-	if transport == "sse" {
-		sseServer := server.NewSSEServer(mcpServer, server.WithBaseURL("http://localhost:8080"))
-		log.Printf("SSE server listening on :8080")
-		if err := sseServer.Start(":8080"); err != nil {
+	// Only check for "http" since stdio is the default
+	if transport == "http" {
+		httpServer := server.NewStreamableHTTPServer(mcpServer)
+		log.Printf("HTTP server listening on :8080/mcp")
+		if err := httpServer.Start(":8080"); err != nil {
 			log.Fatalf("Server error: %v", err)
 		}
 	} else {

examples/simple_client/main.go

@@ -17,12 +17,12 @@ import (
 func main() {
 	// Define command line flags
 	stdioCmd := flag.String("stdio", "", "Command to execute for stdio transport (e.g. 'python server.py')")
-	sseURL := flag.String("sse", "", "URL for SSE transport (e.g. 'http://localhost:8080/sse')")
+	httpURL := flag.String("http", "", "URL for HTTP transport (e.g. 'http://localhost:8080/mcp')")
 	flag.Parse()
 
 	// Validate flags
-	if (*stdioCmd == "" && *sseURL == "") || (*stdioCmd != "" && *sseURL != "") {
-		fmt.Println("Error: You must specify exactly one of --stdio or --sse")
+	if (*stdioCmd == "" && *httpURL == "") || (*stdioCmd != "" && *httpURL != "") {
+		fmt.Println("Error: You must specify exactly one of --stdio or --http")
 		flag.Usage()
 		os.Exit(1)
 	}
@@ -51,11 +51,6 @@ func main() {
 		// Create stdio transport with verbose logging
 		stdioTransport := transport.NewStdio(command, nil, cmdArgs...)
 
-		// Start the transport
-		if err := stdioTransport.Start(ctx); err != nil {
-			log.Fatalf("Failed to start stdio transport: %v", err)
-		}
-
 		// Create client with the transport
 		c = client.NewClient(stdioTransport)
 
@@ -78,20 +73,20 @@ func main() {
 			}()
 		}
 	} else {
-		fmt.Println("Initializing SSE client...")
-		// Create SSE transport
-		sseTransport, err := transport.NewSSE(*sseURL)
+		fmt.Println("Initializing HTTP client...")
+		// Create HTTP transport
+		httpTransport, err := transport.NewStreamableHTTP(*httpURL)
 		if err != nil {
-			log.Fatalf("Failed to create SSE transport: %v", err)
-		}
-
-		// Start the transport
-		if err := sseTransport.Start(ctx); err != nil {
-			log.Fatalf("Failed to start SSE transport: %v", err)
+			log.Fatalf("Failed to create HTTP transport: %v", err)
 		}
 
 		// Create client with the transport
-		c = client.NewClient(sseTransport)
+		c = client.NewClient(httpTransport)
+	}
+
+	// Start the client 
+	if err := c.Start(ctx); err != nil {
+		log.Fatalf("Failed to start client: %v", err)
 	}
 
 	// Set up notification handler