add gpt-4o

Kav-K · May 17, 2024 · 883800e · 883800e
1 parent 6461fff
commit 883800e
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/gpt3discord.py b/gpt3discord.py
@@ -34,7 +34,7 @@
 from models.openai_model import Model
 
 
-__version__ = "12.3.8"
+__version__ = "12.3.9"
 
 
 PID_FILE = Path("bot.pid")

diff --git a/models/autocomplete_model.py b/models/autocomplete_model.py
@@ -97,6 +97,9 @@ async def get_function_calling_models(ctx: discord.AutocompleteContext):
             "gpt-3.5-turbo-1106",
             "gpt-3.5-turbo-0613",
             "gpt-4-turbo-preview",
+            "gpt-4-turbo",
+            "gpt-4o"
+
         ]
 
     async def get_models(

diff --git a/models/openai_model.py b/models/openai_model.py
@@ -73,6 +73,8 @@ class Models:
     GPT_4_TURBO = "gpt-4-1106-preview"
     GPT_4_TURBO_VISION = "gpt-4-vision-preview"
     GPT_4_TURBO_CATCHALL = "gpt-4-turbo-preview"
+    GPT_4_TURBO_REGULAR = "gpt-4-turbo"
+    GPT_4_OMEGA = "gpt-4o"
 
     # Model collections
     TEXT_MODELS = [
@@ -89,6 +91,8 @@ class Models:
         GPT_4_TURBO,
         GPT_4_TURBO_VISION,
         GPT_4_TURBO_CATCHALL,
+        GPT_4_TURBO_REGULAR,
+        GPT_4_OMEGA,
     ]
     CHATGPT_MODELS = [
         TURBO,
@@ -99,6 +103,8 @@ class Models:
         GPT_4_TURBO_VISION,
         GPT_4_TURBO,
         GPT_4_TURBO_CATCHALL,
+        GPT_4_TURBO_REGULAR,
+        GPT_4_OMEGA,
     ]
     GPT4_MODELS = [
         GPT4,
@@ -108,6 +114,8 @@ class Models:
         GPT_4_TURBO_VISION,
         GPT_4_TURBO,
         GPT_4_TURBO_CATCHALL,
+        GPT_4_TURBO_REGULAR,
+        GPT_4_OMEGA,
     ]
     EDIT_MODELS = [EDIT]
 
@@ -129,6 +137,8 @@ class Models:
         GPT_4_TURBO_VISION: 128000,
         GPT_4_TURBO: 128000,
         GPT_4_TURBO_CATCHALL: 128000,
+        GPT_4_TURBO_REGULAR: 128000,
+        GPT_4_OMEGA: 128000,
     }
 
     @staticmethod