phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_key_management.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-23 20:28:27 +05:30
parent 829199d538
commit 643e5b755c
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/proxy_admin_ui_tests/test_key_management.py
View file

@ -653,16 +653,18 @@ def test_personal_key_generation_check():
} }
assert _personal_key_generation_check( assert _personal_key_generation_check(
UserAPIKeyAuth( user_api_key_dict=UserAPIKeyAuth(
user_role=LitellmUserRoles.PROXY_ADMIN, api_key="sk-1234", user_id="admin" user_role=LitellmUserRoles.PROXY_ADMIN, api_key="sk-1234", user_id="admin"
) ),
data=GenerateKeyRequest(),
) )
with pytest.raises(HTTPException): with pytest.raises(HTTPException):
_personal_key_generation_check( _personal_key_generation_check(
UserAPIKeyAuth( user_api_key_dict=UserAPIKeyAuth(
user_role=LitellmUserRoles.INTERNAL_USER, user_role=LitellmUserRoles.INTERNAL_USER,
api_key="sk-1234", api_key="sk-1234",
user_id="admin", user_id="admin",
) ),
data=GenerateKeyRequest(),
) )