🐛 fix: Fix spark const

lobehub · Aug 30, 2024 · 68299a5 · 68299a5
1 parent f696abe
commit 68299a5
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/ModelIcon/const.ts b/src/ModelIcon/const.ts
@@ -124,5 +124,16 @@ export const modelMappings: ModelMapping[] = [
   { Icon: Upstage, keywords: ['^solar-'] },
   { Icon: PaLM, keywords: ['palm'] },
   { Icon: Google, keywords: ['google'] },
-  { Icon: Spark, keywords: ['spark', 'general', 'pro-128k', '4.0ultra'] },
+  {
+    Icon: Spark,
+    keywords: [
+      'spark',
+      'general$',
+      'generalv3$',
+      'generalv3.5$',
+      '4.0ultra$',
+      'pro-128k$',
+      '4.0ultra$',
+    ],
+  },
 ];