(fix) specify UI link on proxy server

This commit is contained in:
ishaan-jaff 2024-01-29 13:59:31 -08:00
parent cce1c2fcaf
commit d2de7ba4fe

View file

@ -2959,6 +2959,11 @@ async def google_callback(code: str, request: Request):
user_id = response["user_id"] # type: ignore
litellm_dashboard_ui = "https://litellm-dashboard.vercel.app/"
# if user set LITELLM_UI_LINK in .env, use that
litellm_ui_link_in_env = os.getenv("LITELLM_UI_LINK", None)
if litellm_ui_link_in_env is not None:
litellm_dashboard_ui = litellm_ui_link_in_env
litellm_dashboard_ui += (
"?userID="
+ user_id