add mistral examples

Author: obinnascale3

src/examples/mistral_example/__init__.py

@@ -0,0 +1,26 @@
+import os
+from langtrace_python_sdk import langtrace
+from mistralai import Mistral
+
+langtrace.init(api_key=os.environ["LANGTRACE_API_KEY"])
+
+api_key = os.environ["MISTRAL_API_KEY"]
+model = "mistral-large-latest"
+
+client = Mistral(api_key=api_key)
+
+def main():
+    chat_response = client.chat.complete(
+        model= model,
+        messages = [
+            {
+                "role": "user",
+                "content": "I need 10 cocktail recipes with tequila other than the classics like margarita, tequila"
+            },
+        ]
+    )
+    print(chat_response.choices[0].message.content)
+
+
+if __name__ == "__main__":
+    main()

src/examples/mistral_example/complete.py

@@ -0,0 +1,24 @@
+import asyncio
+from langtrace_python_sdk import langtrace
+from mistralai import Mistral
+import os
+
+langtrace.init(api_key=os.environ["LANGTRACE_API_KEY"])
+
+async def main():
+    s = Mistral(
+        api_key=os.getenv("MISTRAL_API_KEY", ""),
+    )
+    res = await s.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.",
+            "role": "user",
+        },
+    ])
+    if res is not None:
+        # handle response
+        print(res.choices[0].message.content)
+        pass
+
+asyncio.run(main())
+print("hello")

src/examples/mistral_example/complete_async.py

@@ -0,0 +1,17 @@
+import os
+from langtrace_python_sdk import langtrace
+from mistralai import Mistral
+
+langtrace.init(api_key=os.environ["LANGTRACE_API_KEY"])
+
+api_key = os.environ["MISTRAL_API_KEY"]
+model = "mistral-embed"
+
+client = Mistral(api_key=api_key)
+
+embeddings_batch_response = client.embeddings.create(
+    model=model,
+    inputs=["Embed this sentence.", "As well as this one."],
+)
+
+print(embeddings_batch_response.data[0].embedding)