Update openai_helper.py

gpt-4-vision-preview is deprecated n3d1117#622
tyapinalexandr · Jul 31, 2024 · 732f4b0 · 732f4b0
1 parent 205f2f7
commit 732f4b0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bot/openai_helper.py b/bot/openai_helper.py
@@ -27,6 +27,7 @@
 GPT_4_MODELS = ("gpt-4", "gpt-4-0314", "gpt-4-0613", "gpt-4-turbo-preview")
 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_VISION_MODELS = ("gpt-4o",)
 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-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
@@ -70,6 +71,7 @@ def are_functions_available(model: str) -> bool:
     if model in ("gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k-0613"):
         return datetime.date.today() < datetime.date(2024, 6, 13)
     if model == 'gpt-4-vision-preview':
+    if model == 'gpt-4o':
         return False
     return True