Skip to content

Commit f696abe

Browse files
🐛 fix: Fix spark in const
1 parent c0108cf commit f696abe

File tree

1 file changed

+13
-5
lines changed

1 file changed

+13
-5
lines changed

src/ModelIcon/const.ts

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,13 @@ import ChatGLM from '@/ChatGLM';
1313
import Claude from '@/Claude';
1414
import CodeGeeX from '@/CodeGeeX';
1515
import Cohere from '@/Cohere';
16+
import Dalle from '@/Dalle';
1617
import Dbrx from '@/Dbrx';
1718
import DeepSeek from '@/DeepSeek';
1819
import Doubao from '@/Doubao';
1920
import Fireworks from '@/Fireworks';
2021
import FishAudio from '@/FishAudio';
22+
import Flux from '@/Flux';
2123
import Gemini from '@/Gemini';
2224
import Gemma from '@/Gemma';
2325
import Google from '@/Google';
@@ -33,12 +35,14 @@ import Moonshot from '@/Moonshot';
3335
import OpenAI from '@/OpenAI';
3436
import OpenChat from '@/OpenChat';
3537
import OpenRouter from '@/OpenRouter';
38+
import PaLM from '@/PaLM';
3639
import Perplexity from '@/Perplexity';
3740
import Qwen from '@/Qwen';
3841
import Rwkv from '@/Rwkv';
3942
import Spark from '@/Spark';
4043
import Stability from '@/Stability';
4144
import Stepfun from '@/Stepfun';
45+
import Suno from '@/Suno';
4246
import Upstage from '@/Upstage';
4347
import Wenxin from '@/Wenxin';
4448
import Yi from '@/Yi';
@@ -64,21 +68,22 @@ export interface ModelMapping {
6468
export const modelMappings: ModelMapping[] = [
6569
{ Icon: OpenAI, keywords: ['gpt-3'], props: { type: 'gpt3' } },
6670
{ Icon: OpenAI, keywords: ['gpt-4'], props: { type: 'gpt4' } },
67-
{ Icon: OpenAI, keywords: ['^gpt-', '/gpt-', 'openai', 'dalle', 'text-embedding-'] },
71+
{ Icon: Dalle, keywords: ['dalle'] },
72+
{ Icon: OpenAI, keywords: ['^gpt-', '/gpt-', 'openai', 'text-embedding-'] },
6873
{ Icon: ChatGLM, keywords: ['glm-', 'chatglm'] },
6974
{ Icon: CodeGeeX, keywords: ['^codegeex'] },
7075
{ Icon: DeepSeek, keywords: ['deepseek'] },
7176
{ Icon: Claude, keywords: ['claude'] },
7277
{ Icon: Aws, keywords: ['titan'] },
7378
{ Icon: Fireworks, keywords: ['accounts/fireworks/models/fire'] },
74-
{ Icon: Meta, keywords: ['llama'] },
79+
{ Icon: Meta, keywords: ['llama', '/l3'] },
7580
{ Icon: LLaVA, keywords: ['llava'] },
7681
{ Icon: Gemini, keywords: ['gemini'] },
7782
{ Icon: Gemma, keywords: ['gemma'] },
7883
{ Icon: Moonshot, keywords: ['moonshot'] },
7984
{ Icon: Qwen, keywords: ['qwen'] },
8085
{ Icon: Minimax, keywords: ['minmax', 'abab'] },
81-
{ Icon: Mistral, keywords: ['mistral', 'mixtral', 'codestral', 'mathstral'] },
86+
{ Icon: Mistral, keywords: ['mistral', 'mixtral', 'codestral', 'mathstral', '/mn-'] },
8287
{ Icon: Perplexity, keywords: ['pplx', 'sonar'] },
8388
{ Icon: Yi, keywords: ['^yi-', '/yi-'] },
8489
{ Icon: OpenRouter, keywords: ['^openrouter'] },
@@ -90,8 +95,9 @@ export const modelMappings: ModelMapping[] = [
9095
{ Icon: AiMass, keywords: ['taichu'] },
9196
{ Icon: Ai360, keywords: ['360gpt'] },
9297
{ Icon: Baichuan, keywords: ['baichuan'] },
93-
{ Icon: Rwkv, keywords: ['rwkv'] },
98+
{ Icon: Rwkv, keywords: ['rwkv', '/eagle-'] },
9499
{ Icon: Wenxin, keywords: ['ernie'] },
100+
{ Icon: Doubao, keywords: ['^doubao', 'doubao-'] },
95101
{ Icon: Hunyuan, keywords: ['hunyuan'] },
96102
{ Icon: FishAudio, keywords: ['^d_', '^g_', '^wd_'] },
97103
{ Icon: ByteDance, keywords: ['skylark'] },
@@ -109,12 +115,14 @@ export const modelMappings: ModelMapping[] = [
109115
'^sd1',
110116
],
111117
},
118+
{ Icon: Flux, keywords: ['flux'] },
119+
{ Icon: Suno, keywords: ['suno'] },
112120
{ Icon: Azure, keywords: ['wizardlm', 'phi3', 'phi-3'] },
113121
{ Icon: Adobe, keywords: ['firefly'] },
114122
{ Icon: Ai21, keywords: ['jamba', '^j2-'] },
115123
{ Icon: InternLM, keywords: ['internlm'] },
116124
{ Icon: Upstage, keywords: ['^solar-'] },
117-
{ Icon: Doubao, keywords: ['^doubao', 'doubao-'] },
125+
{ Icon: PaLM, keywords: ['palm'] },
118126
{ Icon: Google, keywords: ['google'] },
119127
{ Icon: Spark, keywords: ['spark', 'general', 'pro-128k', '4.0ultra'] },
120128
];

0 commit comments

Comments
 (0)