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

(fix) better error when no sso is setup on proxy

Browse source
This commit is contained in:
ishaan-jaff 2024-01-31 19:28:18 -08:00
parent 29ba66d78d
commit 399cb4a414
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/proxy_server.py
View file

@ -2954,6 +2954,8 @@ async def google_login(request: Request):
microsoft_client_id = os.getenv("MICROSOFT_CLIENT_ID", None) microsoft_client_id = os.getenv("MICROSOFT_CLIENT_ID", None)
google_client_id = os.getenv("GOOGLE_CLIENT_ID", None) google_client_id = os.getenv("GOOGLE_CLIENT_ID", None)
redirect_url = os.getenv("PROXY_BASE_URL", None) redirect_url = os.getenv("PROXY_BASE_URL", None)
ui_username = os.getenv("UI_USERNAME")
if redirect_url is None: if redirect_url is None:
raise ProxyException( raise ProxyException(
message="PROXY_BASE_URL not set. Set it in .env file", message="PROXY_BASE_URL not set. Set it in .env file",
@ -3015,12 +3017,19 @@ async def google_login(request: Request):
) )
with microsoft_sso: with microsoft_sso:
return await microsoft_sso.get_login_redirect() return await microsoft_sso.get_login_redirect()
else: elif ui_username is not None:
# No Google, Microsoft SSO # No Google, Microsoft SSO
# Use UI Credentials set in .env # Use UI Credentials set in .env
from fastapi.responses import HTMLResponse from fastapi.responses import HTMLResponse
return HTMLResponse(content=html_form, status_code=200) return HTMLResponse(content=html_form, status_code=200)
else:
return JSONResponse(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
content={
"error": "No SSO/Auth provider configured. See https://docs.litellm.ai/docs/proxy/ui"
},
)
@router.post( @router.post(