Backport PR jupyterlab#1127: Removes outdated OpenAI models and adds …

…new ones
meeseeksmachine · Dec 2, 2024 · 6ec38ba · 6ec38ba
1 parent 9fd8b57
commit 6ec38ba
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py
@@ -37,22 +37,17 @@ 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",
         "gpt-4-turbo-preview",
         "gpt-4-0613",
-        "gpt-4-32k",
-        "gpt-4-32k-0613",
         "gpt-4-0125-preview",
         "gpt-4-1106-preview",
         "gpt-4o",
+        "gpt-4o-2024-11-20",
         "gpt-4o-mini",
+        "chatgpt-4o-latest",
     ]
     model_id_key = "model_name"
     pypi_package_deps = ["langchain_openai"]