add azureopenai option

agent.py

@@ -1,6 +1,18 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+import os
+from dotenv import load_dotenv
+from smolagents import CodeAgent, DuckDuckGoSearchTool
+from smolagents.models import AzureOpenAIServerModel
 
+load_dotenv()
 
-if __name__ == '__main__':
-    agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=HfApiModel())
-    agent.run("Search for the best music recommendations for a party at the Wayne's mansion.")
+if __name__ == "__main__":
+    # model = HfApiModel()
+    model = AzureOpenAIServerModel(
+        model_id="gpt-4o",
+        api_key=os.environ.get("AZURE_OPENAI_API_KEY"),
+        api_version=os.environ.get("OPENAI_API_VERSION"),
+        azure_endpoint=os.environ.get("AZURE_OPENAI_API_ENDPOINT"),
+        # custom_role_conversions={"system": "assistant", "tool-call": "assistant", "tool-response": "user"}
+    )
+    agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model, add_base_tools=True)
+    agent.run("Could you give me the 118th number in the Fibonacci sequence?")

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import pandas as pd
 from smolagents import DuckDuckGoSearchTool, CodeAgent, VisitWebpageTool, FinalAnswerTool, OpenAIServerModel, \
-    WikipediaSearchTool, PythonInterpreterTool
+    WikipediaSearchTool, PythonInterpreterTool, AzureOpenAIServerModel
 from dotenv import load_dotenv
 # (Keep Constants as is)
 # --- Constants ---
@@ -17,7 +17,13 @@ class BasicAgent:
     def __init__(self):
 
         # model = HfApiModel()
-        model = OpenAIServerModel(model_id="gpt-4o", api_key=os.environ["OPENAI_API_KEY"])
+        # model = OpenAIServerModel(model_id="gpt-4o", api_key=os.environ["OPENAI_API_KEY"])
+        model = AzureOpenAIServerModel(
+            model_id=os.environ.get("AZURE_OPENAI_MODEL"),
+            azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT"),
+            api_key=os.environ.get("AZURE_OPENAI_API_KEY"),
+            api_version=os.environ.get("OPENAI_API_VERSION")
+        )
 
         # tools
         search_tool = DuckDuckGoSearchTool()