From ee3e4f00a3b09fc01ce170b9dee547c99e3b9ea6 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 24 Apr 2026 17:19:05 -0700 Subject: [PATCH] fix: use model_group not resolved model name for per-model spend counter key --- litellm/proxy/hooks/proxy_track_cost_callback.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/hooks/proxy_track_cost_callback.py b/litellm/proxy/hooks/proxy_track_cost_callback.py index af1da3fd335..6513da8d63b 100644 --- a/litellm/proxy/hooks/proxy_track_cost_callback.py +++ b/litellm/proxy/hooks/proxy_track_cost_callback.py @@ -208,8 +208,11 @@ class _ProxyDBLogger(CustomLogger): # Atomically update spend counters (in-memory + Redis) # for cross-pod budget enforcement. + # Use model_group (router group name, e.g. "gpt-4o") not + # sl_object["model"] (resolved name, e.g. "gpt-4o-2024-08-06") + # so the key matches what auth_checks reads from the request. _model_for_counter = ( - sl_object.get("model_group") or sl_object.get("model") + sl_object.get("model_group") or kwargs.get("model") if sl_object is not None else kwargs.get("model") )