Update agent.py

agent.py

@@ -260,6 +260,7 @@ class MagAgent:
     def __init__(self, rate_limiter: Optional[Limiter] = None):
         """Initialize the MagAgent with search tools."""
         self.rate_limiter = rate_limiter
+
         print("Initializing MagAgent with search tools...")
         model = LiteLLMModel(
             model_id="gemini/gemini-2.0-flash",
@@ -309,8 +310,7 @@ class MagAgent:
 #                f"Return the answer as a string."
             )
             print(f"Calling agent.run for task {task_id}...")
-            response =
- await asyncio.to_thread(
+            response = await asyncio.to_thread(
                 self.agent.run,
                 task=task
             )