Update search tool

agents/single_agent.py

@@ -3,13 +3,14 @@ Web agent for finding and extracting song lyrics from online sources.
 """
 
 from loguru import logger
-from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, FinalAnswerTool
+from smolagents import CodeAgent, VisitWebpageTool, FinalAnswerTool
 
 from tools.analysis_tools import analyze_lyrics_tool
+from tool.search_tools import ThrottledDuckDuckGoSearchTool
 
 
 def create_single_agent(model):
-    """
+    """ 
     Create an agent specialized in web browsing and lyrics extraction.
     
     Args:
@@ -24,16 +25,16 @@ def create_single_agent(model):
     agent = CodeAgent(
         tools=[
             FinalAnswerTool(),
-            DuckDuckGoSearchTool(),
+            ThrottledDuckDuckGoSearchTool(),
             VisitWebpageTool(),
             analyze_lyrics_tool
         ],
         model=model,
         additional_authorized_imports=['numpy', 'bs4', 'rich'],
-        max_steps=22,
-        verbosity_level=1,
-        description="Specialized agent for finding and extracting song, specified by the user. Does not use genius and AZlyrics for scraping since they have protection. Performs lyrics search and analysys using given tools. Provides detailed commentary using FinalAnswerTool meaning with beautiful human-readable format.",
+        max_steps=25,
+        verbosity_level=2,
+        description="Specialized agent for finding and extracting song, specified by the user. Does not use genius.com and AZlyrics.com for scraping since they have protection. Performs lyrics search and analysys using given tools. Provides detailed commentary using FinalAnswerTool meaning with beautiful human-readable format.",
     )
     
     logger.info("Web agent (lyrics search) created successfully")
-    return agent
+    return agent

tools/search_tools.py

@@ -6,7 +6,7 @@ import time
 import random
 from typing import Dict, List, Any
 from loguru import logger
-from smolagents import DuckDuckGoSearchTool, Tool
+from smolagents import DuckDuckGoSearchTool
 
 class ThrottledDuckDuckGoSearchTool(DuckDuckGoSearchTool):
     """
@@ -17,7 +17,7 @@ class ThrottledDuckDuckGoSearchTool(DuckDuckGoSearchTool):
     Each search request will be followed by a random delay within the specified range.
     """
     
-    def __init__(self, min_delay: float = 2.0, max_delay: float = 5.0, **kwargs):
+    def __init__(self, min_delay: float = 5.0, max_delay: float = 12.0, **kwargs):
         """
         Initialize the throttled search tool with delay parameters.