diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 86fd69fa6f..45f432f9df 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -3776,6 +3776,9 @@ async def delete_key_fn( ## only allow user to delete keys they own user_id = user_api_key_dict.user_id + verbose_proxy_logger.debug( + f"user_api_key_dict.user_role: {user_api_key_dict.user_role}" + ) if ( user_api_key_dict.user_role is not None and user_api_key_dict.user_role == "proxy_admin" diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 524eee6f29..74ff61abd1 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -737,8 +737,19 @@ def test_delete_key(prisma_client): delete_key_request = KeyRequest(keys=[generated_key]) + bearer_token = "Bearer sk-1234" + + request = Request(scope={"type": "http"}) + request._url = URL(url="/key/delete") + + # use generated key to auth in + result = await user_api_key_auth(request=request, api_key=bearer_token) + print(f"result: {result}") + result.user_role = "proxy_admin" # delete the key - result_delete_key = await delete_key_fn(data=delete_key_request) + result_delete_key = await delete_key_fn( + data=delete_key_request, user_api_key_dict=result + ) print("result from delete key", result_delete_key) assert result_delete_key == {"deleted_keys": [generated_key]}