From c8512e57ca13c6cdfb68e40689acce1fb6a67009 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 2 Jun 2025 15:47:11 -0700
Subject: [PATCH] fix(page.tsx): route correctly to litellm dashboard from new
 user login

---
 ui/litellm-dashboard/src/app/onboarding/page.tsx   | 10 +++++++++-
 ui/litellm-dashboard/src/components/networking.tsx |  6 ++++++
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/ui/litellm-dashboard/src/app/onboarding/page.tsx b/ui/litellm-dashboard/src/app/onboarding/page.tsx
index 382eee52a15..bd5a1096629 100644
--- a/ui/litellm-dashboard/src/app/onboarding/page.tsx
+++ b/ui/litellm-dashboard/src/app/onboarding/page.tsx
@@ -18,6 +18,7 @@ import {
   getOnboardingCredentials,
   claimOnboardingToken,
   getUiConfig,
+  getProxyBaseUrl
 } from "@/components/networking";
 import { jwtDecode } from "jwt-decode";
 import { Form, Button as Button2, message } from "antd";
@@ -100,7 +101,14 @@ export default function Onboarding() {
       document.cookie = "token=" + jwtToken;
       console.log("redirecting to:", litellm_dashboard_ui);
 
-      window.location.href = litellm_dashboard_ui;
+      const proxyBaseUrl = getProxyBaseUrl();
+      console.log("proxyBaseUrl:", proxyBaseUrl);
+
+      if (proxyBaseUrl) {
+        window.location.href = proxyBaseUrl + litellm_dashboard_ui;
+      } else {
+        window.location.href = litellm_dashboard_ui;
+      }
     });
 
     // redirect to login page
diff --git a/ui/litellm-dashboard/src/components/networking.tsx b/ui/litellm-dashboard/src/components/networking.tsx
index 90c945cd20b..8870b4ad7be 100644
--- a/ui/litellm-dashboard/src/components/networking.tsx
+++ b/ui/litellm-dashboard/src/components/networking.tsx
@@ -10,6 +10,8 @@ import { EmailEventSettingsResponse, EmailEventSettingsUpdateRequest } from "./e
 
 const isLocal = process.env.NODE_ENV === "development";
 export const defaultProxyBaseUrl = isLocal ? "http://localhost:4000" : null;
+const defaultServerRootPath = "/";
+export let serverRootPath = defaultServerRootPath;
 export let proxyBaseUrl = defaultProxyBaseUrl;
 if (isLocal != true) {
   console.log = function() {};
@@ -30,6 +32,10 @@ const updateProxyBaseUrl = (serverRootPath: string, receivedProxyBaseUrl: string
   console.log("Updated proxyBaseUrl:", proxyBaseUrl);
 };
 
+const updateServerRootPath = (receivedServerRootPath: string) => {
+  serverRootPath = receivedServerRootPath;
+};
+
 export const getProxyBaseUrl = () => {
   return proxyBaseUrl;
 };