From 012e79a7e2983cb95efa0cc5b17d82fc5290da33 Mon Sep 17 00:00:00 2001 From: one Date: Mon, 3 Mar 2025 17:17:25 +0800 Subject: [PATCH] fix: show correct logo for qwen-omni-turbo (#2656) * fix: show correct logo for qwen-omni-turbo * fix: match omni in the middle --- src/renderer/src/config/models.ts | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/src/renderer/src/config/models.ts b/src/renderer/src/config/models.ts index db0b09ad..723ac332 100644 --- a/src/renderer/src/config/models.ts +++ b/src/renderer/src/config/models.ts @@ -191,17 +191,14 @@ export function getModelLogo(modelId: string) { 'text-moderation': isLight ? ChatGptModelLogo : ChatGptModelLogoDakr, 'babbage-': isLight ? ChatGptModelLogo : ChatGptModelLogoDakr, 'sora-': isLight ? ChatGptModelLogo : ChatGptModelLogoDakr, - 'omni-': isLight ? ChatGptModelLogo : ChatGptModelLogoDakr, + '(^|/)omni-': isLight ? ChatGptModelLogo : ChatGptModelLogoDakr, 'Embedding-V1': isLight ? WenxinModelLogo : WenxinModelLogoDark, 'text-embedding-v': isLight ? QwenModelLogo : QwenModelLogoDark, 'text-embedding': isLight ? ChatGptModelLogo : ChatGptModelLogoDakr, 'davinci-': isLight ? ChatGptModelLogo : ChatGptModelLogoDakr, glm: isLight ? ChatGLMModelLogo : ChatGLMModelLogoDark, deepseek: isLight ? DeepSeekModelLogo : DeepSeekModelLogoDark, - qwen: isLight ? QwenModelLogo : QwenModelLogoDark, - 'qwq-': isLight ? QwenModelLogo : QwenModelLogoDark, - 'qvq-': isLight ? QwenModelLogo : QwenModelLogoDark, - Omni: isLight ? QwenModelLogo : QwenModelLogoDark, + '(qwen|qwq-|qvq-)': isLight ? QwenModelLogo : QwenModelLogoDark, gemma: isLight ? GemmaModelLogo : GemmaModelLogoDark, 'yi-': isLight ? YiModelLogo : YiModelLogoDark, llama: isLight ? LlamaModelLogo : LlamaModelLogoDark, @@ -287,7 +284,8 @@ export function getModelLogo(modelId: string) { } for (const key in logoMap) { - if (modelId.toLowerCase().includes(key)) { + const regex = new RegExp(key, 'i') + if (regex.test(modelId)) { return logoMap[key] } }