Merge pull request #431 from aurelio-labs/ashraq/fix-cohere-types

fix: cohere type
aurelio-labs · Sep 23, 2024 · e308619 · e308619
2 parents c3fa297 + ed368b1
commit e308619
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git 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
@@ -2,7 +2,7 @@
 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 @@ def __call__(self, docs: List[str]) -> List[List[float]]:
                 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."
                 )