From 36fb6183973323a82ed5e9f3089523fb40f88bf1 Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Wed, 15 May 2024 01:02:01 +0400
Subject: [PATCH] Linting.
---
semantic_router/encoders/cohere.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/semantic_router/encoders/cohere.py b/semantic_router/encoders/cohere.py
index 59dd1c63..8b747e97 100644
--- a/semantic_router/encoders/cohere.py
+++ b/semantic_router/encoders/cohere.py
@@ -2,7 +2,10 @@ 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_EmbeddingsFloats,
+ EmbedResponse_EmbeddingsByType,
+)
from semantic_router.encoders import BaseEncoder
from semantic_router.utils.defaults import EncoderDefault
@@ -50,7 +53,9 @@ class CohereEncoder(BaseEncoder):
if isinstance(embeds, EmbedResponse_EmbeddingsFloats):
return embeds.embeddings
elif isinstance(embeds, EmbedResponse_EmbeddingsByType):
- raise NotImplementedError("Handling of EmbedByTypeResponseEmbeddings is not implemented.")
+ raise NotImplementedError(
+ "Handling of EmbedByTypeResponseEmbeddings is not implemented."
+ )
else:
raise ValueError("Unexpected response type from Cohere API")
except Exception as e:
--
GitLab