scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test: fix testing

Browse source
This commit is contained in:
Krrish Dholakia 2025-01-22 19:18:40 -08:00
parent 2fd9de1351
commit 62cabf7178
1 changed files with 9 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
litellm/proxy/utils.py
View file

@ -2625,12 +2625,10 @@ async def update_spend( # noqa: PLR0915
)
)
if len(prisma_client.end_user_list_transactons.keys()) > 0:
asyncio.create_task(
ProxyUpdateSpend.update_end_user_spend(
n_retry_times=n_retry_times,
prisma_client=prisma_client,
proxy_logging_obj=proxy_logging_obj,
)
await ProxyUpdateSpend.update_end_user_spend(
n_retry_times=n_retry_times,
prisma_client=prisma_client,
proxy_logging_obj=proxy_logging_obj,
)
### UPDATE KEY TABLE ###
verbose_proxy_logger.debug(
@ -2789,13 +2787,11 @@ async def update_spend( # noqa: PLR0915
)
if len(prisma_client.spend_log_transactions) > 0:
asyncio.create_task(
ProxyUpdateSpend.update_spend_logs(
n_retry_times=n_retry_times,
prisma_client=prisma_client,
proxy_logging_obj=proxy_logging_obj,
db_writer_client=db_writer_client,
)
await ProxyUpdateSpend.update_spend_logs(
n_retry_times=n_retry_times,
prisma_client=prisma_client,
proxy_logging_obj=proxy_logging_obj,
db_writer_client=db_writer_client,
)