From 09f0335bb142f61a398f4ba6de91272263d2b0d0 Mon Sep 17 00:00:00 2001
From: Zahid Syed <98535579+zahid-syed@users.noreply.github.com>
Date: Thu, 14 Mar 2024 12:10:25 -0400
Subject: [PATCH] Update semantic_router/encoders/mistral.py
Co-authored-by: Ismail Ashraq <59014721+ashraq1455@users.noreply.github.com>
---
semantic_router/encoders/mistral.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/semantic_router/encoders/mistral.py b/semantic_router/encoders/mistral.py
index 3fc23051..bd6f5045 100644
--- a/semantic_router/encoders/mistral.py
+++ b/semantic_router/encoders/mistral.py
@@ -67,7 +67,7 @@ class MistralEncoder(BaseEncoder):
if (
not embeds
- or not isinstance(embeds, self._embedding_response)
+ or not isinstance(embeds, self._mistralai.models.embeddings.EmbeddingResponse)
or not embeds.data
):
raise ValueError(f"No embeddings returned from MistralAI: {error_message}")
--
GitLab