phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) make page and page size optional params

Browse source
This commit is contained in:
ishaan-jaff 2024-02-29 13:11:07 -08:00
parent 1e8478c1d3
commit 6d30e26845
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/proxy_server.py
View file

@ -4311,11 +4311,11 @@ async def user_info(
default=False, default=False,
description="set to true to View all users. When using view_all, don't pass user_id", description="set to true to View all users. When using view_all, don't pass user_id",
), ),
page: int = fastapi.Query( page: Optional[int] = fastapi.Query(
default=0, default=0,
description="Page number for pagination. Only use when view_all is true", description="Page number for pagination. Only use when view_all is true",
), ),
page_size: int = fastapi.Query( page_size: Optional[int] = fastapi.Query(
default=25, default=25,
description="Number of items per page. Only use when view_all is true", description="Number of items per page. Only use when view_all is true",
), ),
@ -4340,6 +4340,10 @@ async def user_info(
if user_id is not None: if user_id is not None:
user_info = await prisma_client.get_data(user_id=user_id) user_info = await prisma_client.get_data(user_id=user_id)
elif view_all == True: elif view_all == True:
if page is None:
page = 0
if page_size is None:
page_size = 25
offset = (page) * page_size # default is 0 offset = (page) * page_size # default is 0
limit = page_size # default is 10 limit = page_size # default is 10
user_info = await prisma_client.get_data( user_info = await prisma_client.get_data(