Merge pull request #90 from faizan2700/mistralmodel1

mistral model changes
swarmauri · Aug 16, 2024 · 2365323 · 2365323
2 parents 6e2709d + 52b9d92
commit 2365323
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/swarmauri/standard/llms/concrete/MistralModel.py b/swarmauri/standard/llms/concrete/MistralModel.py
@@ -1,6 +1,6 @@
 import json
 from typing import List, Literal, Dict
-from mistralai.client import MistralClient
+from mistralai import Mistral 
 from swarmauri.core.typing import SubclassUnion
 
 from swarmauri.standard.messages.base.MessageBase import MessageBase
@@ -35,9 +35,9 @@ def predict(self,
 
         formatted_messages = self._format_messages(conversation.history)
 
-        client =  MistralClient(api_key=self.api_key)        
+        client =  Mistral(api_key=self.api_key)        
         if enable_json:
-            response = client.chat(
+            response = client.chat.complete(
                 model=self.name,
                 messages=formatted_messages,
                 temperature=temperature,
@@ -47,7 +47,7 @@ def predict(self,
                 safe_prompt=safe_prompt
             )
         else:
-            response = client.chat(
+            response = client.chat.complete(
                 model=self.name,
                 messages=formatted_messages,
                 temperature=temperature,