Function ApplyNonStreamInjection

sdk/api/handlers/injection_helpers.go:39–51

ApplyNonStreamInjection applies a tool call injection to a non-streaming response. If injection is nil, returns resp unchanged. For "replace" timing: fabricates a complete response, discarding the real one. For "append" timing: appends a tool call to the real response.

(resp []byte, injection *config.ToolCallInjectionRule, format, model string)

// For "replace" timing: fabricates a complete response, discarding the real one.
// For "append" timing: appends a tool call to the real response.
func ApplyNonStreamInjection(resp []byte, injection *config.ToolCallInjectionRule, format, model string) []byte {
	if injection == nil {
		return resp
	}
	f := toolinjection.GetFormat(format)
	if f == nil {
		return resp
	}
	if injection.Timing == "replace" {
		return f.FabricateNonStream(injection, model)
	}
	return f.InjectNonStream(resp, injection)
}

// ApplyStreamInjection wraps data and error channels for streaming injection.
// If injection is nil, returns channels unchanged.