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

refactor(test_users.py): refactor test for user info to use mock endpoints

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-12 18:47:25 -07:00
parent 539d0962f5
commit 4bbabb4039
3 changed files with 47 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -312,7 +312,7 @@ async def user_info(
try:
if prisma_client is None:
raise Exception(
f"Database not connected. Connect a database to your proxy - https://docs.litellm.ai/docs/simple_proxy#managing-auth---virtual-keys"
"Database not connected. Connect a database to your proxy - https://docs.litellm.ai/docs/simple_proxy#managing-auth---virtual-keys"
)
## GET USER ROW ##
if user_id is not None:
@ -365,7 +365,14 @@ async def user_info(
getattr(caller_user_info, "user_role", None)
== LitellmUserRoles.PROXY_ADMIN
):
teams_2 = await prisma_client.db.litellm_teamtable.find_many()
from litellm.proxy.management_endpoints.team_endpoints import list_team
teams_2 = await list_team(
http_request=Request(
scope={"type": "http", "path": "/user/info"},
),
user_api_key_dict=user_api_key_dict,
)
else:
teams_2 = await prisma_client.get_data(
team_id_list=caller_user_info.teams,