phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(UI) fix + test displaying number of keys an internal user owns (#6507)

Browse source 
* fix view internal user key count

* add test for /user/list

* fix test user list

* testing ui change

* ui new build
This commit is contained in:
Ishaan Jaff 2024-10-30 20:44:15 +05:30 • committed by GitHub
parent 9b9f763872
commit 8f3b634eca
31 changed files with 94 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -797,11 +797,20 @@ async def get_users(
total_count = await prisma_client.db.litellm_usertable.count(where=query) # type: ignore
# Get paginated users
users = await prisma_client.db.litellm_usertable.find_many(
_users = await prisma_client.db.litellm_usertable.find_many(
where=query, # type: ignore
skip=skip,
take=take,
)
# Add key_count to each user object directly
users = []
for user in _users:
user = user.model_dump()
key_count = await prisma_client.db.litellm_verificationtoken.count(
where={"user_id": user["user_id"]}
)
user["key_count"] = key_count
users.append(user)
# Calculate total pages
total_pages = -(-total_count // page_size) # Ceiling division