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 index a1347073f..106580c8a 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py @@ -25,6 +25,7 @@ def is_api_key_exc(cls, e: Exception): return False +# https://platform.openai.com/docs/models/ class ChatOpenAIProvider(BaseProvider, ChatOpenAI): id = "openai-chat" name = "OpenAI" @@ -37,12 +38,14 @@ class ChatOpenAIProvider(BaseProvider, ChatOpenAI): "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", ] model_id_key = "model_name" pypi_package_deps = ["langchain_openai"]