scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

fix(proxy): include relations and format customer response on /customer/block (#49934)

Browse source
This commit is contained in:
Siddharth Srivastava 2026-11-04 02:27:38 +05:30 • committed by sidsri14
parent dfabf32cd5
commit b8a46d7c6d
2 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -192,8 +192,9 @@ async def block_user(data: BlockUsers):
"create": {"user_id": id, "blocked": True},
"update": {"blocked": True},
},
include={"litellm_budget_table": True, "object_permission": True},
)
records.append(record)
records.append(_to_customer_response(record))
await _evict_end_user_cache_keys(_end_user_cache_keys(data.user_ids))
else:
raise HTTPException(

4
tests/unit/proxy/management_endpoints/test_customer_endpoints.py
View file

@ -1289,6 +1289,10 @@ def test_customer_block_success_serializes_db_model_instance(mock_prisma_client,
assert body["blocked_users"][0]["user_id"] == "c1"
assert body["blocked_users"][0]["blocked"] is True
assert body["blocked_users"][0]["spend"] == 0.0
assert mock_prisma_client.db.litellm_endusertable.upsert.call_args.kwargs["include"] == {
"litellm_budget_table": True,
"object_permission": True,
}
def test_customer_delete_invalidates_end_user_and_registry_caches(mock_prisma_client, mock_user_api_key_auth):