From 10a3304e6a55d4f3b187668aa46127eb0de583c2 Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Mon, 23 Feb 2026 20:18:39 -0800
Subject: [PATCH] refactor: simplify onboarding page to use OnboardingForm
 component

---
 .../src/app/onboarding/page.tsx               | 145 ++----------------
 1 file changed, 9 insertions(+), 136 deletions(-)

diff --git a/ui/litellm-dashboard/src/app/onboarding/page.tsx b/ui/litellm-dashboard/src/app/onboarding/page.tsx
index 3bdf57907ee..d7840a7be3a 100644
--- a/ui/litellm-dashboard/src/app/onboarding/page.tsx
+++ b/ui/litellm-dashboard/src/app/onboarding/page.tsx
@@ -1,149 +1,22 @@
 "use client";
-import React, { Suspense, useEffect, useState } from "react";
+import React, { Suspense } from "react";
 import { useSearchParams } from "next/navigation";
-import { Card, Title, Text, TextInput, Callout, Button, Grid, Col } from "@tremor/react";
-import { RiCheckboxCircleLine } from "@remixicon/react";
-import {
-  getOnboardingCredentials,
-  claimOnboardingToken,
-  getUiConfig,
-  getProxyBaseUrl,
-} from "@/components/networking";
-import { jwtDecode } from "jwt-decode";
-import { Form, Button as Button2 } from "antd";
-import { getCookie } from "@/utils/cookieUtils";
+import { OnboardingForm } from "./OnboardingForm";
 
 function OnboardingContent() {
-  const [form] = Form.useForm();
   const searchParams = useSearchParams()!;
-  const token = getCookie("token");
-  const inviteID = searchParams.get("invitation_id");
   const action = searchParams.get("action");
-  const [accessToken, setAccessToken] = useState<string | null>(null);
-  const [defaultUserEmail, setDefaultUserEmail] = useState<string>("");
-  const [userEmail, setUserEmail] = useState<string>("");
-  const [userID, setUserID] = useState<string | null>(null);
-  const [loginUrl, setLoginUrl] = useState<string>("");
-  const [jwtToken, setJwtToken] = useState<string>("");
-  const [getUiConfigLoading, setGetUiConfigLoading] = useState<boolean>(true);
-
-  useEffect(() => {
-    getUiConfig().then((data) => {
-      // get the information for constructing the proxy base url, and then set the token and auth loading
-      console.log("ui config in onboarding.tsx:", data);
-      setGetUiConfigLoading(false);
-    });
-  }, []);
-
-  useEffect(() => {
-    if (!inviteID || getUiConfigLoading) {
-      // wait for the ui config to be loaded
-      return;
-    }
-
-    getOnboardingCredentials(inviteID).then((data) => {
-      const login_url = data.login_url;
-      console.log("login_url:", login_url);
-      setLoginUrl(login_url);
-
-      const token = data.token;
-      const decoded = jwtDecode(token) as { [key: string]: any };
-      setJwtToken(token);
-
-      console.log("decoded:", decoded);
-      setAccessToken(decoded.key);
-
-      console.log("decoded user email:", decoded.user_email);
-      const user_email = decoded.user_email;
-      setUserEmail(user_email);
-
-      const user_id = decoded.user_id;
-      setUserID(user_id);
-    });
-  }, [inviteID, getUiConfigLoading]);
-
-  const handleSubmit = (formValues: Record<string, any>) => {
-    console.log("in handle submit. accessToken:", accessToken, "token:", jwtToken, "formValues:", formValues);
-    if (!accessToken || !jwtToken) {
-      return;
-    }
-
-    formValues.user_email = userEmail;
-
-    if (!userID || !inviteID) {
-      return;
-    }
-    claimOnboardingToken(accessToken, inviteID, userID, formValues.password).then((data) => {
-      // set cookie "token" to jwtToken
-      document.cookie = "token=" + jwtToken;
-      
-      const proxyBaseUrl = getProxyBaseUrl();
-      console.log("proxyBaseUrl:", proxyBaseUrl);
-      
-      // Construct the full redirect URL using the proxyBaseUrl which includes the server root path
-      let redirectUrl = proxyBaseUrl ? `${proxyBaseUrl}/ui/?login=success` : "/ui/?login=success";
-      console.log("redirecting to:", redirectUrl);
-
-      window.location.href = redirectUrl;
-    });
-
-    // redirect to login page
-  };
-  return (
-    <div className="mx-auto w-full max-w-md mt-10">
-      <Card>
-        <Title className="text-sm mb-5 text-center">🚅 LiteLLM</Title>
-        <Title className="text-xl">{action === "reset_password" ? "Reset Password" : "Sign up"}</Title>
-        <Text>
-          {action === "reset_password"
-            ? "Reset your password to access Admin UI."
-            : "Claim your user account to login to Admin UI."}
-        </Text>
-
-        {action !== "reset_password" && (
-          <Callout className="mt-4" title="SSO" icon={RiCheckboxCircleLine} color="sky">
-            <Grid numItems={2} className="flex justify-between items-center">
-              <Col>SSO is under the Enterprise Tier.</Col>
-
-              <Col>
-                <Button variant="primary" className="mb-2">
-                  <a href="https://forms.gle/W3U4PZpJGFHWtHyA9" target="_blank">
-                    Get Free Trial
-                  </a>
-                </Button>
-              </Col>
-            </Grid>
-          </Callout>
-        )}
-
-        <Form className="mt-10 mb-5 mx-auto" layout="vertical" onFinish={handleSubmit}>
-          <>
-            <Form.Item label="Email Address" name="user_email">
-              <TextInput type="email" disabled={true} value={userEmail} defaultValue={userEmail} className="max-w-md" />
-            </Form.Item>
-
-            <Form.Item
-              label="Password"
-              name="password"
-              rules={[{ required: true, message: "password required to sign up" }]}
-              help={action === "reset_password" ? "Enter your new password" : "Create a password for your account"}
-            >
-              <TextInput placeholder="" type="password" className="max-w-md" />
-            </Form.Item>
-          </>
-
-          <div className="mt-10">
-            <Button2 htmlType="submit">{action === "reset_password" ? "Reset Password" : "Sign Up"}</Button2>
-          </div>
-        </Form>
-      </Card>
-    </div>
-  );
+  const variant = action === "reset_password" ? "reset_password" : "signup";
+  return <OnboardingForm variant={variant} />;
 }
 
 export default function Onboarding() {
   return (
-    <Suspense fallback={<div className="flex items-center justify-center min-h-screen">Loading...</div>}>
+    <Suspense
+      fallback={
+        <div className="flex items-center justify-center min-h-screen">Loading...</div>
+      }
+    >
       <OnboardingContent />
     </Suspense>
   );