diff --git a/examples/hello/main.go b/examples/hello/main.go index 77fc191..69a9247 100644 --- a/examples/hello/main.go +++ b/examples/hello/main.go @@ -32,11 +32,11 @@ func SayHi(ctx context.Context, ss *mcp.ServerSession, params *mcp.CallToolParam // TODO(jba): it should be OK for args to be a pointer, but this fails in // jsonschema. Needs investigation. -func PromptHi(ctx context.Context, ss *mcp.ServerSession, args HiArgs, params *mcp.GetPromptParams) (*mcp.GetPromptResult, error) { +func PromptHi(ctx context.Context, ss *mcp.ServerSession, args HiArgs, _ *mcp.GetPromptParams) (*mcp.GetPromptResult, error) { return &mcp.GetPromptResult{ Description: "Code review prompt", Messages: []*mcp.PromptMessage{ - {Role: "user", Content: mcp.NewTextContent("Say hi to " + params.Name)}, + {Role: "user", Content: mcp.NewTextContent("Say hi to " + args.Name)}, }, }, nil } diff --git a/mcp/prompt.go b/mcp/prompt.go index c690325..de5555d 100644 --- a/mcp/prompt.go +++ b/mcp/prompt.go @@ -75,11 +75,11 @@ func NewServerPrompt[In any](name, description string, handler func(context.Cont if err != nil { return nil, err } - var v In - if err := unmarshalSchema(rawArgs, resolved, &v); err != nil { + var args In + if err := unmarshalSchema(rawArgs, resolved, &args); err != nil { return nil, err } - return handler(ctx, ss, v, params) + return handler(ctx, ss, args, params) } for _, opt := range opts { opt.set(prompt)