agents/web_agent.py

"""
Web agent for finding and extracting song lyrics from online sources.
"""

from smolagents import CodeAgent, VisitWebpageTool
from loguru import logger

from config import AGENT_CONFIG, load_prompt_templates
from tools.search_tools import ThrottledDuckDuckGoSearchTool


def create_web_agent(model):
    """
    Create an agent specialized in web browsing and lyrics extraction.
    
    Args:
        model: The LLM model to use with this agent
        
    Returns:
        A configured CodeAgent for web searches
    """
    # Get configuration values
    config = AGENT_CONFIG['web_agent']
    prompt_templates = load_prompt_templates()
    
    # Create the throttled search tool
    throttled_search_tool = ThrottledDuckDuckGoSearchTool(
        min_delay=3.0, 
        max_delay=7.0
    )
    
    # Create and return the agent
    agent = CodeAgent(
        model=model,
        tools=[throttled_search_tool, VisitWebpageTool()],
        name="lyrics_search_agent",
        description=config['description'],
        additional_authorized_imports=["numpy", "bs4"],
        max_steps=config['max_steps'],
        verbosity_level=config['verbosity_level'],
        prompt_templates=prompt_templates
    )
    
    logger.info("Web agent (lyrics search) created successfully")
    return agent