🐛 fix: Fix dalle

bentwnghk · Oct 31, 2024 · 7e5e2d5 · 7e5e2d5
1 parent d474b29
commit 7e5e2d5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/features/modelConfig.ts b/src/features/modelConfig.ts
@@ -70,7 +70,7 @@ export const modelMappings: ModelMapping[] = [
   { Icon: OpenAI, keywords: ['gpt-3'], props: { type: 'gpt3' } },
   { Icon: OpenAI, keywords: ['gpt-4'], props: { type: 'gpt4' } },
   { Icon: OpenAI, keywords: ['o1-'], props: { type: 'o1' } },
-  { Icon: Dalle, keywords: ['dalle'] },
+  { Icon: Dalle, keywords: ['dalle', 'dall-e'] },
   { Icon: OpenAI, keywords: ['^gpt-', '/gpt-', 'openai', 'text-embedding-'] },
   { Icon: ChatGLM, keywords: ['glm-', 'chatglm'] },
   { Icon: CodeGeeX, keywords: ['^codegeex'] },
@@ -86,7 +86,10 @@ export const modelMappings: ModelMapping[] = [
   { Icon: Moonshot, keywords: ['moonshot'] },
   { Icon: Qwen, keywords: ['qwen'] },
   { Icon: Minimax, keywords: ['minmax', 'abab'] },
-  { Icon: Mistral, keywords: ['mistral', 'mixtral', 'codestral', 'mathstral', '/mn-', 'pixtral', 'ministral'] },
+  {
+    Icon: Mistral,
+    keywords: ['mistral', 'mixtral', 'codestral', 'mathstral', '/mn-', 'pixtral', 'ministral'],
+  },
   { Icon: Perplexity, keywords: ['pplx', 'sonar'] },
   { Icon: Yi, keywords: ['^yi-', '/yi-', '-yi-'] },
   { Icon: OpenRouter, keywords: ['^openrouter'] },