Skip to content

Commit a9e539c

Browse files
committed
Refactor argument retrieval in makeHandler to use req.GetArguments()
1 parent 0b7b889 commit a9e539c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

main.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,8 +121,9 @@ func makeHandler(cfg ForgeConfig, tcfg ToolConfig) server.ToolHandlerFunc {
121121
return func(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
122122
// 1. Gather variables
123123
vars := map[string]interface{}{}
124+
args := req.GetArguments()
124125
for _, inp := range tcfg.Inputs {
125-
val, ok := req.Params.Arguments[inp.Name]
126+
val, ok := args[inp.Name]
126127
if !ok && inp.Required {
127128
return mcp.NewToolResultError(fmt.Sprintf("missing required argument: %s", inp.Name)), nil
128129
}

0 commit comments

Comments
 (0)