From 4cf7fb2c6881ec43cd717a6b7e22a18e16664ab2 Mon Sep 17 00:00:00 2001
From: Clivia <132346501+Yanyutin753@users.noreply.github.com>
Date: Fri, 19 Jul 2024 09:39:20 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20=E6=94=AF=E6=8C=81=E6=9C=80?=
 =?UTF-8?q?=E6=96=B0=E7=9A=84gpt-4o-mini=20=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/models.py          | 1 +
 chatgpt/ChatService.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/api/models.py b/api/models.py
index d759246..1598fb0 100644
--- a/api/models.py
+++ b/api/models.py
@@ -7,6 +7,7 @@ model_proxy = {
     "gpt-4-vision-preview": "gpt-4-1106-vision-preview",
     "gpt-4-turbo": "gpt-4-turbo-2024-04-09",
     "gpt-4o": "gpt-4o-2024-05-13",
+    "gpt-4o-mini": "gpt-4o-mini",
     "claude-3-opus": "claude-3-opus-20240229",
     "claude-3-sonnet": "claude-3-sonnet-20240229",
     "claude-3-haiku": "claude-3-haiku-20240307",
diff --git a/chatgpt/ChatService.py b/chatgpt/ChatService.py
index 04542e9..cb5f0d5 100644
--- a/chatgpt/ChatService.py
+++ b/chatgpt/ChatService.py
@@ -128,6 +128,8 @@ class ChatService:
             self.req_model = "gpt-4o"
         elif "gpt-4" in self.origin_model:
             self.req_model = "gpt-4"
+        elif "gpt-4o-mini" in self.origin_model:
+            self.req_model = "gpt-4o-mini"
         else:
             self.req_model = "text-davinci-002-render-sha"