diff --git a/pyproject.toml b/pyproject.toml index 97d92edf56aafcee734263fcc3216c64a26f8dd5..1b963c743b5d70d3e82e0ea7467ec65bbb52bb67 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,7 +11,7 @@ license = "MIT" python = ">=3.9,<3.13" pydantic = "^2.5.3" openai = ">=1.10.0,<2.0.0" -cohere = ">=5.00,<6.00" +cohere = ">=5.9.4,<6.00" mistralai= {version = ">=0.0.12,<0.1.0", optional = true} numpy = "^1.25.2" colorlog = "^6.8.0" diff --git a/semantic_router/encoders/cohere.py b/semantic_router/encoders/cohere.py index 3fcac85c1cb6f9cb9165a4dfa5fcb19be1336afa..01426e9f23f693e24833be8a9f2ae7529d091fb9 100644 --- a/semantic_router/encoders/cohere.py +++ b/semantic_router/encoders/cohere.py @@ -2,7 +2,7 @@ import os from typing import List, Optional import cohere -from cohere.types.embed_response import EmbedResponse_EmbeddingsByType +from cohere.types.embed_response import EmbeddingsByTypeEmbedResponse from semantic_router.encoders import BaseEncoder from semantic_router.utils.defaults import EncoderDefault @@ -46,7 +46,7 @@ class CohereEncoder(BaseEncoder): texts=docs, input_type=self.input_type, model=self.name ) # Check for unsupported type. - if isinstance(embeds, EmbedResponse_EmbeddingsByType): + if isinstance(embeds, EmbeddingsByTypeEmbedResponse): raise NotImplementedError( "Handling of EmbedByTypeResponseEmbeddings is not implemented." )