From 071d70ec2d0ad819db8838e2299ed0850f2ef89f Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 2 Feb 2024 11:05:48 -0800 Subject: [PATCH] (fix) proxy sso - use PROXY_BASE_URL --- litellm/proxy/proxy_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c28f47de9..6a38eb68a 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -3040,7 +3040,7 @@ async def google_login(request: Request): google_client_id = os.getenv("GOOGLE_CLIENT_ID", None) # get url from request - redirect_url = str(request.base_url) + redirect_url = os.getenv("PROXY_BASE_URL", str(request.base_url)) ui_username = os.getenv("UI_USERNAME") if redirect_url.endswith("/"): @@ -3174,7 +3174,7 @@ async def auth_callback(request: Request): google_client_id = os.getenv("GOOGLE_CLIENT_ID", None) # get url from request - redirect_url = str(request.base_url) + redirect_url = os.getenv("PROXY_BASE_URL", str(request.base_url)) if redirect_url.endswith("/"): redirect_url += "sso/callback"