From fe22a9253d1d19fdcd0ab63184d78daf71d599f5 Mon Sep 17 00:00:00 2001 From: Emerson Gomes Date: Tue, 10 Feb 2026 14:19:53 -0600 Subject: [PATCH] chore(router-budget): simplify redis queue lock getter --- litellm/router_strategy/budget_limiter.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/litellm/router_strategy/budget_limiter.py b/litellm/router_strategy/budget_limiter.py index fdd4a45e0e1..154f778d6cf 100644 --- a/litellm/router_strategy/budget_limiter.py +++ b/litellm/router_strategy/budget_limiter.py @@ -353,13 +353,7 @@ class RouterBudgetLimiting(CustomLogger): self.redis_increment_operation_queue.append(increment_op) def _get_redis_increment_queue_lock(self) -> asyncio.Lock: - queue_lock: Optional[asyncio.Lock] = getattr( - self, "_redis_increment_queue_lock", None - ) - if queue_lock is None: - queue_lock = asyncio.Lock() - self._redis_increment_queue_lock = queue_lock - return queue_lock + return self._redis_increment_queue_lock async def async_log_success_event(self, kwargs, response_obj, start_time, end_time): """Original method now uses helper functions"""