Skip to content

Commit c9f6ddb

Browse files
committed
update design.md
1 parent 3cd375c commit c9f6ddb

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

design/design.md

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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
453453
As 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

Comments
 (0)