From 3df367bdb8ff9400943292f779a8307605c1bd3c Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaan@berri.ai>
Date: Sat, 11 Apr 2026 11:38:47 -0700
Subject: [PATCH] fix(reset_budget_job): use spend_counter_cache from
 proxy_server in reset_budget_windows

---
 litellm/proxy/common_utils/reset_budget_job.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/litellm/proxy/common_utils/reset_budget_job.py b/litellm/proxy/common_utils/reset_budget_job.py
index 099d68209e0..84740975c2f 100644
--- a/litellm/proxy/common_utils/reset_budget_job.py
+++ b/litellm/proxy/common_utils/reset_budget_job.py
@@ -565,9 +565,9 @@ class ResetBudgetJob:
         reset_at_str = window.get("reset_at")
         if not reset_at_str:
             return False
-        reset_at = datetime.fromisoformat(
-            reset_at_str.replace("Z", "+00:00")
-        ).replace(tzinfo=None)
+        reset_at = datetime.fromisoformat(reset_at_str.replace("Z", "+00:00")).replace(
+            tzinfo=None
+        )
         if reset_at > now:
             return False
         spend_counter_cache.in_memory_cache.set_cache(key=counter_key, value=0.0)
@@ -591,6 +591,8 @@ class ResetBudgetJob:
         reset_at <= now. Only the expired windows are reset; other windows are untouched.
         """
 
+        from litellm.proxy.proxy_server import spend_counter_cache
+
         now = datetime.utcnow()
 
         # --- Keys ---
@@ -605,9 +607,11 @@ class ResetBudgetJob:
                 windows: list = raw if isinstance(raw, list) else json.loads(raw)
                 changed = False
                 for window in windows:
-                    counter_key = f"spend:key:{key.token}:window:{window['budget_duration']}"
+                    counter_key = (
+                        f"spend:key:{key.token}:window:{window['budget_duration']}"
+                    )
                     if await ResetBudgetJob._reset_expired_window(
-                        window, counter_key, ProxyLogging.spend_counter_cache, now
+                        window, counter_key, spend_counter_cache, now
                     ):
                         changed = True
                 if changed:
@@ -632,7 +636,9 @@ class ResetBudgetJob:
                 windows = raw if isinstance(raw, list) else json.loads(raw)
                 changed = False
                 for window in windows:
-                    counter_key = f"spend:team:{team.team_id}:window:{window['budget_duration']}"
+                    counter_key = (
+                        f"spend:team:{team.team_id}:window:{window['budget_duration']}"
+                    )
                     if await ResetBudgetJob._reset_expired_window(
                         window, counter_key, spend_counter_cache, now
                     ):