From 9b96ae6d78d3ff4b28b4acbe0a9a8ac903e6eeae Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 23 Nov 2024 21:29:22 +0530 Subject: [PATCH] test: fix test --- tests/proxy_admin_ui_tests/test_key_management.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/proxy_admin_ui_tests/test_key_management.py b/tests/proxy_admin_ui_tests/test_key_management.py index a73e33f1f..0b392a268 100644 --- a/tests/proxy_admin_ui_tests/test_key_management.py +++ b/tests/proxy_admin_ui_tests/test_key_management.py @@ -562,21 +562,23 @@ def test_team_key_generation_team_member_check(): } assert _team_key_generation_check( - UserAPIKeyAuth( + user_api_key_dict=UserAPIKeyAuth( user_role=LitellmUserRoles.INTERNAL_USER, api_key="sk-1234", team_member=Member(role="admin", user_id="test_user_id"), - ) + ), + data=GenerateKeyRequest(), ) with pytest.raises(HTTPException): _team_key_generation_check( - UserAPIKeyAuth( + user_api_key_dict=UserAPIKeyAuth( user_role=LitellmUserRoles.INTERNAL_USER, api_key="sk-1234", user_id="test_user_id", team_member=Member(role="user", user_id="test_user_id"), - ) + ), + data=GenerateKeyRequest(), )