Update agent.py

agent.py

@@ -2,7 +2,7 @@ import os
 import dotenv
 import yaml
 from smolagents import CodeAgent, LiteLLMModel, OpenAIServerModel, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
-from tools import tool_read_files, tool_reverse_string, tool_download_image 
+from tools import ToolReadFiles, ToolReverseString, ToolDownloadImage, FinalAnswerTool 
 
 dotenv.load_dotenv()
 
@@ -35,7 +35,7 @@ class Agent:
         self.agent = CodeAgent(
             model=self.model,
             add_base_tools=True,
-            tools=[tool_read_files, tool_reverse_string, tool_download_image],
+            tools=[ToolReadFiles, ToolReverseString, ToolDownloadImage, FinalAnswerTool],
             max_steps=10,
             additional_authorized_imports=['numpy','beautifulsoup4','wiki','re','pandas']
         )