rearrange list, change defaults

Kav-K · May 17, 2024 · 0e16004 · 0e16004
1 parent 3034c85
commit 0e16004
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/models/openai_model.py b/models/openai_model.py
@@ -78,6 +78,8 @@ class Models:
 
     # Model collections
     TEXT_MODELS = [
+        GPT_4_TURBO_REGULAR,
+        GPT_4_OMEGA,
         DAVINCI,
         CURIE,
         TURBO,
@@ -91,10 +93,10 @@ class Models:
         GPT_4_TURBO,
         GPT_4_TURBO_VISION,
         GPT_4_TURBO_CATCHALL,
-        GPT_4_TURBO_REGULAR,
-        GPT_4_OMEGA,
     ]
     CHATGPT_MODELS = [
+        GPT_4_TURBO_REGULAR,
+        GPT_4_OMEGA,
         TURBO,
         TURBO_16,
         TURBO_DEV,
@@ -103,23 +105,21 @@ class Models:
         GPT_4_TURBO_VISION,
         GPT_4_TURBO,
         GPT_4_TURBO_CATCHALL,
-        GPT_4_TURBO_REGULAR,
-        GPT_4_OMEGA,
     ]
     GPT4_MODELS = [
+        GPT_4_TURBO_REGULAR,
+        GPT_4_OMEGA,
         GPT4,
         GPT4_32,
         GPT4_DEV,
         GPT4_32_DEV,
         GPT_4_TURBO_VISION,
         GPT_4_TURBO,
         GPT_4_TURBO_CATCHALL,
-        GPT_4_TURBO_REGULAR,
-        GPT_4_OMEGA,
     ]
     EDIT_MODELS = [EDIT]
 
-    DEFAULT = TURBO
+    DEFAULT = GPT_4_OMEGA
     LOW_USAGE_MODEL = CURIE
 
     # Tokens Mapping