Added GPT-4o-mini support

GaiZhenbiao · Jul 23, 2024 · a7c3971 · a7c3971
1 parent a87f2b2
commit a7c3971
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/locale/en_US.json b/locale/en_US.json
@@ -278,5 +278,6 @@
     "chuanhu_question_2": "Any new releases from Apple?",
     "chuanhu_question_3": "Current prices of graphics cards?",
     "chuanhu_question_4": "Any new trends on TikTok?",
-    "gpt4o_description": "OpenAI's most advanced, multimodal flagship model that’s cheaper and faster than GPT-4 Turbo."
+    "gpt4o_description": "OpenAI's most advanced, multimodal flagship model that’s cheaper and faster than GPT-4 Turbo.",
+    "gpt4omini_description": "OpenAI's affordable and intelligent small model for fast, lightweight tasks."
 }
diff --git a/locale/zh_CN.json b/locale/zh_CN.json
@@ -20,5 +20,6 @@
     "chuanhu_question_2": "最近 Apple 发布了什么新品？",
     "chuanhu_question_3": "现在显卡的价格如何？",
     "chuanhu_question_4": "TikTok 上有什么新梗？",
-    "gpt4o_description": "OpenAI 的最先进的多模态旗舰模型，比 GPT-4 Turbo 更便宜、更快。"
+    "gpt4o_description": "OpenAI 的最先进的多模态旗舰模型，比 GPT-4 Turbo 更便宜、更快。",
+    "gpt4omini_description": "OpenAI 的经济实惠且智能的小型模型，适用于快速、轻量级任务。"
 }
diff --git a/modules/presets.py b/modules/presets.py
@@ -55,6 +55,7 @@
 ONLINE_MODELS = [
     "GPT3.5 Turbo",
     "GPT-4o",
+    "GPT-4o-mini",
     "GPT4 Turbo",
     "GPT3.5 Turbo Instruct",
     "GPT4",
@@ -238,6 +239,16 @@
             "slogan": i18n("gpt_default_slogan"),
         }
     },
+    "GPT-4o-mini": {
+        "model_name": "gpt-4o-mini",
+        "description": "gpt4omini_description",
+        "token_limit": 128000,
+        "multimodal": True,
+        "placeholder": {
+            "logo": "file=web_assets/model_logos/openai-black.webp",
+            "slogan": i18n("gpt_default_slogan"),
+        }
+    },
     "Claude 3 Haiku": {
         "model_name": "claude-3-haiku-20240307",
         "description": "claude3_haiku_description",