Skip to content

Commit 3a1ac96

Browse files
yuneng-jiangclaude
andcommitted
fix: pass organization_ids=None in get_users test calls
When calling get_users() directly (not via FastAPI), Query() defaults are not resolved. Pass organization_ids=None explicitly to avoid 'Query' object has no attribute 'split' error. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 7006642 commit 3a1ac96

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_litellm/proxy/management_endpoints/test_internal_user_endpoints.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -502,7 +502,7 @@ async def mock_get_user_key_counts(*args, **kwargs):
502502

503503
# Call get_users function directly with proxy admin auth
504504
admin_key = UserAPIKeyAuth(user_id="admin", user_role=LitellmUserRoles.PROXY_ADMIN)
505-
response = await get_users(page=1, page_size=1, user_api_key_dict=admin_key)
505+
response = await get_users(page=1, page_size=1, user_api_key_dict=admin_key, organization_ids=None)
506506

507507
print("user /list response: ", response)
508508

@@ -1427,15 +1427,15 @@ async def mock_get_user_key_counts(*args, **kwargs):
14271427
admin_key = UserAPIKeyAuth(user_id="admin", user_role=LitellmUserRoles.PROXY_ADMIN)
14281428

14291429
captured_where_conditions.clear()
1430-
await get_users(user_ids="test-user", page=1, page_size=1, user_api_key_dict=admin_key)
1430+
await get_users(user_ids="test-user", page=1, page_size=1, user_api_key_dict=admin_key, organization_ids=None)
14311431

14321432
assert "user_id" in captured_where_conditions
14331433
assert "contains" in captured_where_conditions["user_id"]
14341434
assert captured_where_conditions["user_id"]["contains"] == "test-user"
14351435
assert captured_where_conditions["user_id"]["mode"] == "insensitive"
14361436

14371437
captured_where_conditions.clear()
1438-
await get_users(user_ids="user1,user2,user3", page=1, page_size=1, user_api_key_dict=admin_key)
1438+
await get_users(user_ids="user1,user2,user3", page=1, page_size=1, user_api_key_dict=admin_key, organization_ids=None)
14391439

14401440
assert "user_id" in captured_where_conditions
14411441
assert "in" in captured_where_conditions["user_id"]

0 commit comments

Comments
 (0)