💄 style: Add Doubao Model (#37)

* Update const.ts * Update const.ts
lobehub · Sep 13, 2024 · 6f91171 · 6f91171
1 parent a8c8163
commit 6f91171
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/ModelIcon/const.ts b/src/ModelIcon/const.ts
@@ -98,7 +98,7 @@ export const modelMappings: ModelMapping[] = [
   { Icon: Baichuan, keywords: ['baichuan'] },
   { Icon: Rwkv, keywords: ['rwkv', '/eagle-'] },
   { Icon: Wenxin, keywords: ['ernie'] },
-  { Icon: Doubao, keywords: ['^doubao', 'doubao-'] },
+  { Icon: Doubao, keywords: ['^ep-'] },
   { Icon: Hunyuan, keywords: ['hunyuan'] },
   { Icon: FishAudio, keywords: ['^d_', '^g_', '^wd_'] },
   { Icon: ByteDance, keywords: ['skylark'] },

diff --git a/src/ProviderIcon/const.ts b/src/ProviderIcon/const.ts
@@ -8,6 +8,7 @@ import Azure from '@/Azure';
 import Baichuan from '@/Baichuan';
 import Bedrock from '@/Bedrock';
 import DeepSeek from '@/DeepSeek';
+import Doubao from '@/Doubao';
 import Fireworks from '@/Fireworks';
 import Github from '@/Github';
 import Google from '@/Google';
@@ -56,6 +57,7 @@ export enum ModelProvider {
   Baichuan = 'baichuan',
   Bedrock = 'bedrock',
   DeepSeek = 'deepseek',
+  Doubao = 'Doubao',
   FireworksAI = 'fireworksai',
   Github = 'github',
   Google = 'google',
@@ -112,4 +114,5 @@ export const providerMappings: ProviderMapping[] = [
   { Icon: Upstage, keywords: [ModelProvider.Upstage] },
   { Icon: Ai21, keywords: [ModelProvider.Ai21] },
   { Icon: Github, keywords: [ModelProvider.Github] },
+  { Icon: Doubao, keywords: [ModelProvider.Doubao] },
 ];