Function WithModelOverride

pkg/teamloader/model_override.go:11–20

WithModelOverride wraps a toolset so that every tool it produces carries the given model in its ModelOverride field, enabling per-toolset model routing.

(inner tools.ToolSet, model string)

// WithModelOverride wraps a toolset so that every tool it produces carries the
// given model in its ModelOverride field, enabling per-toolset model routing.
func WithModelOverride(inner tools.ToolSet, model string) tools.ToolSet {
	if model == "" {
		return inner
	}

	return &modelOverrideToolset{
		ToolSet: inner,
		model:   model,
	}
}

type modelOverrideToolset struct {
	tools.ToolSet