支持最新的gpt-4o-mini 模型

This commit is contained in:
Clivia 2024-07-19 09:39:20 +08:00
parent 4c0b22429e
commit 4cf7fb2c68
2 changed files with 3 additions and 0 deletions

View File

@ -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",

View File

@ -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"