phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix test_list_key_helper_team_filtering

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-08 17:21:32 -08:00
parent de0467b8e6
commit 83062b892b
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/proxy_admin_ui_tests/test_key_management.py
View file

@ -1116,7 +1116,9 @@ async def test_list_key_helper_team_filtering(prisma_client):
# Test 1: Get all keys with pagination (exclude litellm-dashboard)
all_keys = []
page = 1
while True:
max_pages_to_check = 3 # Only check the first 3 pages
while page <= max_pages_to_check:
result = await _list_key_helper(
prisma_client=prisma_client,
size=100,
@ -1130,7 +1132,7 @@ async def test_list_key_helper_team_filtering(prisma_client):
all_keys.extend(result["keys"])
if page >= result["total_pages"]:
if page >= result["total_pages"] or page >= max_pages_to_check:
break
page += 1