From cf4136665392b8bcdacd04f9ded5559dadbab76d Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 2 Feb 2024 18:41:40 -0800
Subject: [PATCH] (fix) ui

---
 ui/litellm-dashboard/src/components/user_dashboard.tsx | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/user_dashboard.tsx b/ui/litellm-dashboard/src/components/user_dashboard.tsx
index dccbd21dbf..ab9be7fcd3 100644
--- a/ui/litellm-dashboard/src/components/user_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/user_dashboard.tsx
@@ -8,6 +8,9 @@ import EnterProxyUrl from "./enter_proxy_url";
 import { useSearchParams } from "next/navigation";
 import { jwtDecode } from "jwt-decode";
 
+const proxyBaseUrl = process.env.PROXY_BASE_URL || null;
+console.log("Proxy Base URL:", proxyBaseUrl);
+
 const UserDashboard = () => {
   const [data, setData] = useState<null | any[]>(null); // Keep the initialization of state here
   // Assuming useSearchParams() hook exists and works in your setup
@@ -53,8 +56,8 @@ const UserDashboard = () => {
 
   
     // Now you can construct the full URL
-    const url = `/sso/key/generate`;
-
+    const url = proxyBaseUrl ? `${proxyBaseUrl}/sso/key/generate` : `/sso/key/generate`;
+    console.log("Full URL:", url);
     window.location.href = url;
 
     return null;