deeplearninglearner/First_agent_template

1from typing import Any, Optional2from smolagents.tools import Tool3import duckduckgo_search4 5class DuckDuckGoSearchTool(Tool):6    name = "web_search"7    description = "Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results."8    inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}9    output_type = "string"10 11    def __init__(self, max_results=10, **kwargs):12        super().__init__()13        self.max_results = max_results14        try:15            from duckduckgo_search import DDGS16        except ImportError as e:17            raise ImportError(18                "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."19            ) from e20        self.ddgs = DDGS(**kwargs)21 22    def forward(self, query: str) -> str:23        results = self.ddgs.text(query, max_results=self.max_results)24        if len(results) == 0:25            raise Exception("No results found! Try a less restrictive/shorter query.")26        postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]27        return "## Search Results\n\n" + "\n\n".join(postprocessed_results)28