Adds new gen-3 embedding models

jupyterlab · Feb 7, 2024 · 3ca4d7b · 3ca4d7b
1 parent 7b8e314
commit 3ca4d7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,7 @@ 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")