native azure model

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, VisitWebpageTool, PythonInterpreterTool, FinalAnswerTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, AzureOpenAIServerModel, VisitWebpageTool, PythonInterpreterTool, FinalAnswerTool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -11,21 +11,15 @@ 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 = AzureOpenAIServerModel("gpt-4o")
+        model = AzureOpenAIServerModel(
+            "gpt-4o",
+            azure_endpoint=os.environ["AZURE_ENDPOINT"],
+            api_version=os.environ["OPENAI_API_VERSION"],
+            api_key=os.environ["OPENAI_API_KEY"]
+        )
         search_tool = DuckDuckGoSearchTool()
         visit_page_tool = VisitWebpageTool()
         interpreter_tool = PythonInterpreterTool()