From 2b6903365819f8c08a3913553de64dafeb40311b Mon Sep 17 00:00:00 2001
From: Zahid Syed <98535579+zahid-syed@users.noreply.github.com>
Date: Thu, 14 Mar 2024 12:10:18 -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 cfefda82..3fc23051 100644
--- a/semantic_router/encoders/mistral.py
+++ b/semantic_router/encoders/mistral.py
@@ -59,7 +59,7 @@ class MistralEncoder(BaseEncoder):
                 embeds = self._client.embeddings(model=self.name, input=docs)
                 if embeds.data:
                     break
-            except self._mistral_exception as e:
+            except self._mistralai.exceptions.MistralException as e:
                 sleep(2**_)
                 error_message = str(e)
             except Exception as e:
-- 
GitLab