scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

docs(proxy): refresh get_user_key_counts docstring after group_by switch

Browse source
This commit is contained in:
lengkejun 2026-05-21 14:55:09 +08:00 • committed by silencedoctor
parent 880ee42c42
commit eba2685ece
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/management_endpoints/internal_user_endpoints.py
View file

@ -1894,14 +1894,16 @@ async def get_user_key_counts(
user_ids: list[str] | None = None,
) -> Mapping[str, int]:
"""
Helper function to get the count of keys for each user using Prisma's group_by 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