From 707b885a3fbdfbcca965cf95992f493f83dc3423 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 15 Feb 2025 16:41:47 -0800 Subject: [PATCH] test_list_key_helper_team_filtering --- tests/proxy_admin_ui_tests/test_key_management.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/proxy_admin_ui_tests/test_key_management.py b/tests/proxy_admin_ui_tests/test_key_management.py index 55738923a34..a1c4f66ebe0 100644 --- a/tests/proxy_admin_ui_tests/test_key_management.py +++ b/tests/proxy_admin_ui_tests/test_key_management.py @@ -1011,6 +1011,7 @@ async def test_list_key_helper_team_filtering(prisma_client): from litellm.proxy.management_endpoints.key_management_endpoints import ( _list_key_helper, ) + import uuid # Setup setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client) @@ -1024,7 +1025,7 @@ async def test_list_key_helper_team_filtering(prisma_client): for i in range(3): key = await generate_key_fn( data=GenerateKeyRequest( - key_alias=f"no_team_key_{i}", + key_alias=f"no_team_key_{i}.{uuid.uuid4()}", ), user_api_key_dict=UserAPIKeyAuth( user_role=LitellmUserRoles.PROXY_ADMIN, @@ -1039,7 +1040,7 @@ async def test_list_key_helper_team_filtering(prisma_client): key = await generate_key_fn( data=GenerateKeyRequest( team_id="litellm-dashboard", - key_alias=f"dashboard_key_{i}", + key_alias=f"dashboard_key_{i}.{uuid.uuid4()}", ), user_api_key_dict=UserAPIKeyAuth( user_role=LitellmUserRoles.PROXY_ADMIN, @@ -1055,7 +1056,7 @@ async def test_list_key_helper_team_filtering(prisma_client): key = await generate_key_fn( data=GenerateKeyRequest( team_id=other_team_id, - key_alias=f"other_team_key_{i}", + key_alias=f"other_team_key_{i}.{uuid.uuid4()}", ), user_api_key_dict=UserAPIKeyAuth( user_role=LitellmUserRoles.PROXY_ADMIN,