💄 style: Fix model const

lobehub · Aug 27, 2024 · ddba21f · ddba21f
1 parent 4873aaa
commit ddba21f
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 14 deletions.
diff --git a/src/Groq/style.ts b/src/Groq/style.ts
@@ -1,4 +1,4 @@
 export const TITLE = 'Groq';
 export const TEXT_MULTIPLE = 0.75;
-export const SPACE_MULTIPLE = 0.1;
+export const SPACE_MULTIPLE = 0.2;
 export const COLOR_PRIMARY = '#F55036';
diff --git a/src/ModelIcon/const.ts b/src/ModelIcon/const.ts
@@ -5,11 +5,13 @@ import Ai21 from '@/Ai21';
 import Ai360 from '@/Ai360';
 import AiMass from '@/AiMass';
 import Aws from '@/Aws';
+import Aya from '@/Aya';
 import Azure from '@/Azure';
 import Baichuan from '@/Baichuan';
 import ByteDance from '@/ByteDance';
 import ChatGLM from '@/ChatGLM';
 import Claude from '@/Claude';
+import CodeGeeX from '@/CodeGeeX';
 import Cohere from '@/Cohere';
 import Dbrx from '@/Dbrx';
 import DeepSeek from '@/DeepSeek';
@@ -55,10 +57,11 @@ export interface ModelMapping {
 
 // Create a mapping of model keywords to their respective Icon functions
 export const modelMappings: ModelMapping[] = [
-  { Icon: OpenAI, keywords: ['^gpt-3'], props: { type: 'gpt3' } },
-  { Icon: OpenAI, keywords: ['^gpt-4'], props: { type: 'gpt4' } },
-  { Icon: OpenAI, keywords: ['^gpt', 'openai', 'dalle'] },
-  { Icon: ChatGLM, keywords: ['glm', 'chatglm'] },
+  { Icon: OpenAI, keywords: ['gpt-3'], props: { type: 'gpt3' } },
+  { Icon: OpenAI, keywords: ['gpt-4'], props: { type: 'gpt4' } },
+  { Icon: OpenAI, keywords: ['gpt', 'openai', 'dalle', 'text-embedding-'] },
+  { Icon: ChatGLM, keywords: ['glm-', 'chatglm'] },
+  { Icon: CodeGeeX, keywords: ['^codegeex'] },
   { Icon: DeepSeek, keywords: ['deepseek'] },
   { Icon: Claude, keywords: ['claude'] },
   { Icon: Aws, keywords: ['titan'] },
@@ -67,25 +70,20 @@ export const modelMappings: ModelMapping[] = [
   { Icon: Gemini, keywords: ['gemini'] },
   { Icon: Gemma, keywords: ['gemma'] },
   { Icon: Moonshot, keywords: ['moonshot'] },
-  {
-    Icon: Tongyi,
-    keywords: ['qwen'],
-  },
+  { Icon: Tongyi, keywords: ['qwen'] },
   { Icon: Minimax, keywords: ['minmax', 'abab'] },
-  { Icon: Mistral, keywords: ['mistral', 'mixtral'] },
+  { Icon: Mistral, keywords: ['mistral', 'mixtral', 'codestral', 'mathstral'] },
   { Icon: Perplexity, keywords: ['pplx', 'sonar'] },
   { Icon: Yi, keywords: ['^yi-'] },
   { Icon: OpenRouter, keywords: ['^openrouter'] },
   { Icon: OpenChat, keywords: ['^openchat'] },
+  { Icon: Aya, keywords: ['aya'] },
   { Icon: Cohere, keywords: ['command'] },
   { Icon: Dbrx, keywords: ['dbrx'] },
   { Icon: Stepfun, keywords: ['step'] },
   { Icon: AiMass, keywords: ['taichu'] },
   { Icon: Ai360, keywords: ['360gpt'] },
-  {
-    Icon: Baichuan,
-    keywords: ['baichuan'],
-  },
+  { Icon: Baichuan, keywords: ['baichuan'] },
   { Icon: Rwkv, keywords: ['rwkv'] },
   { Icon: Wenxin, keywords: ['ernie'] },
   { Icon: Spark, keywords: ['spark'] },

diff --git a/src/ProviderCombine/const.tsx b/src/ProviderCombine/const.tsx
@@ -18,11 +18,13 @@ import LobeHub from '@/LobeHub';
 import Minimax from '@/Minimax';
 import Mistral from '@/Mistral';
 import Moonshot from '@/Moonshot';
+import Novita from '@/Novita';
 import Ollama from '@/Ollama';
 import OpenAI from '@/OpenAI';
 import OpenRouter from '@/OpenRouter';
 import Perplexity from '@/Perplexity';
 import { ModelProvider } from '@/ProviderIcon/const';
+import SiliconCloud from '@/SiliconCloud';
 import Stepfun from '@/Stepfun';
 import Together from '@/Together';
 import Tongyi from '@/Tongyi';
@@ -127,4 +129,6 @@ export const providerMappings: ProviderMapping[] = [
   { Icon: Baichuan.Combine, keywords: [ModelProvider.Baichuan], multiple: 0.83 },
   { Icon: AiMass.Combine, keywords: [ModelProvider.Taichu], multiple: 1.16 },
   { Icon: Ai360.Combine, keywords: [ModelProvider.Ai360], multiple: 0.83 },
+  { Icon: Novita.Combine, keywords: [ModelProvider.Novita], multiple: 0.95 },
+  { Icon: SiliconCloud.Combine, keywords: [ModelProvider.SiliconCloud], multiple: 0.75 },
 ];
diff --git a/src/ProviderIcon/const.ts b/src/ProviderIcon/const.ts
@@ -15,10 +15,12 @@ import LobeHub from '@/LobeHub';
 import Minimax from '@/Minimax';
 import Mistral from '@/Mistral';
 import Moonshot from '@/Moonshot';
+import Novita from '@/Novita';
 import Ollama from '@/Ollama';
 import OpenAI from '@/OpenAI';
 import OpenRouter from '@/OpenRouter';
 import Perplexity from '@/Perplexity';
+import SiliconCloud from '@/SiliconCloud';
 import Stepfun from '@/Stepfun';
 import Together from '@/Together';
 import Tongyi from '@/Tongyi';
@@ -54,11 +56,13 @@ export enum ModelProvider {
   Minimax = 'minimax',
   Mistral = 'mistral',
   Moonshot = 'moonshot',
+  Novita = 'novita',
   Ollama = 'ollama',
   OpenAI = 'openai',
   OpenRouter = 'openrouter',
   Perplexity = 'perplexity',
   Qwen = 'qwen',
+  SiliconCloud = 'siliconcloud',
   Stepfun = 'stepfun',
   Taichu = 'taichu',
   TogetherAI = 'togetherai',
@@ -76,6 +80,7 @@ export const providerMappings: ProviderMapping[] = [
   { Icon: Google, keywords: [ModelProvider.Google] },
   { Icon: Azure, keywords: [ModelProvider.Azure] },
   { Icon: Moonshot, keywords: [ModelProvider.Moonshot] },
+  { Icon: Novita, keywords: [ModelProvider.Novita] },
   { Icon: OpenAI, keywords: [ModelProvider.OpenAI] },
   { Icon: Ollama, keywords: [ModelProvider.Ollama] },
   { Icon: Perplexity, keywords: [ModelProvider.Perplexity] },
@@ -91,4 +96,5 @@ export const providerMappings: ProviderMapping[] = [
   { Icon: Baichuan, keywords: [ModelProvider.Baichuan] },
   { Icon: AiMass, keywords: [ModelProvider.Taichu] },
   { Icon: Ai360, keywords: [ModelProvider.Ai360] },
+  { Icon: SiliconCloud, keywords: [ModelProvider.SiliconCloud] },
 ];