From 932d07c5e67806ea37df6b99a5749bfffdf6761e Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 1 Feb 2024 13:09:28 -0800
Subject: [PATCH] (fix) proxy - remove PROXY_BASE_URL

---
 litellm/proxy/proxy_server.py | 20 ++++----------------
 1 file changed, 4 insertions(+), 16 deletions(-)

diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 2a0495dbd..df4dec756 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -3059,14 +3059,7 @@ async def login(request: Request):
         if litellm_ui_link_in_env is not None:
             litellm_dashboard_ui = litellm_ui_link_in_env
 
-        litellm_dashboard_ui += (
-            "?userID="
-            + user_id
-            + "&accessToken="
-            + key
-            + "&proxyBaseUrl="
-            + os.getenv("PROXY_BASE_URL")
-        )
+        litellm_dashboard_ui += "?userID=" + user_id + "&accessToken=" + key
         return RedirectResponse(url=litellm_dashboard_ui)
     else:
         raise ProxyException(
@@ -3084,14 +3077,9 @@ async def auth_callback(request: Request):
     microsoft_client_id = os.getenv("MICROSOFT_CLIENT_ID", None)
     google_client_id = os.getenv("GOOGLE_CLIENT_ID", None)
 
-    redirect_url = os.getenv("PROXY_BASE_URL", None)
-    if redirect_url is None:
-        raise ProxyException(
-            message="PROXY_BASE_URL not set. Set it in .env file",
-            type="auth_error",
-            param="PROXY_BASE_URL",
-            code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-        )
+    # get url from request
+    redirect_url = str(request.base_url)
+
     if redirect_url.endswith("/"):
         redirect_url += "sso/callback"
     else: