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

check litellm header in login on ui

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-29 17:03:04 -07:00
parent 59b91c35d7
commit 34ad3dec68
2 changed files with 14 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/proxy/proxy_server.py
View file

@ -7773,7 +7773,7 @@ async def fallback_login(request: Request):
"/login", include_in_schema=False
) # hidden since this is a helper for UI sso login
async def login(request: Request):
global premium_user
global premium_user, general_settings
try:
import multipart
except ImportError:
@ -7875,6 +7875,9 @@ async def login(request: Request):
"user_role": user_role, # this is the path without sso - we can assume only admins will use this
"login_method": "username_password",
"premium_user": premium_user,
"auth_header_name": general_settings.get(
"litellm_key_header_name", "Authorization"
),
},
master_key,
algorithm="HS256",
@ -7939,6 +7942,9 @@ async def login(request: Request):
"user_role": user_role,
"login_method": "username_password",
"premium_user": premium_user,
"auth_header_name": general_settings.get(
"litellm_key_header_name", "Authorization"
),
},
master_key,
algorithm="HS256",
@ -7987,7 +7993,7 @@ async def onboarding(invite_link: str):
- Get user from db
- Pass in user_email if set
"""
global prisma_client, master_key
global prisma_client, master_key, general_settings
if master_key is None:
raise ProxyException(
message="Master Key not set for Proxy. Please set Master Key to use Admin UI. Set `LITELLM_MASTER_KEY` in .env or set general_settings:master_key in config.yaml. https://docs.litellm.ai/docs/proxy/virtual_keys. If set, use `--detailed_debug` to debug issue.",
@ -8074,6 +8080,9 @@ async def onboarding(invite_link: str):
"user_role": user_obj.user_role,
"login_method": "username_password",
"premium_user": premium_user,
"auth_header_name": general_settings.get(
"litellm_key_header_name", "Authorization"
),
},
master_key,
algorithm="HS256",
@ -8491,6 +8500,9 @@ async def auth_callback(request: Request):
"user_role": user_role,
"login_method": "sso",
"premium_user": premium_user,
"auth_header_name": general_settings.get(
"litellm_key_header_name", "Authorization"
),
},
master_key,
algorithm="HS256",