diff --git a/docs/aliases.md b/docs/aliases.md index 787a303..3fa5c53 100644 --- a/docs/aliases.md +++ b/docs/aliases.md @@ -33,7 +33,13 @@ gpt-4-turbo-preview : gpt-4-turbo 4t : gpt-4-turbo 3.5-instruct : gpt-3.5-turbo-instruct chatgpt-instruct : gpt-3.5-turbo-instruct -ada : ada-002 (embedding) +ada : text-embedding-ada-002 (embedding) +ada-002 : text-embedding-ada-002 (embedding) +3-small : text-embedding-3-small (embedding) +3-large : text-embedding-3-large (embedding) +3-small-512 : text-embedding-3-small-512 (embedding) +3-large-256 : text-embedding-3-large-256 (embedding) +3-large-1024 : text-embedding-3-large-1024 (embedding) ``` diff --git a/llm/default_plugins/openai_models.py b/llm/default_plugins/openai_models.py index a59690f..33d0566 100644 --- a/llm/default_plugins/openai_models.py +++ b/llm/default_plugins/openai_models.py @@ -134,14 +134,39 @@ def register_models(register): @hookimpl def register_embedding_models(register): register( - OpenAIEmbeddingModel("text-embedding-ada-002", "text-embedding-ada-002"), aliases=("ada","ada-002",) + OpenAIEmbeddingModel("text-embedding-ada-002", "text-embedding-ada-002"), + aliases=( + "ada", + "ada-002", + ), + ) + register( + OpenAIEmbeddingModel("text-embedding-3-small", "text-embedding-3-small"), + aliases=("3-small",), + ) + register( + OpenAIEmbeddingModel("text-embedding-3-large", "text-embedding-3-large"), + aliases=("3-large",), ) - register(OpenAIEmbeddingModel("text-embedding-3-small", "text-embedding-3-small"), aliases=("3-small",)) - register(OpenAIEmbeddingModel("text-embedding-3-large", "text-embedding-3-large"), aliases=("3-large",)) # With varying dimensions - register(OpenAIEmbeddingModel("text-embedding-3-small-512", "text-embedding-3-small", 512), aliases=("3-small-512",)) - register(OpenAIEmbeddingModel("text-embedding-3-large-256", "text-embedding-3-large", 256), aliases=("3-large-256",)) - register(OpenAIEmbeddingModel("text-embedding-3-large-1024", "text-embedding-3-large", 1024), aliases=("3-large-1024",)) + register( + OpenAIEmbeddingModel( + "text-embedding-3-small-512", "text-embedding-3-small", 512 + ), + aliases=("3-small-512",), + ) + register( + OpenAIEmbeddingModel( + "text-embedding-3-large-256", "text-embedding-3-large", 256 + ), + aliases=("3-large-256",), + ) + register( + OpenAIEmbeddingModel( + "text-embedding-3-large-1024", "text-embedding-3-large", 1024 + ), + aliases=("3-large-1024",), + ) class OpenAIEmbeddingModel(EmbeddingModel):