From 457436e12e214b183cf739e437db8510742fc16f Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 1 Aug 2024 09:58:37 -0700
Subject: [PATCH] ui fix entering custom model names for azure

---
 .../src/components/model_dashboard.tsx        | 22 ++++++++++++-------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/model_dashboard.tsx b/ui/litellm-dashboard/src/components/model_dashboard.tsx
index bf93e3416..f96916cfd 100644
--- a/ui/litellm-dashboard/src/components/model_dashboard.tsx
+++ b/ui/litellm-dashboard/src/components/model_dashboard.tsx
@@ -1730,14 +1730,20 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
                     rules={[{ required: true, message: "Required" }]}
                     noStyle
                   >
-                    <MultiSelect>
-                    <MultiSelectItem value="custom">Custom Model Name (Enter below)</MultiSelectItem>
-                      {providerModels.map((model, index) => (
-                        <MultiSelectItem key={index} value={model}>
-                          {model}
-                        </MultiSelectItem>
-                      ))}
-                    </MultiSelect>
+                     { (selectedProvider === Providers.Azure) || (selectedProvider === Providers.OpenAI_Compatible) || (selectedProvider === Providers.Ollama) ? (
+                      <TextInput placeholder={getPlaceholder(selectedProvider.toString())} />
+                    ) : providerModels.length > 0 ? (
+                      <MultiSelect>
+                      <MultiSelectItem value="custom">Custom Model Name (Enter below)</MultiSelectItem>
+                        {providerModels.map((model, index) => (
+                          <MultiSelectItem key={index} value={model}>
+                            {model}
+                          </MultiSelectItem>
+                        ))}
+                      </MultiSelect>
+                    ) : (
+                      <TextInput placeholder={getPlaceholder(selectedProvider.toString())} />
+                    )}
                   </Form.Item>
 
                   <Form.Item