add azure openai

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -10,10 +10,22 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+#
+class AzureOpenAIServerModel(OpenAIServerModel):
+    def __init__(self, model_id: str):
+        super().__init__(model_id=model_id)
+        # if we've reached this point, it means the openai package is available (baseclass check) so go ahead and import it
+        import openai
+
+        self.client = openai.AzureOpenAI(
+            api_key=os.environ["OPENAI_API_KEY"],
+            api_version=os.environ["OPENAI_API_VERSION"],
+            azure_endpoint=os.environ["AZURE_ENDPOINT"]
+        )
 
 class BasicAgent:
     def __init__(self):
-        model = InferenceClientModel("deepseek-ai/DeepSeek-V3")
+        model = AzureOpenAIServerModel("deepseek-ai/DeepSeek-V3")
         search_tool = DuckDuckGoSearchTool()
 
         self.agent = CodeAgent(

requirements.txt

@@ -1,4 +1,5 @@
 gradio
 requests
 smolagents
-duckduckgo-search
+duckduckgo-search
+openai