From 5a8091ff19bfe743e2ff3ef9450ab9b0d5e41a15 Mon Sep 17 00:00:00 2001
From: Billy Trend <billy@cohere.com>
Date: Mon, 23 Sep 2024 12:14:17 +0100
Subject: [PATCH] Fix types
---
pyproject.toml | 2 +-
semantic_router/encoders/cohere.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/pyproject.toml b/pyproject.toml
index 97d92edf..1b963c74 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,7 @@ license = "MIT"
python = ">=3.9,<3.13"
pydantic = "^2.5.3"
openai = ">=1.10.0,<2.0.0"
-cohere = ">=5.00,<6.00"
+cohere = ">=5.9.4,<6.00"
mistralai= {version = ">=0.0.12,<0.1.0", optional = true}
numpy = "^1.25.2"
colorlog = "^6.8.0"
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