From 145aa8854fb2b755aede2453a06db7ed96d3a5a5 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 14 Dec 2024 12:13:49 -0800
Subject: [PATCH] ui - new build

---
 ui/litellm-dashboard/src/app/model_hub/page.tsx        | 2 +-
 ui/litellm-dashboard/src/app/onboarding/page.tsx       | 2 +-
 ui/litellm-dashboard/src/app/page.tsx                  | 2 +-
 ui/litellm-dashboard/src/components/user_dashboard.tsx | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/ui/litellm-dashboard/src/app/model_hub/page.tsx b/ui/litellm-dashboard/src/app/model_hub/page.tsx
index cca9f877b7..35abf9d12e 100644
--- a/ui/litellm-dashboard/src/app/model_hub/page.tsx
+++ b/ui/litellm-dashboard/src/app/model_hub/page.tsx
@@ -5,7 +5,7 @@ import { modelHubCall } from "@/components/networking";
 import ModelHub from "@/components/model_hub";
 
 export default function PublicModelHub() {
-  const searchParams = useSearchParams();
+  const searchParams = useSearchParams()!;
   const key = searchParams.get("key");
   const [accessToken, setAccessToken] = useState<string | null>(null);
 
diff --git a/ui/litellm-dashboard/src/app/onboarding/page.tsx b/ui/litellm-dashboard/src/app/onboarding/page.tsx
index 373944a98d..ff301d7a47 100644
--- a/ui/litellm-dashboard/src/app/onboarding/page.tsx
+++ b/ui/litellm-dashboard/src/app/onboarding/page.tsx
@@ -31,7 +31,7 @@ function getCookie(name: string) {
 
 export default function Onboarding() {
   const [form] = Form.useForm();
-  const searchParams = useSearchParams();
+  const searchParams = useSearchParams()!;
   const token = getCookie('token');
   const inviteID = searchParams.get("invitation_id");
   const [accessToken, setAccessToken] = useState<string | null>(null);
diff --git a/ui/litellm-dashboard/src/app/page.tsx b/ui/litellm-dashboard/src/app/page.tsx
index 94af125740..4f1c6411f2 100644
--- a/ui/litellm-dashboard/src/app/page.tsx
+++ b/ui/litellm-dashboard/src/app/page.tsx
@@ -76,7 +76,7 @@ const CreateKeyPage = () => {
   });
 
   const [showSSOBanner, setShowSSOBanner] = useState<boolean>(true);
-  const searchParams = useSearchParams();
+  const searchParams = useSearchParams()!;
   const [modelData, setModelData] = useState<any>({ data: [] });
   const userID = searchParams.get("userID");
   const invitation_id = searchParams.get("invitation_id");
diff --git a/ui/litellm-dashboard/src/components/user_dashboard.tsx b/ui/litellm-dashboard/src/components/user_dashboard.tsx
index 9c213bc3c5..ed1a18be4c 100644
--- a/ui/litellm-dashboard/src/components/user_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/user_dashboard.tsx
@@ -83,7 +83,7 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
   );
 
   // Assuming useSearchParams() hook exists and works in your setup
-  const searchParams = useSearchParams();
+  const searchParams = useSearchParams()!;
   const viewSpend = searchParams.get("viewSpend");
   const router = useRouter();