🐛 fix: Yi & Solar icon display in TogetherAI provider (#35)

lobehub · Sep 13, 2024 · 2e9dafb · 2e9dafb
1 parent 6f91171
commit 2e9dafb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ModelIcon/const.ts b/src/ModelIcon/const.ts
@@ -86,7 +86,7 @@ export const modelMappings: ModelMapping[] = [
   { Icon: Minimax, keywords: ['minmax', 'abab'] },
   { Icon: Mistral, keywords: ['mistral', 'mixtral', 'codestral', 'mathstral', '/mn-'] },
   { Icon: Perplexity, keywords: ['pplx', 'sonar'] },
-  { Icon: Yi, keywords: ['^yi-', '/yi-'] },
+  { Icon: Yi, keywords: ['^yi-', '/yi-', '-yi-'] },
   { Icon: OpenRouter, keywords: ['^openrouter'] },
   { Icon: OpenChat, keywords: ['^openchat'] },
   { Icon: Aya, keywords: ['aya'] },
@@ -122,7 +122,7 @@ export const modelMappings: ModelMapping[] = [
   { Icon: Adobe, keywords: ['firefly'] },
   { Icon: Ai21, keywords: ['jamba', '^j2-', 'ai21'] },
   { Icon: InternLM, keywords: ['internlm'] },
-  { Icon: Upstage, keywords: ['^solar-'] },
+  { Icon: Upstage, keywords: ['^solar-', '/solar'] },
   { Icon: PaLM, keywords: ['palm'] },
   { Icon: Google, keywords: ['google'] },
   {