From 151f75b68fb480aa87677f8a50fb2b59d63f2a96 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 12 Feb 2024 16:55:11 -0800
Subject: [PATCH] (feat) ui - show users models when creating key

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

diff --git a/ui/litellm-dashboard/src/components/user_dashboard.tsx b/ui/litellm-dashboard/src/components/user_dashboard.tsx
index 0448b14c79..dff8046654 100644
--- a/ui/litellm-dashboard/src/components/user_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/user_dashboard.tsx
@@ -47,6 +47,7 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
 
   const token = searchParams.get("token");
   const [accessToken, setAccessToken] = useState<string | null>(null);
+  let userModels = ["gpt-4", "bedrock"];
 
   function formatUserRole(userRole: string) {
     if (!userRole) {
@@ -158,6 +159,7 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
           <CreateKey
             userID={userID}
             userRole={userRole}
+            userModels={userModels}
             accessToken={accessToken}
             data={data}
             setData={setData}