From 23c2b308932c2918cb7d1b919dbcb184341c40e6 Mon Sep 17 00:00:00 2001
From: Yufei Kang <kou.uhi.x@gmail.com>
Date: Wed, 4 Dec 2024 18:24:34 +0900
Subject: [PATCH] feat(models): update OpenAI models and add new o1 mini and o1
 preview models

---
 app/models.py | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/app/models.py b/app/models.py
index d213caf..44c5c2b 100644
--- 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",