Update app.py

app.py

@@ -11,6 +11,7 @@ from huggingface_hub import login
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 login(token=os.environ["HUGGINGFACEHUB_API_TOKEN"])
+openai_key = os.environ.get("OPENAI_API_KEY")
 
 search_tool = DuckDuckGoSearchTool()
 
@@ -18,11 +19,20 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
     log_output = ""
 
     try:
-        agent = CodeAgent(
+        
+        agent = ToolCallingAgentAgent(
             tools=[search_tool],
-            model=InferenceClientModel(model="mistralai/Magistral-Small-2506", provider="featherless-ai"),
-            max_steps=5,
-            verbosity_level=2        
+            model=InferenceClientModel(
+                model="gpt-3.5-turbo",
+                provider="openai",
+                model_kwargs={
+                    "api_key": os.environ.get("OPENAI_API_KEY"),
+                    "temperature": 0,
+                    "max_tokens": 256
+                }
+            ),
+            max_steps=7,
+            verbosity_level=2
         )
     except Exception as e:
         yield f"Error initializing agent: {e}", None, log_output