update examples

Author: obinnascale3

src/examples/mistral_example/__init__.py

@@ -0,0 +1,13 @@
+import asyncio
+from examples.mistral_example.complete import chat_complete
+from examples.mistral_example.complete_async import complete_async
+from examples.mistral_example.embeddings import embeddings_create
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class MistralRunner:
+    @with_langtrace_root_span("Mistral")
+    def run(self):
+        chat_complete()
+        asyncio.run(complete_async())
+        embeddings_create()

src/examples/mistral_example/complete.py

@@ -1,15 +1,15 @@
-import os
-from langtrace_python_sdk import langtrace
+from dotenv import find_dotenv, load_dotenv
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 from mistralai import Mistral
 
-langtrace.init(api_key=os.environ["LANGTRACE_API_KEY"])
+_ = load_dotenv(find_dotenv())
 
-api_key = os.environ["MISTRAL_API_KEY"]
-model = "mistral-large-latest"
+langtrace.init()
 
-client = Mistral(api_key=api_key)
-
-def main():
+@with_langtrace_root_span("chat_complete")
+def chat_complete():
+    model = "mistral-large-latest"
+    client = Mistral()
     chat_response = client.chat.complete(
         model= model,
         messages = [
@@ -20,7 +20,3 @@ def main():
         ]
     )
     print(chat_response.choices[0].message.content)
-
-
-if __name__ == "__main__":
-    main()

src/examples/mistral_example/complete_async.py

@@ -1,15 +1,15 @@
-import asyncio
-from langtrace_python_sdk import langtrace
+from dotenv import find_dotenv, load_dotenv
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 from mistralai import Mistral
-import os
 
-langtrace.init(api_key=os.environ["LANGTRACE_API_KEY"])
+_ = load_dotenv(find_dotenv())
 
-async def main():
-    s = Mistral(
-        api_key=os.getenv("MISTRAL_API_KEY", ""),
-    )
-    res = await s.chat.complete_async(model="mistral-small-latest", messages=[
+langtrace.init()
+
+@with_langtrace_root_span("chat_complete_async")
+async def complete_async():
+    client = Mistral()
+    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.",
             "role": "user",
@@ -18,7 +18,3 @@ async def main():
     if res is not None:
         # handle response
         print(res.choices[0].message.content)
-        pass
-
-asyncio.run(main())
-print("hello")

src/examples/mistral_example/embeddings.py

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