diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py index 9ef8720ba..55f74bc1c 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py @@ -71,7 +71,11 @@ def __init__(self, *args, **kwargs): class OpenAIEmbeddingsProvider(BaseEmbeddingsProvider, OpenAIEmbeddings): id = "openai" name = "OpenAI" - models = ["text-embedding-ada-002"] + models = [ + "text-embedding-ada-002", + "text-embedding-3-small", + "text-embedding-3-large", + ] model_id_key = "model" pypi_package_deps = ["openai"] auth_strategy = EnvAuthStrategy(name="OPENAI_API_KEY")