phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +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) # Test 1: Get all keys with pagination (exclude litellm-dashboard)
all_keys = [] all_keys = []
page = 1 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( result = await _list_key_helper(
prisma_client=prisma_client, prisma_client=prisma_client,
size=100, size=100,
@ -1130,7 +1132,7 @@ async def test_list_key_helper_team_filtering(prisma_client):
all_keys.extend(result["keys"]) all_keys.extend(result["keys"])
if page >= result["total_pages"]: if page >= result["total_pages"] or page >= max_pages_to_check:
break break
page += 1 page += 1