diff --git a/ui/litellm-dashboard/src/components/model_dashboard.tsx b/ui/litellm-dashboard/src/components/model_dashboard.tsx index 27608f0ae48..bf93e341634 100644 --- a/ui/litellm-dashboard/src/components/model_dashboard.tsx +++ b/ui/litellm-dashboard/src/components/model_dashboard.tsx @@ -220,6 +220,9 @@ const handleSubmit = async ( else if (key === "base_model") { // Add key-value pair to model_info dictionary modelInfoObj[key] = value; + } + else if (key === "custom_model_name") { + litellmParamsObj["model"] = value; } else if (key == "litellm_extra_params") { console.log("litellm_extra_params:", value); let litellmExtraParams = {}; @@ -1718,26 +1721,43 @@ const ModelDashboard: React.FC = ({ + - { (selectedProvider === Providers.Azure) || (selectedProvider === Providers.OpenAI_Compatible) || (selectedProvider === Providers.Ollama) ? ( - - ) : providerModels.length > 0 ? ( - - {providerModels.map((model, index) => ( - - {model} - - ))} - - ) : ( - - )} + + Custom Model Name (Enter below) + {providerModels.map((model, index) => ( + + {model} + + ))} + + + prevValues.model !== currentValues.model} + > + {({ getFieldValue }) => { + const selectedModels = getFieldValue('model') || []; + return selectedModels.includes('custom') && ( + + + + ) + }} + +