From ddebd6c7e6e69495f75022edeade45733dab53e7 Mon Sep 17 00:00:00 2001 From: naaa760 Date: Sat, 17 Jan 2026 08:57:55 +0530 Subject: [PATCH] improve Models & Providers organization in Model Management U --- .../src/components/add_model/AddModelForm.tsx | 14 +++++++++++++- .../src/components/provider_info_helpers.tsx | 19 +++++++++++++++---- 2 files changed, 28 insertions(+), 5 deletions(-) diff --git a/ui/litellm-dashboard/src/components/add_model/AddModelForm.tsx b/ui/litellm-dashboard/src/components/add_model/AddModelForm.tsx index 2b3f23a35ae..178a195235d 100644 --- a/ui/litellm-dashboard/src/components/add_model/AddModelForm.tsx +++ b/ui/litellm-dashboard/src/components/add_model/AddModelForm.tsx @@ -85,11 +85,23 @@ const AddModelForm: React.FC = ({ fetchModelAccessGroups(); }, [accessToken]); + const nonProviderIdentifiers = new Set([ + "AIOHTTP_OPENAI", + "CUSTOM_OPENAI", + "OPENAI_LIKE", + "OpenAI_Text", + "OpenAI_Compatible", + "OpenAI_Text_Compatible", + ]); + const sortedProviderMetadata: ProviderCreateInfo[] = useMemo(() => { if (!providerMetadata) { return []; } - return [...providerMetadata].sort((a, b) => a.provider_display_name.localeCompare(b.provider_display_name)); + const filtered = providerMetadata.filter((provider) => { + return !nonProviderIdentifiers.has(provider.provider); + }); + return [...filtered].sort((a, b) => a.provider_display_name.localeCompare(b.provider_display_name)); }, [providerMetadata]); const providerMetadataErrorText = providerMetadataError diff --git a/ui/litellm-dashboard/src/components/provider_info_helpers.tsx b/ui/litellm-dashboard/src/components/provider_info_helpers.tsx index bf9e9449d8e..825c8dc5412 100644 --- a/ui/litellm-dashboard/src/components/provider_info_helpers.tsx +++ b/ui/litellm-dashboard/src/components/provider_info_helpers.tsx @@ -368,6 +368,11 @@ export const getPlaceholder = (selectedProvider: string): string => { } }; +const isValidModelName = (modelName: string): boolean => { + const resolutionOnlyPattern = /^\d{3,4}-x-\d{3,4}\/dall-e-[23]$/; + return !resolutionOnlyPattern.test(modelName); +}; + export const getProviderModels = (provider: Providers, modelMap: any): Array => { let providerKey = provider; console.log(`Provider key: ${providerKey}`); @@ -384,7 +389,9 @@ export const getProviderModels = (provider: Providers, modelMap: any): Array