diff --git a/semantic_router/encoders/cohere.py b/semantic_router/encoders/cohere.py index 9716a9b5f39deefac2b9f1ff08541bc36ba5cfc3..3fcac85c1cb6f9cb9165a4dfa5fcb19be1336afa 100644 --- a/semantic_router/encoders/cohere.py +++ b/semantic_router/encoders/cohere.py @@ -45,8 +45,7 @@ class CohereEncoder(BaseEncoder): embeds = self.client.embed( texts=docs, input_type=self.input_type, model=self.name ) - # Check the type of response and handle accordingly - # Only EmbedResponse_EmbeddingsFloats has embeddings of type List[List[float]] + # Check for unsupported type. if isinstance(embeds, EmbedResponse_EmbeddingsByType): raise NotImplementedError( "Handling of EmbedByTypeResponseEmbeddings is not implemented."