import importlib

import yaml
from smolagents import CodeAgent

from tools import browse_webpage, extract_dates, find_in_page, web_search


def create_web_agent(model):
    """
    Create a specialized agent for web browsing tasks.

    Args:
        model: The model to use for the agent

    Returns:
        Configured CodeAgent for web browsing
    """
    # Load default prompts
    prompt_templates = yaml.safe_load(
        importlib.resources.files("smolagents.prompts")
        .joinpath("code_agent.yaml")
        .read_text()
    )

    web_agent = CodeAgent(
        tools=[web_search, browse_webpage, find_in_page, extract_dates],
        model=model,
        name="web_agent",
        description="Specialized agent for web browsing and searching. Use this agent to find information online, browse websites, and extract information from web pages.",
        add_base_tools=True,
        additional_authorized_imports=["requests", "bs4", "re", "json"],
        prompt_templates=prompt_templates,
    )

    return web_agent