diff --git a/semantic_router/encoders/mistral.py b/semantic_router/encoders/mistral.py index c515508e7a3819104c01662936efce415346f2b2..a3768d340dd7bedd270d715731a02def7b9b7e7f 100644 --- a/semantic_router/encoders/mistral.py +++ b/semantic_router/encoders/mistral.py @@ -33,9 +33,8 @@ class MistralEncoder(BaseEncoder): def _initialize_client(self, api_key): try: + import mistralai from mistralai.client import MistralClient - from mistralai.exceptions import MistralException - from mistralai.models.embeddings import EmbeddingResponse except ImportError: raise ImportError( "Please install MistralAI to use MistralEncoder. "