Update agent description to avoid Genius and AZLyrics websites due to protection

- Modified description in single_agent.py to explicitly mention avoidance of Genius and AZLyrics
- Added 'rich' to authorized imports for better formatting
- Cleaned up code structure and imports

agents/single_agent.py

@@ -5,9 +5,7 @@ Web agent for finding and extracting song lyrics from online sources.
 from loguru import logger
 from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, FinalAnswerTool
 
-from config import load_prompt_templates
 from tools.analysis_tools import analyze_lyrics_tool
-from tools.search_tools import ThrottledDuckDuckGoSearchTool
 
 
 def create_single_agent(model):
@@ -20,8 +18,6 @@ def create_single_agent(model):
     Returns:
         A configured CodeAgent for web searches
     """
-    # Load prompt templates
-    prompt_templates = load_prompt_templates()
     
     # Define agent parameters directly in the code instead of using config
     # Example usage within the agent
@@ -33,10 +29,10 @@ def create_single_agent(model):
             analyze_lyrics_tool
         ],
         model=model,
-        additional_authorized_imports=['numpy', 'bs4'],
+        additional_authorized_imports=['numpy', 'bs4', 'rich'],
         max_steps=22,
         verbosity_level=1,
-        description="Specialized agent for finding and extracting song, specified by the user. Performs lyrics search and analysys using given tools. Provides detailed commentary using FinalAnswerTool meaning.",
+        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.",
     )
     
     logger.info("Web agent (lyrics search) created successfully")

tools/search_tools.py

@@ -57,39 +57,3 @@ class ThrottledDuckDuckGoSearchTool(DuckDuckGoSearchTool):
             logger.error(f"Error in DuckDuckGo search: {str(e)}")
             # Return empty results on error to allow the agent to continue
             return [{"title": "Search error", "href": "", "body": f"Error performing search: {str(e)}"}]
-
-
-class LyricsSearchTool(Tool):
-    """
-    Uses web search to find song lyrics based on song title and artist name
-
-    The search query should include the song title and artist name. The tool
-    will return the lyrics of the song if found.
-
-    Parameters
-    ----------
-    query : str
-        The search query for finding song lyrics. Should include song title and artist name.
-
-    Returns
-    -------
-    str
-        The lyrics of the song if found, otherwise an empty string.
-    """
-    name = "lyrics_search_tool"
-    description = "Uses web search to find song lyrics based on song title and artist name"
-    inputs = {
-        "query": {
-            "type": "string",
-            "description": "The search query for finding song lyrics. Should include song title and artist name.",
-        }
-    }
-    output_type = "string"
-
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-
-    def forward(self, query: str) -> str:
-        assert isinstance(query, str), "Your search query must be a string"
-        # TODO: Implement lyrics search functionality
-        return "Lyrics search not implemented yet"