scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

fix(proxy): run blocked-customer upserts concurrently instead of via an async generator

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mateo 2026-08-07 20:22:49 +00:00
parent 1002f16ebd
commit 3c61c4c251
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
litellm/proxy/management_endpoints/customer_endpoints.py
View file

@ -10,6 +10,7 @@ All /customer management endpoints
"""
#### END-USER/CUSTOMER MANAGEMENT ####
import asyncio
from collections.abc import Iterable, Sequence
from datetime import datetime, timedelta
from typing import Final
@ -59,20 +60,19 @@ async def _set_end_users_blocked(user_ids: Sequence[str], blocked: bool) -> tupl
raise HTTPException(status_code=500, detail={"error": CommonProxyErrors.db_not_connected_error.value})
try:
records: Final = tuple(
LiteLLM_EndUserTable.model_validate(
(
await EndUserRepository(prisma_client).table.upsert(
where={"user_id": user_id},
data={
"create": {"user_id": user_id, "blocked": blocked},
"update": {"blocked": blocked},
},
)
).model_dump()
rows: Final = await asyncio.gather(
*(
EndUserRepository(prisma_client).table.upsert(
where={"user_id": user_id},
data={
"create": {"user_id": user_id, "blocked": blocked},
"update": {"blocked": blocked},
},
)
for user_id in user_ids
)
for user_id in user_ids
)
records: Final = tuple(LiteLLM_EndUserTable.model_validate(row.model_dump()) for row in rows)
except Exception as e:
verbose_proxy_logger.error("An error occurred - %s", e)
raise HTTPException(status_code=500, detail={"error": str(e)})