changed model

app.py

@@ -10,6 +10,10 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, LiteL
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+from langchain_openai import AzureChatOpenAI
+
+load_dotenv()
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
@@ -17,7 +21,7 @@ class BasicAgent:
         print("Initializing the BasicAgent")
         # model = OpenAIServerModel(model_id="gpt-4o")
         # model = OpenAIServerModel(model_id="gpt-4o-mini")
-        model = OpenAIServerModel(model_id="o1-mini")
+        # model = OpenAIServerModel(model_id="o1-mini")
         # google_api_key = os.getenv('GOOGLE_API_KEY')
         # if google_api_key:
         #     print(f"GOOGLE_API_KEY found. Last 10 characters: ...{google_api_key[-10:]}")
@@ -35,6 +39,17 @@ class BasicAgent:
         #     print("and that your GOOGLE_API_KEY environment variable is set.")
         #     raise e # Re-raise the exception to fail initialization if model fails
 
+        model = AzureChatOpenAI(
+        azure_endpoint=os.getenv("AZURE_API_ENDPOINT"),
+        azure_deployment=os.getenv("AZURE_DEPLOYMENT_NME"),  # or your deployment
+        api_key=os.getenv("AZURE_OPENAI_KEY"),  
+        api_version=os.getenv("AZURE_API_VERSION"),
+        temperature=0.5,
+        max_tokens=None,
+        timeout=None,
+        max_retries=2,
+        )
+
         # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
         print("DuckDuckSearchTool instantiated")