diff --git a/ui/litellm-dashboard/src/components/model_dashboard.tsx b/ui/litellm-dashboard/src/components/model_dashboard.tsx index d0d4b66a09e..de92805b0bf 100644 --- a/ui/litellm-dashboard/src/components/model_dashboard.tsx +++ b/ui/litellm-dashboard/src/components/model_dashboard.tsx @@ -930,7 +930,26 @@ const ModelDashboard: React.FC = ({ _providerModels.push(key); } }); + + // Special case for cohere_chat + // we need both cohere_chat and cohere models to show on dropdown + if (providerKey == Providers.Cohere) { + console.log("adding cohere chat model") + Object.entries(modelMap).forEach(([key, value]) => { + if ( + value !== null && + typeof value === "object" && + "litellm_provider" in (value as object) && + ((value as any)["litellm_provider"] === "cohere") + ) { + _providerModels.push(key); + } + }); + } } + + + setProviderModels(_providerModels); console.log(`providerModels: ${providerModels}`); } @@ -1787,7 +1806,7 @@ const ModelDashboard: React.FC = ({