From 37db367ba62aee809a6c817dff00cc06681f8091 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 28 Mar 2024 09:07:10 -0700
Subject: [PATCH] (fix) creating key for a team

---
 ui/litellm-dashboard/src/components/create_key_button.tsx | 7 ++++---
 ui/litellm-dashboard/src/components/user_dashboard.tsx    | 5 ++++-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/create_key_button.tsx b/ui/litellm-dashboard/src/components/create_key_button.tsx
index 3f2d8e7b5..7ce84b1f5 100644
--- a/ui/litellm-dashboard/src/components/create_key_button.tsx
+++ b/ui/litellm-dashboard/src/components/create_key_button.tsx
@@ -18,7 +18,7 @@ const { Option } = Select;
 
 interface CreateKeyProps {
   userID: string;
-  teamID: string | null;
+  team: any | null;
   userRole: string | null;
   accessToken: string;
   data: any[] | null;
@@ -27,7 +27,7 @@ interface CreateKeyProps {
 
 const CreateKey: React.FC<CreateKeyProps> = ({
   userID,
-  teamID,
+  team,
   userRole,
   accessToken,
   data,
@@ -130,7 +130,8 @@ const CreateKey: React.FC<CreateKeyProps> = ({
               <Form.Item label="Team ID" name="team_id">
                 <Input
                   placeholder="ai_team"
-                  defaultValue={teamID ? teamID : ""}
+                  defaultValue={team && team["team_alias"] ? team["team_alias"]  : ""}
+                  disabled={true}
                 />
               </Form.Item>
               <Form.Item label="Models" name="models">
diff --git a/ui/litellm-dashboard/src/components/user_dashboard.tsx b/ui/litellm-dashboard/src/components/user_dashboard.tsx
index d164a35a1..ecbd0f6f9 100644
--- a/ui/litellm-dashboard/src/components/user_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/user_dashboard.tsx
@@ -203,6 +203,8 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
     );
   }
 
+  console.log("inside user dashboard, selected team", selectedTeam);
+
   return (
     <div>
       <Grid numItems={1} className="gap-0 p-10 h-[75vh] w-full">
@@ -220,8 +222,9 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
             setData={setKeys}
           />
           <CreateKey
+            key={selectedTeam ? selectedTeam.team_id : null}
             userID={userID}
-            teamID={selectedTeam ? selectedTeam["team_id"] : null}
+            team={selectedTeam ? selectedTeam : null}
             userRole={userRole}
             userModels={userModels}
             accessToken={accessToken}