Skip to content
Snippets Groups Projects
Commit a997b974 authored by Ismail Ashraq's avatar Ismail Ashraq
Browse files

fix: cohere type

parent c3fa297b
Branches
Tags
No related merge requests found
...@@ -2,7 +2,7 @@ import os ...@@ -2,7 +2,7 @@ import os
from typing import List, Optional from typing import List, Optional
import cohere 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.encoders import BaseEncoder
from semantic_router.utils.defaults import EncoderDefault from semantic_router.utils.defaults import EncoderDefault
...@@ -46,7 +46,7 @@ class CohereEncoder(BaseEncoder): ...@@ -46,7 +46,7 @@ class CohereEncoder(BaseEncoder):
texts=docs, input_type=self.input_type, model=self.name texts=docs, input_type=self.input_type, model=self.name
) )
# Check for unsupported type. # Check for unsupported type.
if isinstance(embeds, EmbedResponse_EmbeddingsByType): if isinstance(embeds, EmbeddingsByTypeEmbedResponse):
raise NotImplementedError( raise NotImplementedError(
"Handling of EmbedByTypeResponseEmbeddings is not implemented." "Handling of EmbedByTypeResponseEmbeddings is not implemented."
) )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment