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

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-23 21:29:22 +05:30
parent 643e5b755c
commit 9b96ae6d78
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

@ -562,21 +562,23 @@ def test_team_key_generation_team_member_check():
} }
assert _team_key_generation_check( assert _team_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",
team_member=Member(role="admin", user_id="test_user_id"), team_member=Member(role="admin", user_id="test_user_id"),
) ),
data=GenerateKeyRequest(),
) )
with pytest.raises(HTTPException): with pytest.raises(HTTPException):
_team_key_generation_check( _team_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="test_user_id", user_id="test_user_id",
team_member=Member(role="user", user_id="test_user_id"), team_member=Member(role="user", user_id="test_user_id"),
) ),
data=GenerateKeyRequest(),
) )