U di @sLddlZddlZddlmZddlmZGdddeZGdddeZdS) N)RequestException)Toolc@s6eZdZdZdZddddiZdZeedd d Zd S) DuckDuckGoSearchToolZ web_searchzPerform a web search based on your query (think a Google search) then returns the top search results as a list of dict elements. Each result has keys 'title', 'href' and 'body'.querystringzThe search query to perform.type descriptionany)rreturncCsBzddlm}Wntk r,tdYnX|j|dd}|S)Nr)DDGSzpYou must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`.)Z max_results)Zduckduckgo_searchr ImportErrortext)selfrr resultsr>/tmp/pip-unpacked-wheel-bm_b0l5e/transformers/agents/search.pyforward s zDuckDuckGoSearchTool.forwardN __name__ __module__ __qualname__namer inputs output_typestrrrrrrrs rc@s6eZdZdZdZddddiZdZeeddd Zd S) VisitWebpageToolZ visit_webpagezOVisits a wbepage at the given url and returns its content as a markdown string.urlrz The url of the webpage to visit.r)rr c Cszddlm}Wntk r,tdYnXz4t|}|||j}tdd|}|WSt k r}zdt |WYSd}~XYn4t k r}zdt |WYSd}~XYnXdS)Nr) markdownifyzdYou must install package `markdownify` to run this tool: for instance run `pip install markdownify`.z\n{3,}z zError fetching the webpage: zAn unexpected error occurred: ) r rrequestsgetraise_for_statusrstripresubrr Exception)rrr responseZmarkdown_contenterrrr6s   "zVisitWebpageTool.forwardNrrrrrr+sr)r%r!Zrequests.exceptionsrZtoolsrrrrrrrs