fix mistral examples

Author: obinnascale3

src/examples/mistral_example/complete.py

@@ -1,13 +1,11 @@
-from dotenv import find_dotenv, load_dotenv
+import os
 from langtrace_python_sdk import with_langtrace_root_span
 from mistralai import Mistral
 
-_ = load_dotenv(find_dotenv())
-
 @with_langtrace_root_span("chat_complete")
 def chat_complete():
     model = "mistral-large-latest"
-    client = Mistral()
+    client = Mistral(api_key=os.environ["MISTRAL_API_KEY"])
     chat_response = client.chat.complete(
         model= model,
         messages = [

src/examples/mistral_example/complete_async.py

@@ -1,12 +1,10 @@
-from dotenv import find_dotenv, load_dotenv
+import os
 from langtrace_python_sdk import with_langtrace_root_span
 from mistralai import Mistral
 
-_ = load_dotenv(find_dotenv())
-
 @with_langtrace_root_span("chat_complete_async")
 async def complete_async():
-    client = Mistral()
+    client = Mistral(api_key=os.environ["MISTRAL_API_KEY"])
     res = await client.chat.complete_async(model="mistral-small-latest", messages=[
         {
             "content": "Which locations should I visit when I travel to New york? Answer in one short sentence.",

src/examples/mistral_example/embeddings.py

@@ -1,15 +1,13 @@
-from dotenv import find_dotenv, load_dotenv
+import os
 from langtrace_python_sdk import with_langtrace_root_span
 from mistralai import Mistral
 
-_ = load_dotenv(find_dotenv())
-
 
 @with_langtrace_root_span("create_embeddings")
 def embeddings_create():
     model = "mistral-embed"
 
-    client = Mistral()
+    client = Mistral(api_key=os.environ["MISTRAL_API_KEY"])
 
     embeddings_batch_response = client.embeddings.create(
         model=model,

src/run_example.py

@@ -16,7 +16,8 @@
     "ollama": False,
     "groq": False,
     "vertexai": False,
-    "gemini": True,
+    "gemini": False,
+    "mistral": True,
 }
 
 if ENABLED_EXAMPLES["anthropic"]:
@@ -103,3 +104,9 @@
 
     print(Fore.BLUE + "Running Gemini example" + Fore.RESET)
     GeminiRunner().run()
+
+if ENABLED_EXAMPLES["mistral"]:
+    from examples.mistral_example import MistralRunner
+
+    print(Fore.BLUE + "Running Mistral example" + Fore.RESET)
+    MistralRunner().run()