phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #9395 from BerriAI/litellm_ui_fixes_03_19_2025
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 18s
Details
Helm unit test / unit-test (push) Successful in 21s
Details

Browse source 
fix(internal_user_endpoints.py): re-introduce upsert on user not found
This commit is contained in:
Krish Dholakia 2025-03-19 22:58:32 -07:00 • committed by GitHub
commit 7cf7e80c5a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 9 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/proxy/_new_secret_config.yaml
View file

@ -8,7 +8,6 @@ model_list:
litellm_settings: litellm_settings:
num_retries: 0 num_retries: 0
callbacks: ["otel"]
router_settings: router_settings:
routing_strategy: usage-based-routing-v2 # 👈 KEY CHANGE routing_strategy: usage-based-routing-v2 # 👈 KEY CHANGE

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

@ -737,13 +737,16 @@ async def user_update(
existing_user_row = await prisma_client.db.litellm_usertable.find_first( existing_user_row = await prisma_client.db.litellm_usertable.find_first(
where={"user_id": data.user_id} where={"user_id": data.user_id}
) )
if existing_user_row is None: if existing_user_row is not None:
raise Exception(f"User not found, passed user_id={data.user_id}") existing_user_row = LiteLLM_UserTable(
existing_user_row = LiteLLM_UserTable( **existing_user_row.model_dump(exclude_none=True)
**existing_user_row.model_dump(exclude_none=True) )
)
existing_metadata = cast(Dict, getattr(existing_user_row, "metadata", {}) or {}) existing_metadata = (
cast(Dict, getattr(existing_user_row, "metadata", {}) or {})
if existing_user_row is not None
else {}
)
non_default_values = prepare_metadata_fields( non_default_values = prepare_metadata_fields(
data=data, data=data,