o Zh @sDdZddlmZmZmZmZddlmZmZm Z GdddeZ dS)ziUtil that calls DuckDuckGo Search. No setup required. Free. https://pypi.org/project/duckduckgo-search/ )AnyDictListOptional) BaseModel ConfigDictmodel_validatorc @sTeZdZUdZdZeeed< dZeed< dZ eeed< dZ e ed <d Z eed < d Z eed < eddZeddededefddZ d"ded ee deeeeffddZ d"ded ee deeeeffddZ d"ded ee deeeeffddZdedefddZ d"ded e d eedeeeeffd d!ZdS)#DuckDuckGoSearchAPIWrapperzQWrapper for DuckDuckGo Search API. Free and does not require any setup. zwt-wtregionZmoderate safesearchytime max_resultsautobackendtextsourceZforbid)extrabefore)modevaluesreturncCs*z ddlm}W|Stytdw)z3Validate that python package exists in environment.rDDGSzmCould not import duckduckgo-search python package. Please install it with `pip install -U duckduckgo-search`.)duckduckgo_searchr ImportError)clsrrrf/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/utilities/duckduckgo_search.pyvalidate_environment,s z/DuckDuckGoSearchAPIWrapper.validate_environmentNqueryc Cs|ddlm}|+}|j||j|j|j|p|j|jd}|r,dd|DWdSWdgS1s7wYgS)zIz9DuckDuckGoSearchAPIWrapper._ddgs_text..N)rrrr r r rrselfr!rrZddgsZddgs_genrrr _ddgs_text9s$     z%DuckDuckGoSearchAPIWrapper._ddgs_textcCsxddlm}|)}|j||j|j|j|p|jd}|r*dd|DWdSWdgS1s5wYgS)z.N)rrnewsr r r rr)rrr _ddgs_newsLs"     z%DuckDuckGoSearchAPIWrapper._ddgs_newscCstddlm}|'}|j||j|j|p|jd}|r(dd|DWdSWdgS1s3wYgS)z=Run query through DuckDuckGo image search and return results.rr)r r rcSr#rrr$rrrr'lr(z;DuckDuckGoSearchAPIWrapper._ddgs_images..N)rrimagesr r rr)rrr _ddgs_images^s     z'DuckDuckGoSearchAPIWrapper._ddgs_imagescCsb|jdkr ||}n|jdkr||}n |jdkr!||}ng}|s'dSddd|DS)z=Run query through DuckDuckGo and return concatenated results.rr,r.*No good DuckDuckGo Search Result was found css|]}|dVqdS)bodyNrr$rrr |sz1DuckDuckGoSearchAPIWrapper.run..)rr+r-r/join)r*r!resultsrrrrunos      zDuckDuckGoSearchAPIWrapper.runcCs|p|j}|dkrdd|j||dD}n$|dkr'dd|j||dD}n|dkr8dd|j||dD}ng}|d urCd d ig}|S) aRun query through DuckDuckGo and return metadata. Args: query: The query to search for. max_results: The number of results to return. source: The source to look from. Returns: A list of dictionaries with the following keys: snippet - The description of the result. title - The title of the result. link - The link to the result. rcSs$g|]}|d|d|ddqS)r2titlehref)snippetr7linkrr$rrrr'sz6DuckDuckGoSearchAPIWrapper.results..)rr,cSs0g|]}|d|d|d|d|ddqS)r2r7urldater)r9r7r:r<rrr$rrrr'sr.c Ss<g|]}|d|d|d|d|d|d|ddqS)r7 thumbnailimager;heightwidthr)r7r=r>r;r?r@rrr$rrrr's NResultr0)rr+r-r/)r*r!rrr5rrrr5~s"       z"DuckDuckGoSearchAPIWrapper.results)N)__name__ __module__ __qualname____doc__r rstr__annotations__r r rintrrrZ model_configr classmethodrrr rr+r-r/r6r5rrrrr sl         r N) rEtypingrrrrZpydanticrrrr rrrrs