o  h @sLddlZddlZddlmZddlmZGdddeZGdddeZdS) N)RequestException)Toolc@s8eZdZdZdZddddiZdZdedefd d Zd S) DuckDuckGoSearchTool 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 descriptionanyreturncCs<zddlm}Wn tytdw|j|dd}|S)Nr)DDGSzpYou must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`.) max_results)duckduckgo_searchr ImportErrortext)selfrrresultsrn/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/transformers/agents/search.pyforward s zDuckDuckGoSearchTool.forwardN __name__ __module__ __qualname__namer inputs output_typestrrrrrrrs rc@s8eZdZdZdZddddiZdZdedefdd Zd S) VisitWebpageTool visit_webpagezOVisits a webpage at the given url and returns its content as a markdown string.urlrz The url of the webpage to visit.r r c Cszddlm}Wn tytdwzt|}|||j}tdd|}|WSt yG}z dt |WYd}~Sd}~wt y_}z dt |WYd}~Sd}~ww)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)rr#r$responsemarkdown_contenterrrr6s(  zVisitWebpageTool.forwardNrrrrrr!+sr!)r)r%requests.exceptionsrtoolsrrr!rrrrs