diff --git a/semantic_router/encoders/mistral.py b/semantic_router/encoders/mistral.py index cfefda825ed497f689b8da77d7d25e631c5ebc6a..3fc230512945bb4b3f6d7495425b2c004f732daf 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: