Add DuckDuckGoSearchTool and VisitWebpageTool to agent initialization in main.py. Update tools configuration to enhance agent functionality and enable visualization of tools used during execution.

main.py

@@ -12,6 +12,10 @@ import litellm
 from smolagents import CodeAgent, LiteLLMModel
 from smolagents.memory import ActionStep, FinalAnswerStep
 from smolagents.monitoring import LogLevel
+from smolagents.default_tools import (
+    DuckDuckGoSearchTool,
+    VisitWebpageTool,
+)
 
 litellm._turn_on_debug()
 
@@ -54,18 +58,29 @@ except Exception as e:
     logger.error(f"Failed to initialize model: {str(e)}")
     raise
 
+tools = [
+    DuckDuckGoSearchTool(max_results=3),
+    # VisitWebpageTool(max_output_length=1000),
+]
+
 # Initialize agent with error handling
 try:
     agent = CodeAgent(
-        add_base_tools=True,
+        # add_base_tools=True,
         additional_authorized_imports=["pandas", "numpy"],
         max_steps=10,
         model=model,
-        tools=[],
+        tools=tools,
         step_callbacks=None,
         verbosity_level=LogLevel.ERROR,
     )
     agent.logger.console.width = 66
+
+    agent.visualize()
+
+    tools = agent.tools
+    print(f"Tools: {tools}")
+
 except Exception as e:
     logger.error(f"Failed to initialize agent: {str(e)}")
     raise