scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

fix linting

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-28 20:57:28 -07:00
parent 4fb9d27a35
commit 29d6968f58
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/proxy/db/redis_update_buffer.py
View file

@ -5,10 +5,11 @@ This is to prevent deadlocks and improve reliability
"""
import json
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional, Union
from litellm._logging import verbose_proxy_logger
from litellm.caching import RedisCache
from litellm.constants import MAX_REDIS_BUFFER_DEQUEUE_COUNT, REDIS_UPDATE_BUFFER_KEY
from litellm.litellm_core_utils.safe_json_dumps import safe_dumps
from litellm.proxy._types import DBSpendUpdateTransactions
from litellm.secret_managers.main import str_to_bool
@ -18,9 +19,6 @@ if TYPE_CHECKING:
else:
PrismaClient = Any
REDIS_UPDATE_BUFFER_KEY = "litellm_spend_update_buffer"
MAX_REDIS_BUFFER_DEQUEUE_COUNT = 100
class RedisUpdateBuffer:
"""
@ -245,9 +243,9 @@ class RedisUpdateBuffer:
# Process each field type
for field in transaction_fields:
if transaction.get(field):
for entity_id, amount in transaction[field].items():
combined_transaction[field][entity_id] = (
combined_transaction[field].get(entity_id, 0) + amount
for entity_id, amount in transaction[field].items(): # type: ignore
combined_transaction[field][entity_id] = ( # type: ignore
combined_transaction[field].get(entity_id, 0) + amount # type: ignore
)
return combined_transaction