From 720a470e3ae38a692ae5f51460ff6e7ac1fb0a16 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 30 May 2024 15:10:16 -0700 Subject: [PATCH] fix test_generate_and_call_with_valid_key --- litellm/tests/test_key_generate_prisma.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 6be720ffd1..c818b4f4a2 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -209,7 +209,7 @@ def test_generate_and_call_with_valid_key(prisma_client, api_route): await litellm.proxy.proxy_server.prisma_client.connect() from litellm.proxy.proxy_server import user_api_key_cache - request = NewUserRequest(user_role="app_owner") + request = NewUserRequest(user_role=LitellmUserRoles.INTERNAL_USER) key = await new_user(request) print(key) user_id = key.user_id @@ -218,7 +218,7 @@ def test_generate_and_call_with_valid_key(prisma_client, api_route): new_user_info = await user_info(user_id=user_id) new_user_info = new_user_info["user_info"] print("new_user_info=", new_user_info) - assert new_user_info.user_role == "app_owner" + assert new_user_info.user_role == LitellmUserRoles.INTERNAL_USER assert new_user_info.user_id == user_id generated_key = key.key