Refactored agent.py

agent.py

@@ -12,11 +12,12 @@ model = LiteLLMModel("gemini/gemini-2.5-flash",api_key=api_key)
 
 # Initialize tools
 search_tool = DuckDuckGoSearchTool(max_results=5)
-wiki_tool = WikipediaSearchTool()
+wiki_search_tool = WikipediaSearchTool()
 visit_webpage_tool = VisitWebpageTool()
 python_interpreter_tool = PythonInterpreterTool()
 calculator_tools = [addition_tool, subtraction_tool, multiplication_tool, division_tool, exponent_tool, modulus_tool]
-main_tools = [search_tool, wiki_tool, transcriber_tool, visit_webpage_tool, python_interpreter_tool, excel_loader_tool]
+main_tools = [search_tool, wiki_search_tool, transcriber_tool, visit_webpage_tool, python_interpreter_tool, excel_loader_tool]
+authorized_imports = ['requests', 'bs4', 'whisper', 'langchain_community', 'langchain', 'markdownify', 'openpyxl', 'tabulate']
 
 # Agent Definitions
 calculator_agent = CodeAgent(
@@ -24,16 +25,16 @@ calculator_agent = CodeAgent(
     tools = calculator_tools,
     name = "arithmetic_calculator_agent",
     description = "You perform basic arithmetic calculations between two numbers.",
-    max_steps = 7,
-    verbosity_level=3
+    max_steps = 4,
+    verbosity_level=3,
 )
 
 main_agent = CodeAgent(
     tools=main_tools,
     model=model,
-    additional_authorized_imports = ['requests', 'bs4', 'whisper', 'langchain_community', 'langchain', 'markdownify', 'openpyxl'],
-    description = system_prompt,
-    managed_agents = [calculator_agent],
-    max_steps=10,
+    additional_authorized_imports=authorized_imports,
+    instructions=system_prompt,
+    managed_agents=[calculator_agent],
+    max_steps=8,
     verbosity_level=3,
 )