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 @@ def __call__(self, docs: List[str]) -> List[List[float]]: 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: