@@ -433,28 +433,28 @@ We provide a mechanism to add MCP-level middleware on the both the client and se
433433// A MethodHandler handles MCP messages.
434434// For methods, exactly one of the return values must be nil.
435435// For notifications, both must be nil.
436- type MethodHandler[S Session] func (ctx context.Context , method string , req * Request[S] ) (result Result, err error )
436+ type MethodHandler func (ctx context.Context , method string , req Request) (result Result, err error )
437437
438438// Middleware is a function from MethodHandlers to MethodHandlers.
439- type Middleware[S Session] func (MethodHandler[S] ) MethodHandler[S]
439+ type Middleware func (MethodHandler) MethodHandler
440440
441441// AddMiddleware wraps the client/server's current method handler using the provided
442442// middleware. Middleware is applied from right to left, so that the first one
443443// is executed first.
444444//
445445// For example, AddMiddleware(m1, m2, m3) augments the server method handler as
446446// m1(m2(m3(handler))).
447- func (c *Client) AddSendingMiddleware (middleware ...Middleware [*ClientSession] )
448- func (c *Client) AddReceivingMiddleware (middleware ...Middleware [*ClientSession] )
449- func (s *Server) AddSendingMiddleware (middleware ...Middleware [*ServerSession] )
450- func (s *Server) AddReceivingMiddleware (middleware ...Middleware [*ServerSession] )
447+ func (c *Client) AddSendingMiddleware (middleware ...Middleware )
448+ func (c *Client) AddReceivingMiddleware (middleware ...Middleware )
449+ func (s *Server) AddSendingMiddleware (middleware ...Middleware )
450+ func (s *Server) AddReceivingMiddleware (middleware ...Middleware )
451451` ` `
452452
453453As an example, this code adds server-side logging:
454454
455455` ` ` go
456- func withLogging (h mcp.MethodHandler [*mcp. ServerSession ] ) mcp.MethodHandler [*mcp. ServerSession ] {
457- return func (ctx context.Context , s *mcp. ServerSession , method string , params mcp.Params ) (res mcp.Result , err error ) {
456+ func withLogging (h mcp.MethodHandler ) mcp.MethodHandler {
457+ return func (ctx context.Context , method string , req mcp.Request ) (res mcp.Result , err error ) {
458458 log.Printf (" request: %s %v " , method, params)
459459 defer func () { log.Printf (" response: %v , %v " , res, err) }()
460460 return h (ctx, s , method, params)
0 commit comments