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

(fix) proxy - allow _has_user_setup_sso to override allow_auth

Browse source
This commit is contained in:
ishaan-jaff 2024-02-01 08:50:38 -08:00
parent ee50e97c98
commit fe4138853c
1 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

19
litellm/proxy/proxy_server.py
View file

@ -548,6 +548,7 @@ async def user_api_key_auth(
or route.startswith("/model/") or route.startswith("/model/")
) )
and (not is_master_key_valid) and (not is_master_key_valid)
and (not _has_user_setup_sso())
and (not general_settings.get("allow_user_auth", False)) and (not general_settings.get("allow_user_auth", False))
): ):
# enters this block when allow_user_auth is set to False # enters this block when allow_user_auth is set to False
@ -3869,6 +3870,24 @@ async def get_routes():
return {"routes": routes} return {"routes": routes}
def _has_user_setup_sso():
"""
Check if the user has set up single sign-on (SSO) by verifying the presence of Microsoft client ID, Google client ID, and UI username environment variables.
Returns a boolean indicating whether SSO has been set up.
"""
microsoft_client_id = os.getenv("MICROSOFT_CLIENT_ID", None)
google_client_id = os.getenv("GOOGLE_CLIENT_ID", None)
ui_username = os.getenv("UI_USERNAME")
sso_setup = (
(microsoft_client_id is not None)
or (google_client_id is not None)
or (ui_username is not None)
)
return sso_setup
@router.on_event("shutdown") @router.on_event("shutdown")
async def shutdown_event(): async def shutdown_event():
global prisma_client, master_key, user_custom_auth, user_custom_key_generate global prisma_client, master_key, user_custom_auth, user_custom_key_generate