michael/merlyn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add fallback key to mono model provider

Browse Source
This commit is contained in:
Timothy Carambat 2026-01-20 23:04:09 -08:00
parent af998ee0a7
commit b90bbc8363
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frontend/src/components/lib/MonoProviderIcon/index.jsx
View File

@ -46,7 +46,8 @@ const modelPatterns = [
{ pattern: /^claude-/i, icon: Anthropic }, { pattern: /^claude-/i, icon: Anthropic },
{ pattern: /^gemini-/i, icon: Gemini }, { pattern: /^gemini-/i, icon: Gemini },
{ pattern: /gemma/i, icon: Gemma }, { pattern: /gemma/i, icon: Gemma },
{ pattern: /^llama/i, icon: Meta }, { pattern: /llama/i, icon: Meta },
{ pattern: /^meta/i, icon: Meta },
{ {
pattern: /^(mistral|devstral|mixtral|magistral|codestral|ministral)/i, pattern: /^(mistral|devstral|mixtral|magistral|codestral|ministral)/i,
icon: Mistral, icon: Mistral,
@ -80,6 +81,8 @@ function findIconByModelName(modelName) {
* @param {string} props.provider - The provider key (for exact match) or model name (for pattern match). * @param {string} props.provider - The provider key (for exact match) or model name (for pattern match).
* @param {('exact'|'pattern')} props.match - Match mode: 'exact' for provider key, 'pattern' for model name matching. * @param {('exact'|'pattern')} props.match - Match mode: 'exact' for provider key, 'pattern' for model name matching.
* @param {number} props.size - The size of the icon. * @param {number} props.size - The size of the icon.
* @param {string} props.className - The class name of the icon.
* @param {string} props.fallbackIconKey - The key of the fallback icon to use if no icon is found.
* @returns {React.ReactNode} * @returns {React.ReactNode}
*/ */
export default function MonoProviderIcon({ export default function MonoProviderIcon({
@ -87,11 +90,14 @@ export default function MonoProviderIcon({
match = "exact", match = "exact",
size = 24, size = 24,
className = "", className = "",
fallbackIconKey = null,
}) { }) {
let Icon = null; let Icon = null;
if (match === "exact") Icon = providerIcons[provider?.toLowerCase()]; if (match === "exact") Icon = providerIcons[provider?.toLowerCase()];
else if (match === "pattern") Icon = findIconByModelName(provider); else if (match === "pattern") Icon = findIconByModelName(provider);
if (!Icon && fallbackIconKey && providerIcons[fallbackIconKey])
Icon = providerIcons[fallbackIconKey];
if (!Icon) return null; if (!Icon) return null;
return <Icon size={size} className={className} />; return <Icon size={size} className={className} />;
} }