From c45ede7187782abf8b9476803314cf32abe66346 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 9 Sep 2025 21:31:34 -0700 Subject: [PATCH] test: update test --- tests/proxy_unit_tests/test_key_generate_prisma.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py index 983f68b89e..22bf404342 100644 --- a/tests/proxy_unit_tests/test_key_generate_prisma.py +++ b/tests/proxy_unit_tests/test_key_generate_prisma.py @@ -2844,7 +2844,12 @@ async def test_update_user_role(prisma_client): await user_update( data=UpdateUserRequest( user_id=key.user_id, user_role=LitellmUserRoles.PROXY_ADMIN - ) + ), + user_api_key_dict=UserAPIKeyAuth( + user_role=LitellmUserRoles.PROXY_ADMIN, + api_key="sk-1234", + user_id="1234", + ), ) # await asyncio.sleep(3) @@ -2882,7 +2887,12 @@ async def test_update_user_unit_test(prisma_client): tpm_limit=100, rpm_limit=100, metadata={"very-new-metadata": "something"}, - ) + ), + user_api_key_dict=UserAPIKeyAuth( + user_role=LitellmUserRoles.PROXY_ADMIN, + api_key="sk-1234", + user_id="1234", + ), ) print("user_info", user_info)