(ui) show team models on ui

2024-03-28 20:36:32 -07:00 · 2024-03-28 20:36:32 -07:00 · 9dde7abf4c
commit 9dde7abf4c
parent 69ae445e65
1 changed files with 16 additions and 5 deletions
--- 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