Function NewHandler

pkg/tools/tools.go:25–44

NewHandler creates a type-safe tool handler from a function that accepts typed parameters. It unmarshals the tool-call arguments via [aijson.Unmarshal], which runs strict [encoding/json.Unmarshal] first and only falls back to a narrow set of shape repairs (stringified array, bare scalar where an arr

(fn func(context.Context, T) (*ToolCallResult, error))

// tool_input_repaired log entry so per-(model, tool) repair rates can be
// tracked.
func NewHandler[T any](fn func(context.Context, T) (*ToolCallResult, error)) ToolHandler {
	return func(ctx context.Context, toolCall ToolCall) (*ToolCallResult, error) {
		var params T
		args := toolCall.Function.Arguments
		if args == "" {
			args = "{}"
		}

		err := aijson.Unmarshal([]byte(args), &params, aijson.OnRepair(func(kinds []aijson.Kind) {
			slog.InfoContext(ctx, "tool_input_repaired",
				"tool", toolCall.Function.Name,
				"repairs", kinds,
			)
		}))
		if err != nil {
			return nil, err
		}
		return fn(ctx, params)
	}
}

type ToolHandler func(ctx context.Context, toolCall ToolCall) (*ToolCallResult, error)