From 04d78c36cb0eb50bece71f03642082bc33c30f2b Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Wed, 15 May 2024 01:20:27 +0400
Subject: [PATCH] Linting.
---
semantic_router/encoders/cohere.py | 11 +++--------
1 file changed, 3 insertions(+), 8 deletions(-)
diff --git a/semantic_router/encoders/cohere.py b/semantic_router/encoders/cohere.py
index 935195e6..9716a9b5 100644
--- a/semantic_router/encoders/cohere.py
+++ b/semantic_router/encoders/cohere.py
@@ -2,10 +2,7 @@ import os
from typing import List, Optional
import cohere
-from cohere.types.embed_response import (
- EmbedResponse_EmbeddingsFloats,
- EmbedResponse_EmbeddingsByType,
-)
+from cohere.types.embed_response import EmbedResponse_EmbeddingsByType
from semantic_router.encoders import BaseEncoder
from semantic_router.utils.defaults import EncoderDefault
@@ -50,13 +47,11 @@ class CohereEncoder(BaseEncoder):
)
# Check the type of response and handle accordingly
# Only EmbedResponse_EmbeddingsFloats has embeddings of type List[List[float]]
- if isinstance(embeds, EmbedResponse_EmbeddingsFloats):
- return embeds.embeddings
- elif isinstance(embeds, EmbedResponse_EmbeddingsByType):
+ if isinstance(embeds, EmbedResponse_EmbeddingsByType):
raise NotImplementedError(
"Handling of EmbedByTypeResponseEmbeddings is not implemented."
)
else:
- raise ValueError(f"Unexpected response type from Cohere API: type(embeds) = {type(embeds)}")
+ return embeds.embeddings
except Exception as e:
raise ValueError(f"Cohere API call failed. Error: {e}") from e
--
GitLab