Updated back to docs further

app.py

@@ -418,28 +418,41 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
+# Initialize the agent using standard smolagents patterns
 agent = CodeAgent(
     model=model,
     tools=[
-        final_answer, 
+        final_answer,
         Sonar_Web_Search_Tool,
-        primary_search_tool,  # Use whichever search tool was successfully initialized
-        get_current_time_in_timezone, 
+        primary_search_tool,  # This is already set to either DuckDuckGo, Google, or fallback
+        get_current_time_in_timezone,
         image_generation_tool,
         Dataset_Creator_Tool,
         Check_Dataset_Validity,
-        visit_webpage_tool,
-    ],  
+        visit_webpage_tool,  # This is correctly initialized as VisitWebpageTool()
+    ],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
+    planning_interval=3,
+    name="Research Assistant",
+    description="""An AI assistant that can search the web, create datasets, and answer questions # Note about working within token limits
+                # When using with queries that might exceed token limits, consider:
+                # 1. Breaking tasks into smaller sub-tasks
+                # 2. Limiting the amount of data returned by search tools
+                # 3. Using the planning_interval to enable more effective reasoning""",
     prompt_templates=prompt_templates
 )
 
+# Add informative message about which search tool is being used
 print(f"Agent initialized with {search_tool_name} as primary search tool")
+print(f"Available tools: final_answer, Sonar_Web_Search_Tool, {search_tool_name}, get_current_time_in_timezone, image_generation_tool, Dataset_Creator_Tool, Check_Dataset_Validity, visit_webpage_tool")
+
+# Note about working within token limits - add this comment
+# When using with queries that might exceed token limits, consider:
+# 1. Breaking tasks into smaller sub-tasks
+# 2. Limiting the amount of data returned by search tools
+# 3. Using the planning_interval to enable more effective reasoning
 
 # To fix the TypeError in Gradio_UI.py, you would need to modify that file
 # For now, we'll just use the agent directly