from smolagents import CodeAgent, VisitWebpageTool, DuckDuckGoSearchTool, WikipediaSearchTool, Model

from config import authorized_libraries


def create_web_agent(model: Model) -> CodeAgent:
    return CodeAgent(
        model=model,
        tools=[
            DuckDuckGoSearchTool(),
            VisitWebpageTool(),
            WikipediaSearchTool(),
        ],
        add_base_tools=True,
        additional_authorized_imports=authorized_libraries,
        name="web_agent",
        description="Browses the web to find information, can also look for information using the search engine DuckDuckGo, Wikipedia, etc",
        verbosity_level=0,
        max_steps=8,
    )