From 40ff68c2f479bc5965703b6a4ae503772678f7c7 Mon Sep 17 00:00:00 2001
From: EduardDurech <39579228+EduardDurech@users.noreply.github.com>
Date: Fri, 9 Feb 2024 00:05:32 +0100
Subject: [PATCH] Add new OpenAI models (#625)

* Update providers.py

Added new OpenAI models gpt-3.5-turbo-0125, gpt-4-0125-preview, and gpt-4-turbo-preview (currently points to 0125-preview)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update providers.py

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 packages/jupyter-ai-magics/jupyter_ai_magics/providers.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
index 850e24a99..2704e8f32 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -657,15 +657,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"