Merge pull request #78 from bucolucas/update-default-model

Update default model to gpt-4o
This commit is contained in:
2024-08-19 13:52:22 -05:00
committed by GitHub
+2 -2
View File
@@ -9,8 +9,8 @@ class ChatGPTTelegramInferenceBot(BaseTelegramInferenceBot):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
self.client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY")) self.client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
self.model = "gpt-4o-mini" self.model = "gpt-4o"
self.max_tokens = 16384 self.max_tokens = 4096
def get_chat_response(self, messages): def get_chat_response(self, messages):
response = self.client.chat.completions.create( response = self.client.chat.completions.create(