Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-03-17 05:00:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update embedding model names in docs, refs #654

Browse source 
Also ran Black.
This commit is contained in:
Simon Willison 2025-01-22 20:34:59 -08:00
parent 6f7ea406bf
commit 57d3baac42
2 changed files with 38 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
docs/aliases.md
View file

@ -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)
```
<!-- [[[end]]] -->

37
llm/default_plugins/openai_models.py
View file

@ -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):