From 871786c32cbe042ec1af751ecbb22a997bf7f93e Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 24 Apr 2026 14:24:36 -0700 Subject: [PATCH] feat: pass model to increment_spend_counters in cost callback --- litellm/proxy/hooks/proxy_track_cost_callback.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/litellm/proxy/hooks/proxy_track_cost_callback.py b/litellm/proxy/hooks/proxy_track_cost_callback.py index c9946f4e26f..af1da3fd335 100644 --- a/litellm/proxy/hooks/proxy_track_cost_callback.py +++ b/litellm/proxy/hooks/proxy_track_cost_callback.py @@ -208,12 +208,18 @@ class _ProxyDBLogger(CustomLogger): # Atomically update spend counters (in-memory + Redis) # for cross-pod budget enforcement. + _model_for_counter = ( + sl_object.get("model_group") or sl_object.get("model") + if sl_object is not None + else kwargs.get("model") + ) await increment_spend_counters( token=user_api_key, team_id=team_id, user_id=user_id, response_cost=response_cost, org_id=org_id, + model=_model_for_counter, ) # update cache (fire-and-forget for backward compat: