scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

chore(router-budget): simplify redis queue lock getter

Browse source
This commit is contained in:
Emerson Gomes 2026-02-10 14:19:53 -06:00
parent c57edc0615
commit fe22a9253d
1 changed files with 1 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/router_strategy/budget_limiter.py
View file

@ -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"""