Update agent.py

agent.py

@@ -1,4 +1,4 @@
-from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel, VisitWebpageTool, PythonInterpreterTool
+from smolagents import WebSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel, VisitWebpageTool, PythonInterpreterTool, FinalAnswerTool, UserInputTool
 from agent_tools import *
 import os
 
@@ -11,12 +11,22 @@ api_key = os.environ.get("GOOGLE_API_KEY")
 model = LiteLLMModel("gemini/gemini-2.5-flash",api_key=api_key)
 
 # Initialize tools
-search_tool = DuckDuckGoSearchTool(max_results=5)
+search_tool = WebSearchTool()
 wiki_search_tool = WikipediaSearchTool()
 visit_webpage_tool = VisitWebpageTool()
 python_interpreter_tool = PythonInterpreterTool()
+final_answer_tool = FinalAnswerTool()
+user_input_tool = UserInputTool()
 calculator_tools = [addition_tool, subtraction_tool, multiplication_tool, division_tool, exponent_tool, modulus_tool]
-main_tools = [search_tool, wiki_search_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,
+              final_answer_tool,
+              user_input_tool,
+             ]
 authorized_imports = ['requests', 'bs4', 'whisper', 'langchain_community', 'langchain', 'markdownify', 'openpyxl', 'tabulate']
 
 # Agent Definitions