From f2799d4e870728e6e72d2028d2836dfef1db4b6a Mon Sep 17 00:00:00 2001 From: lengkejun Date: Thu, 21 May 2026 14:55:09 +0800 Subject: [PATCH] docs(proxy): refresh get_user_key_counts docstring after group_by switch --- .../proxy/management_endpoints/internal_user_endpoints.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/management_endpoints/internal_user_endpoints.py b/litellm/proxy/management_endpoints/internal_user_endpoints.py index f2b34480eaa..d185069bde7 100644 --- a/litellm/proxy/management_endpoints/internal_user_endpoints.py +++ b/litellm/proxy/management_endpoints/internal_user_endpoints.py @@ -1763,14 +1763,16 @@ async def get_user_key_counts( user_ids: Optional[List[str]] = None, ): """ - Helper function to get the count of keys for each user using Prisma's count method. + Get the count of (non-UI-session) keys for each user via a single + Prisma ``group_by`` query, avoiding an N+1 per-user ``count`` loop. Args: prisma_client: The Prisma client instance user_ids: List of user IDs to get key counts for Returns: - Dictionary mapping user_id to key count + Dictionary mapping user_id to key count. Users in ``user_ids`` with no + keys are present with a value of ``0``. """ from litellm.constants import UI_SESSION_TOKEN_TEAM_ID