scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

remove unsupported select param from find_many call

Browse source 
LiteLLM_VerificationTokenActions.find_many() does not support the
select keyword argument. Remove it and drop the corresponding test.

Co-Authored-By: Claude Haiku 4.5 <noreply@anthropic.com>
This commit is contained in:
yuneng-jiang 2026-02-25 21:40:56 -08:00
parent 0b0809a3d5
commit 06e87eea87
2 changed files with 0 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -4152,7 +4152,6 @@ async def key_aliases(
)
rows = await prisma_client.db.litellm_verificationtoken.find_many(
where=where,
select={"key_alias": True},
order=[{"key_alias": "asc"}],
skip=(page - 1) * size,
take=size,

12
tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py
View file

@ -6297,15 +6297,3 @@ async def test_key_aliases_no_search_omits_contains_filter():
assert "contains" not in json.dumps(where)
@pytest.mark.asyncio
async def test_key_aliases_select_only_key_alias():
"""Test that find_many is called with select={key_alias: True} to avoid fetching full rows."""
mock_prisma_client = AsyncMock()
mock_prisma_client.db.litellm_verificationtoken.count = AsyncMock(return_value=0)
mock_prisma_client.db.litellm_verificationtoken.find_many = AsyncMock(return_value=[])
with patch("litellm.proxy.proxy_server.prisma_client", mock_prisma_client):
await key_aliases(page=1, size=50, search=None)
find_many_kwargs = mock_prisma_client.db.litellm_verificationtoken.find_many.call_args.kwargs
assert find_many_kwargs.get("select") == {"key_alias": True}