From e32c5e3840be6057f87d9806af4d29b09869a0ff Mon Sep 17 00:00:00 2001
From: Zahid Syed <98535579+zahid-syed@users.noreply.github.com>
Date: Thu, 14 Mar 2024 12:09:47 -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 | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/semantic_router/encoders/mistral.py b/semantic_router/encoders/mistral.py
index c515508e..a3768d34 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. "
-- 
GitLab