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

fix test_generate_and_call_with_valid_key

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-30 15:10:16 -07:00
parent 6eb8186740
commit 720a470e3a
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_key_generate_prisma.py
View file

@ -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