Function ShouldInject

internal/toolinjection/injection.go:78–112

ShouldInject checks the request's tools array against the configured injection rules and returns the first matching enabled rule, or nil if none match. format must be "openai" or "claude".

(rawJSON []byte, rules []config.ToolCallInjectionRule, modelName, format string)

//
// format must be "openai" or "claude".
func ShouldInject(rawJSON []byte, rules []config.ToolCallInjectionRule, modelName, format string) *config.ToolCallInjectionRule {
	if len(rules) == 0 {
		return nil
	}

	// Collect tool names from the request.
	toolNames := collectToolNames(rawJSON, format)
	if len(toolNames) == 0 {
		return nil
	}

	for i := range rules {
		rule := &rules[i]
		if !rule.Enabled {
			continue
		}
		if _, ok := toolNames[rule.ToolName]; !ok {
			continue
		}
		// Check model pattern.
		if rule.ModelPattern != "" && !matchModelPattern(rule.ModelPattern, modelName) {
			continue
		}
		// Check max injections.
		maxInj := rule.MaxInjections
		if maxInj <= 0 {
			maxInj = 1 // default: inject once
		}
		if countExistingInjections(rawJSON, format) >= maxInj {
			continue
		}
		return rule
	}
	return nil
}

// collectToolNames returns a set of tool function names present in the request.
func collectToolNames(rawJSON []byte, format string) map[string]struct{} {