From 68299a50bea2eef5f630b13b679bd4a9df3acd12 Mon Sep 17 00:00:00 2001
From: canisminor1990 <i@canisminor.cc>
Date: Fri, 30 Aug 2024 18:43:42 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix:=20Fix=20spark=20const?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/ModelIcon/const.ts | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/ModelIcon/const.ts b/src/ModelIcon/const.ts
index 2d35063b..1c40934e 100644
--- 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$',
+    ],
+  },
 ];