update chatgpt list

jupyterlab · Aug 11, 2023 · 7a33711 · 7a33711
1 parent a0b949e
commit 7a33711
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -419,11 +419,13 @@ class ChatOpenAIProvider(BaseProvider, OpenAIChat):
     id = "openai-chat"
     name = "OpenAI"
     models = [
-        "gpt-4",
         "gpt-4-0314",
-        "gpt-4-32k",
-        "gpt-4-32k-0314",
         "gpt-3.5-turbo",
+        "gpt-4-0613",
+        "gpt-4",
+        "gpt-3.5-turbo-16k-0613",
+        "gpt-3.5-turbo-16k",
+        "gpt-3.5-turbo-0613",
         "gpt-3.5-turbo-0301",
     ]
     model_id_key = "model_name"
@@ -443,11 +445,13 @@ class ChatOpenAINewProvider(BaseProvider, ChatOpenAI):
     id = "openai-chat-new"
     name = "OpenAI"
     models = [
-        "gpt-4",
         "gpt-4-0314",
-        "gpt-4-32k",
-        "gpt-4-32k-0314",
         "gpt-3.5-turbo",
+        "gpt-4-0613",
+        "gpt-4",
+        "gpt-3.5-turbo-16k-0613",
+        "gpt-3.5-turbo-16k",
+        "gpt-3.5-turbo-0613",
         "gpt-3.5-turbo-0301",
     ]
     model_id_key = "model_name"