diff --git a/chatgpt/ChatService.py b/chatgpt/ChatService.py index ba3064e..a2bd021 100644 --- a/chatgpt/ChatService.py +++ b/chatgpt/ChatService.py @@ -136,7 +136,9 @@ async def set_model(self): else: self.gizmo_id = None - if "o1-preview" in self.origin_model: + if self.gizmo_id: + self.req_model = "gpt-4o" + elif "o1-preview" in self.origin_model: self.req_model = "o1-preview" elif "o1-pro" in self.origin_model: self.req_model = "o1-pro"