scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix: remove duplicate prisma_client guard and rollback user on password update failure

Browse source
This commit is contained in:
Smeet Agrawal 2026-04-12 14:04:09 +05:30
parent 43a06ab573
commit c94b9b1c8d
1 changed files with 11 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/management_endpoints/internal_user_endpoints.py
View file

@ -421,12 +421,6 @@ async def new_user(
raise HTTPException(
status_code=400, detail=CommonProxyErrors.db_not_connected_error.value
)
if prisma_client is None:
raise HTTPException(
status_code=500,
detail=CommonProxyErrors.db_not_connected_error.value,
)
# Check for duplicate user_id or email
await _check_duplicate_user_id(data.user_id, prisma_client)
await _check_duplicate_user_email(data.user_email, prisma_client)
@ -477,10 +471,17 @@ async def new_user(
status_code=500,
detail="User was created but user_id was not returned — password could not be stored.",
)
await prisma_client.db.litellm_usertable.update(
where={"user_id": created_user_id},
data={"password": hashed_password},
)
try:
await prisma_client.db.litellm_usertable.update(
where={"user_id": created_user_id},
data={"password": hashed_password},
)
except Exception as pwd_err:
# Roll back the user row so the caller can safely retry.
await prisma_client.db.litellm_usertable.delete(
where={"user_id": created_user_id}
)
raise pwd_err
# Admin UI Logic
# Add User to Team and Organization
# if team_id passed add this user to the team