feat(models): update OpenAI models and add new o1 mini and o1 preview…

… models
yufeikang · Dec 4, 2024 · 23c2b30 · 23c2b30
1 parent 302efb7
commit 23c2b30
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/app/models.py b/app/models.py
@@ -308,14 +308,24 @@ def get_models(self):
         default_models = _get_default_model_dict("openai-gpt-4o-mini")
         models = [
             {
-                "id": "openai-gpt-3.5-turbo",
-                "model": "gpt-3.5-turbo",
-                "name": "GPT-3.5 Turbo",
+                "id": "openai-o1-mini",
+                "model": "o1-mini",
+                "name": "o1 mini",
                 "provider": "openai",
                 "provider_name": "OpenAI",
                 "provider_brand": "openai",
                 "context": 16,
-                **_get_model_extra_info("gpt-3.5-turbo"),
+                **_get_model_extra_info("o1-mini"),
+            },
+            {
+                "id": "openai-o1-preview",
+                "model": "o1-preview",
+                "name": "o1 Preview",
+                "provider": "openai",
+                "provider_name": "OpenAI",
+                "provider_brand": "openai",
+                "context": 16,
+                **_get_model_extra_info("o1-preview"),
             },
             {
                 "id": "openai-gpt-4o-mini",
@@ -340,7 +350,7 @@ def get_models(self):
             {
                 "id": "openai-gpt-4-turbo",
                 "model": "gpt-4-turbo",
-                "name": "GPT-4 Turbo",
+                "name": "GPT-4 Turbo (Legacy)",
                 "provider": "openai",
                 "provider_name": "OpenAI",
                 "provider_brand": "openai",