Update agent.py

agent.py

@@ -1,5 +1,4 @@
-from smolagents import CodeAgent, LiteLLMModel, tool, Tool, load_tool, DuckDuckGoSearchTool
-, WikipediaSearchTool
+from smolagents import CodeAgent, LiteLLMModel, tool, Tool, load_tool, DuckDuckGoSearchTool, WikipediaSearchTool
 import asyncio
 import os
 import re
@@ -21,7 +20,8 @@ import logging
 logger = logging.getLogger(__name__)
 
 @tool
-def search_arxiv(query: str) -> str:
+def search_arxiv(query
+: str) -> str:
     """Search Arxiv for a query and return maximum 3 result.
     
     Args: