scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-08 08:08:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

style(proxy): ruff format user_api_key_auth.py

Browse source
This commit is contained in:
lostmartian 2026-08-15 20:45:34 +05:30
parent b1b0fd586c
commit d66e7848cf
1 changed files with 4 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/proxy/auth/user_api_key_auth.py
View file

@ -500,9 +500,7 @@ _GLOBAL_PROXY_SPEND_DB_FLOOR_TTL_SECONDS: Final = 5
_GLOBAL_PROXY_SPEND_DB_FLOOR_MARKER_PREFIX: Final = "global_proxy_spend_db_floor:"
def invalidate_global_proxy_spend_db_floor_marker(
user_api_key_cache: UserApiKeyCache, cache_key: str
) -> None:
def invalidate_global_proxy_spend_db_floor_marker(user_api_key_cache: UserApiKeyCache, cache_key: str) -> None:
"""Drop the worker-local DB-floor marker for ``cache_key``.
ResetBudgetJob calls this after zeroing the aggregate DB row so a stale
@ -583,15 +581,11 @@ async def _fetch_global_spend_with_event_coordination(
# pushed Redis above the DB value is not overwritten by a repair
# carrying a slightly-stale DB total. Use the resulting Redis value
# so the enforced total reflects any concurrent increment.
repaired_value: Final[float | None] = (
await user_api_key_cache.redis_cache.async_set_max(
key=cache_key, value=db_spend
)
repaired_value: Final[float | None] = await user_api_key_cache.redis_cache.async_set_max(
key=cache_key, value=db_spend
)
numeric_value = repaired_value if repaired_value is not None else db_spend
user_api_key_cache.in_memory_cache.set_cache(
key=cache_key, value=numeric_value
)
user_api_key_cache.in_memory_cache.set_cache(key=cache_key, value=numeric_value)
return numeric_value