Commint 3d

app.py

@@ -6,7 +6,8 @@ import pandas as pd
 
 from PIL import Image
 from typing import Optional, Tuple, Any
-from smolagents import CodeAgent,DuckDuckGoSearchTool,FinalAnswerTool,InferenceClientModel,VisitWebpageTool,tool,AzureOpenAIServerModel
+from smolagents import CodeAgent,FinalAnswerTool,InferenceClientModel,tool,AzureOpenAIServerModel
+from tools import wiki_search,web_search,get_current_time_in_timezone,visit_webpage,calculator_add,ocr,parse_excel,arxiv_search,
 from smolagents.tools import Tool
 
 # (Keep Constants as is)
@@ -32,8 +33,6 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initializing...")
 
-        duckduckgo=DuckDuckGoSearchTool()
-        visit_page=VisitWebpageTool()
         final_answer=FinalAnswerTool()
         
         model = AzureOpenAIServerModel(
@@ -45,9 +44,10 @@ class BasicAgent:
         
         self.smart_agent = CodeAgent(
             model=model,
-            tools=[duckduckgo,visit_page,final_answer],
-            additional_authorized_imports=["pandas"],
-            max_steps=20,
+            tools=[wiki_search,web_search,get_current_time_in_timezone,visit_webpage,calculator_add,ocr,parse_excel,arxiv_search,final_answer],
+            additional_authorized_imports=["pandas","bs4"],
+            max_steps=5,
+            verbisity_level=1
             )
         print("BasicAgent initialized.")