From a997b9741a3e0193d972c9abe2239faa9e5f161d Mon Sep 17 00:00:00 2001
From: Ismail Ashraq <ismailashraq@Ismails-MacBook-Pro.local>
Date: Mon, 23 Sep 2024 17:04:02 +0500
Subject: [PATCH] fix: cohere type
---
semantic_router/encoders/cohere.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/semantic_router/encoders/cohere.py b/semantic_router/encoders/cohere.py
index 3fcac85c..01426e9f 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."
)
--
GitLab