scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 129

improve Models & Providers organization in Model Management U

Browse source
This commit is contained in:
naaa760 2026-01-17 08:57:55 +05:30
parent 6e091bb26d
commit 3fdb9ba563
2 changed files with 28 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
ui/litellm-dashboard/src/components/add_model/AddModelForm.tsx
View file

@ -85,11 +85,23 @@ const AddModelForm: React.FC<AddModelFormProps> = ({
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

19
ui/litellm-dashboard/src/components/provider_info_helpers.tsx
View file

@ -211,6 +211,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<string> => {
let providerKey = provider;
console.log(`Provider key: ${providerKey}`);
@ -227,7 +232,9 @@ export const getProviderModels = (provider: Providers, modelMap: any): Array<str
litellmProvider === custom_llm_provider ||
(typeof litellmProvider === "string" && litellmProvider.includes(custom_llm_provider))
) {
providerModels.push(key);
if (isValidModelName(key)) {
providerModels.push(key);
}
}
}
});
@ -242,7 +249,9 @@ export const getProviderModels = (provider: Providers, modelMap: any): Array<str
"litellm_provider" in (value as object) &&
(value as any)["litellm_provider"] === "cohere_chat"
) {
providerModels.push(key);
if (isValidModelName(key)) {
providerModels.push(key);
}
}
});
}
@ -258,11 +267,13 @@ export const getProviderModels = (provider: Providers, modelMap: any): Array<str
"litellm_provider" in (value as object) &&
(value as any)["litellm_provider"] === "sagemaker_chat"
) {
providerModels.push(key);
if (isValidModelName(key)) {
providerModels.push(key);
}
}
});
}
}
return providerModels;
return providerModels.sort();
};