Backport PR jupyterlab#625: Add new OpenAI models

meeseeksmachine · Feb 8, 2024 · 0c76819 · 0c76819
1 parent d1b146f
commit 0c76819
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -608,15 +608,18 @@ class ChatOpenAIProvider(BaseProvider, ChatOpenAI):
     name = "OpenAI"
     models = [
         "gpt-3.5-turbo",
+        "gpt-3.5-turbo-0125",
         "gpt-3.5-turbo-0301",  # Deprecated as of 2024-06-13
         "gpt-3.5-turbo-0613",  # Deprecated as of 2024-06-13
         "gpt-3.5-turbo-1106",
         "gpt-3.5-turbo-16k",
         "gpt-3.5-turbo-16k-0613",  # Deprecated as of 2024-06-13
         "gpt-4",
+        "gpt-4-turbo-preview",
         "gpt-4-0613",
         "gpt-4-32k",
         "gpt-4-32k-0613",
+        "gpt-4-0125-preview",
         "gpt-4-1106-preview",
     ]
     model_id_key = "model_name"