add auzre openai example

Author: obinnascale3

src/examples/azureopenai_example/__init__.py

@@ -0,0 +1,9 @@
+from examples.azureopenai_example.completion import chat_completion
+from langtrace_python_sdk import with_langtrace_root_span, langtrace
+
+langtrace.init()
+
+class AzureOpenAIRunner:
+    @with_langtrace_root_span("AzureOpenAI")
+    def run(self):
+        chat_completion()

src/examples/azureopenai_example/completion.py

@@ -0,0 +1,22 @@
+import os
+from langchain_openai import AzureChatOpenAI
+
+from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+
+model = AzureChatOpenAI(
+    azure_endpoint=os.environ['AZURE_OPENAI_ENDPOINT'],
+    azure_deployment=os.environ['AZURE_OPENAI_DEPLOYMENT_NAME'],
+    openai_api_version=os.environ['AZURE_OPENAI_API_VERSION'],
+)
+
+@with_langtrace_root_span()
+def chat_completion():
+    messages = [
+        (
+            "system",
+            "You are a helpful assistant that translates English to French. Translate the user sentence.",
+        ),
+        ("human", "I love programming."),
+    ]
+    result = model.invoke(messages)
+    print(result)

src/run_example.py

@@ -2,13 +2,14 @@
 
 ENABLED_EXAMPLES = {
     "anthropic": False,
+    "azureopenai": True,
     "chroma": False,
     "cohere": False,
     "fastapi": False,
     "langchain": False,
     "llamaindex": False,
     "hiveagent": False,
-    "openai": True,
+    "openai": False,
     "perplexity": False,
     "pinecone": False,
     "qdrant": False,
@@ -110,3 +111,9 @@
 
     print(Fore.BLUE + "Running Mistral example" + Fore.RESET)
     MistralRunner().run()
+
+if ENABLED_EXAMPLES["azureopenai"]:
+    from examples.azureopenai_example import AzureOpenAIRunner
+
+    print(Fore.BLUE + "Running Azure OpenAI example" + Fore.RESET)
+    AzureOpenAIRunner().run()