diff --git a/relay/common/override.go b/relay/common/override.go index d75216d1..af0b4361 100644 --- a/relay/common/override.go +++ b/relay/common/override.go @@ -27,9 +27,11 @@ const ( var errSourceHeaderNotFound = errors.New("source header does not exist") var paramOverrideKeyAuditPaths = map[string]struct{}{ - "model": {}, - "service_tier": {}, - "inference_geo": {}, + "model": {}, + "original_model": {}, + "upstream_model": {}, + "service_tier": {}, + "inference_geo": {}, } type paramOverrideAuditRecorder struct {