From 9dde7abf4c05977d8b0224810f009d728db83efd Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 28 Mar 2024 20:36:32 -0700
Subject: [PATCH] (ui) show team models on ui

---
 .../src/components/create_key_button.tsx      | 21 ++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/create_key_button.tsx b/ui/litellm-dashboard/src/components/create_key_button.tsx
index 3bb729fe69..dcf66cd063 100644
--- a/ui/litellm-dashboard/src/components/create_key_button.tsx
+++ b/ui/litellm-dashboard/src/components/create_key_button.tsx
@@ -148,11 +148,22 @@ const CreateKey: React.FC<CreateKeyProps> = ({
                   placeholder="Select models"
                   style={{ width: "100%" }}
                 >
-                  {userModels.map((model) => (
-                    <Option key={model} value={model}>
-                      {model}
-                    </Option>
-                  ))}
+
+                  {team && team.models ? (
+                    team.models.map((model: string) => (
+                      <Option key={model} value={model}>
+                        {model}
+                      </Option>
+                    ))
+                  ) : (
+                    userModels.map((model: string) => (
+                      <Option key={model} value={model}>
+                        {model}
+                      </Option>
+                    ))
+                  )}
+
+
                 </Select>
               </Form.Item>
               <Form.Item