From fe15df26f2cd2ffa856bfc37864b8bc743d74592 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 4 Jul 2024 15:40:08 -0700
Subject: [PATCH] fix invite link flow

---
 .../src/components/user_dashboard.tsx                 | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/ui/litellm-dashboard/src/components/user_dashboard.tsx b/ui/litellm-dashboard/src/components/user_dashboard.tsx
index ec66bcea0f..2d2fb0dc1e 100644
--- a/ui/litellm-dashboard/src/components/user_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/user_dashboard.tsx
@@ -12,6 +12,7 @@ import ViewKeyTable from "./view_key_table";
 import ViewUserSpend from "./view_user_spend";
 import ViewUserTeam from "./view_user_team";
 import DashboardTeam from "./dashboard_default_team";
+import Onboarding from "../app/onboarding/page";
 import { useSearchParams, useRouter } from "next/navigation";
 import { jwtDecode } from "jwt-decode";
 import { Typography } from "antd";
@@ -76,6 +77,8 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
 
   const token = getCookie('token');
 
+  const invitation_id = searchParams.get("invitation_id");
+
   const [accessToken, setAccessToken] = useState<string | null>(null);
   const [teamSpend, setTeamSpend] = useState<number | null>(null);
   const [userModels, setUserModels] = useState<string[]>([]);
@@ -257,7 +260,13 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
   }, [selectedTeam]);
 
   if (userID == null || token == null) {
-    // Now you can construct the full URL
+    // user is not logged in as yet 
+    if (invitation_id != null) {
+      return (
+        <Onboarding></Onboarding>
+
+      )
+    }
     const url = proxyBaseUrl
       ? `${proxyBaseUrl}/sso/key/generate`
       : `/sso/key/generate`;