Notifications

internal/protocol/protocol.go

@@ -157,6 +157,7 @@ func NewProtocol(options *ProtocolOptions) *Protocol {
 	}
 
 	// Set up default handlers
+	p.SetNotificationHandler("notifications/initialized", p.handleInitializedNotification)
 	p.SetNotificationHandler("notifications/cancelled", p.handleCancelledNotification)
 	p.SetNotificationHandler("$/progress", p.handleProgressNotification)
 
@@ -297,6 +298,12 @@ func (p *Protocol) handleRequest(ctx context.Context, request *transport.BaseJSO
 	}()
 }
 
+// Note: Not sure what this is for, yet, after perusing the MCP spec, lightly.
+// Just have a placeholder here, for now, so that our server can at least receive it.
+func (p *Protocol) handleInitializedNotification(notification *transport.BaseJSONRPCNotification) error {
+	return nil
+}
+
 func (p *Protocol) handleProgressNotification(notification *transport.BaseJSONRPCNotification) error {
 	var params struct {
 		Progress      int64               `json:"progress"`

transport/http/http_client.go

@@ -4,10 +4,12 @@ import (
 	"bytes"
 	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"io"
 	"net/http"
 	"sync"
+	"time"
 
 	"github.com/rvoh-emccaleb/mcp-golang/transport"
 )
@@ -68,6 +70,30 @@ func (t *HTTPClientTransport) Send(ctx context.Context, message *transport.BaseJ
 		req.Header.Set(key, value)
 	}
 
+	if message.Type == transport.BaseMessageTypeJSONRPCNotificationType {
+		// For notifications, use an arbitrary short timeout
+		ctx, cancel := context.WithTimeout(ctx, 1*time.Millisecond)
+		defer cancel()
+		req = req.WithContext(ctx)
+
+		// Fire and forget for notifications
+		go func() {
+			resp, err := t.client.Do(req)
+			if err != nil && !errors.Is(err, context.DeadlineExceeded) {
+				if t.errorHandler != nil {
+					t.errorHandler(fmt.Errorf("async notification error: %w", err))
+				}
+			}
+			if resp != nil {
+				resp.Body.Close()
+			}
+		}()
+
+		return nil
+	}
+
+	// For non-notifications, continue with normal synchronous request
+
 	resp, err := t.client.Do(req)
 	if err != nil {
 		return fmt.Errorf("failed to send request: %w", err)