Update agent.py

agent.py

@@ -6,7 +6,7 @@ class MagAgent:
         """Initialize the MagAgent with search tools."""
         print("Initializing MagAgent with search tools...")
         self.agent = CodeAgent(
-            model=OpenAIServerModel(model_name="gpt-4o-mini"),  # Use a lightweight model
+            model=OpenAIServerModel(model_id="gpt-4o-mini"), 
             tools=[
                 DuckDuckGoSearchTool(),
                 WikipediaSearchTool()