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"