scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 124

[Feat] Support substring search for user_id and key_alias filters on /key/list

Browse source 
The key list endpoint used exact-match filtering for user_id and key_alias,
requiring users to type the full value (e.g. full email address) to find keys.
Switch to Prisma `contains` with case-insensitive mode, matching the pattern
already used by /v2/team/list for team_alias.
This commit is contained in:
Ryan Crabbe 2026-03-28 14:01:05 -07:00
parent 479c5e21aa
commit 33bd570d5e
No known key found for this signature in database
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/proxy/management_endpoints/key_management_endpoints.py
View file

@ -4181,13 +4181,13 @@ async def list_keys(
user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth), user_api_key_dict: UserAPIKeyAuth = Depends(user_api_key_auth),
page: int = Query(1, description="Page number", ge=1), page: int = Query(1, description="Page number", ge=1),
size: int = Query(10, description="Page size", ge=1, le=100), size: int = Query(10, description="Page size", ge=1, le=100),
user_id: Optional[str] = Query(None, description="Filter keys by user ID"), user_id: Optional[str] = Query(None, description="Filter keys by user ID. Supports partial matching (substring, case-insensitive)."),
team_id: Optional[str] = Query(None, description="Filter keys by team ID"), team_id: Optional[str] = Query(None, description="Filter keys by team ID"),
organization_id: Optional[str] = Query( organization_id: Optional[str] = Query(
None, description="Filter keys by organization ID" None, description="Filter keys by organization ID"
), ),
key_hash: Optional[str] = Query(None, description="Filter keys by key hash"), key_hash: Optional[str] = Query(None, description="Filter keys by key hash"),
key_alias: Optional[str] = Query(None, description="Filter keys by key alias"), key_alias: Optional[str] = Query(None, description="Filter keys by key alias. Supports partial matching (substring, case-insensitive)."),
return_full_object: bool = Query(False, description="Return full key object"), return_full_object: bool = Query(False, description="Return full key object"),
include_team_keys: bool = Query( include_team_keys: bool = Query(
False, description="Include all keys for teams that user is an admin of." False, description="Include all keys for teams that user is an admin of."
@ -4543,9 +4543,15 @@ def _build_key_filter_conditions(
# Base conditions for user's own keys # Base conditions for user's own keys
user_condition: Dict[str, Any] = {} user_condition: Dict[str, Any] = {}
if user_id and isinstance(user_id, str): if user_id and isinstance(user_id, str):
user_condition["user_id"] = user_id user_condition["user_id"] = {
"contains": user_id,
"mode": "insensitive",
}
if key_alias and isinstance(key_alias, str): if key_alias and isinstance(key_alias, str):
user_condition["key_alias"] = key_alias user_condition["key_alias"] = {
"contains": key_alias,
"mode": "insensitive",
}
if exclude_team_id and isinstance(exclude_team_id, str): if exclude_team_id and isinstance(exclude_team_id, str):
user_condition["team_id"] = {"not": exclude_team_id} user_condition["team_id"] = {"not": exclude_team_id}
if organization_id and isinstance(organization_id, str): if organization_id and isinstance(organization_id, str):