Fix types

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"

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."
                 )