diff --git a/docs/docs/cloud/reference/cli.md b/docs/docs/cloud/reference/cli.md index 34629fa4e..914611ffe 100644 --- a/docs/docs/cloud/reference/cli.md +++ b/docs/docs/cloud/reference/cli.md @@ -81,6 +81,15 @@ The `fields` configuration determines which parts of your documents to embed: } ``` +!!! note "Common model dimensions" + - openai:text-embedding-3-large: 3072 + - openai:text-embedding-3-small: 1536 + - openai:text-embedding-ada-002: 1536 + - cohere:embed-english-v3.0: 1024 + - cohere:embed-english-light-v3.0: 384 + - cohere:embed-multilingual-v3.0: 1024 + - cohere:embed-multilingual-light-v3.0: 384 + #### Semantic search with a custom embedding function If you want to use semantic search with a custom embedding function, you can pass a path to a custom embedding function: diff --git a/libs/checkpoint/langgraph/store/base/__init__.py b/libs/checkpoint/langgraph/store/base/__init__.py index 9673fc6cb..7734eff03 100644 --- a/libs/checkpoint/langgraph/store/base/__init__.py +++ b/libs/checkpoint/langgraph/store/base/__init__.py @@ -478,13 +478,13 @@ class IndexConfig(TypedDict, total=False): """Number of dimensions in the embedding vectors. Common embedding models have the following dimensions: - - OpenAI text-embedding-3-large: 256, 1024, or 3072 - - OpenAI text-embedding-3-small: 512 or 1536 - - OpenAI text-embedding-ada-002: 1536 - - Cohere embed-english-v3.0: 1024 - - Cohere embed-english-light-v3.0: 384 - - Cohere embed-multilingual-v3.0: 1024 - - Cohere embed-multilingual-light-v3.0: 384 + - openai:text-embedding-3-large: 3072 + - openai:text-embedding-3-small: 1536 + - openai:text-embedding-ada-002: 1536 + - cohere:embed-english-v3.0: 1024 + - cohere:embed-english-light-v3.0: 384 + - cohere:embed-multilingual-v3.0: 1024 + - cohere:embed-multilingual-light-v3.0: 384 """ embed: Union[Embeddings, EmbeddingsFunc, AEmbeddingsFunc] diff --git a/libs/cli/langgraph_cli/config.py b/libs/cli/langgraph_cli/config.py index 99db440e1..5a15ac479 100644 --- a/libs/cli/langgraph_cli/config.py +++ b/libs/cli/langgraph_cli/config.py @@ -17,13 +17,13 @@ class IndexConfig(TypedDict, total=False): """Number of dimensions in the embedding vectors. Common embedding models have the following dimensions: - - OpenAI text-embedding-3-large: 256, 1024, or 3072 - - OpenAI text-embedding-3-small: 512 or 1536 - - OpenAI text-embedding-ada-002: 1536 - - Cohere embed-english-v3.0: 1024 - - Cohere embed-english-light-v3.0: 384 - - Cohere embed-multilingual-v3.0: 1024 - - Cohere embed-multilingual-light-v3.0: 384 + - openai:text-embedding-3-large: 3072 + - openai:text-embedding-3-small: 1536 + - openai:text-embedding-ada-002: 1536 + - cohere:embed-english-v3.0: 1024 + - cohere:embed-english-light-v3.0: 384 + - cohere:embed-multilingual-v3.0: 1024 + - cohere:embed-multilingual-light-v3.0: 384 """ embed: str