tools/web_search.py

from typing import Any, Optional
from smolagents.tools import Tool

class DuckDuckGoSearchTool(Tool):
    name = "web_search"
    description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."
    inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}
    output_type = "string"

    def __init__(self, max_results=10, **kwargs):
        super().__init__()
        self.max_results = max_results
        try:
            # Preferred package name (new)
            from ddgs import DDGS
        except ImportError as e:
            try:
                # Backward-compatible fallback (old package name)
                from duckduckgo_search import DDGS
            except ImportError as inner_e:
                raise ImportError(
                    "You must install package `ddgs` (or legacy `duckduckgo_search`) to run this tool: for instance run `pip install ddgs`."
                ) from inner_e
        self.ddgs = DDGS(**kwargs)

    def forward(self, query: str) -> str:
        results = list(self.ddgs.text(query, max_results=self.max_results))
        if len(results) == 0:
            raise Exception("No results found! Try a less restrictive/shorter query.")
        postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
        return "## Search Results\n\n" + "\n\n".join(postprocessed_results)