diff --git a/bot/openai_helper.py b/bot/openai_helper.py index 5a1896cf..1fd9e297 100644 --- a/bot/openai_helper.py +++ b/bot/openai_helper.py @@ -28,7 +28,7 @@ GPT_4_32K_MODELS = ("gpt-4-32k", "gpt-4-32k-0314", "gpt-4-32k-0613") GPT_4_VISION_MODELS = ("gpt-4-vision-preview",) GPT_4_128K_MODELS = ("gpt-4-1106-preview","gpt-4-0125-preview","gpt-4-turbo-preview", "gpt-4-turbo", "gpt-4-turbo-2024-04-09") -GPT_4O_MODELS = ("gpt-4o",) +GPT_4O_MODELS = ("gpt-4o","gpt-4o-mini") GPT_ALL_MODELS = GPT_3_MODELS + GPT_3_16K_MODELS + GPT_4_MODELS + GPT_4_32K_MODELS + GPT_4_VISION_MODELS + GPT_4_128K_MODELS + GPT_4O_MODELS def default_max_tokens(model: str) -> int: