o ȳg@sDdZddlmZmZmZmZddlmZmZm Z GdddeZ dS)ziUtil that calls DuckDuckGo Search. No setup required. Free. https://pypi.org/project/duckduckgo-search/ )AnyDictListOptional) BaseModel ConfigDictmodel_validatorc @s*eZdZUdZdZeeed< dZeed< dZ eeed< dZ e ed <d Z eed < d Z eed < eddZeddededefddZ d ded ee deeeeffddZ d ded ee deeeeffddZdedefddZ d ded e d eedeeeeffddZdS)!DuckDuckGoSearchAPIWrapperzQWrapper for DuckDuckGo Search API. Free and does not require any setup. zwt-wtregionmoderate safesearchytime max_resultsautobackendtextsourceforbid)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)clsrrr k/var/www/html/chatdoc2/venv/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 rrrselfr#rrddgsddgs_genr r r! _ddgs_text9s$     z%DuckDuckGoSearchAPIWrapper._ddgs_textcCsxddlm}|)}|j||j|j|j|p|jd}|r*dd|DWdSWdgS1s5wYgS)z.N)rrnewsr r rrr+r r r! _ddgs_newsLs"     z%DuckDuckGoSearchAPIWrapper._ddgs_newscCsL|jdkr ||}n |jdkr||}ng}|sdSddd|DS)z=Run query through DuckDuckGo and return concatenated results.rr0*No good DuckDuckGo Search Result was found css|]}|dVqdS)bodyNr r&r r r! isz1DuckDuckGoSearchAPIWrapper.run..)rr/r1join)r,r#resultsr r r!run^s    zDuckDuckGoSearchAPIWrapper.runcCsh|p|j}|dkrdd|j||dD}n|dkr'dd|j||dD}ng}|dur2dd 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)r4titlehref)snippetr9linkr r&r r r!r)}sz6DuckDuckGoSearchAPIWrapper.results..)rr0cSs0g|]}|d|d|d|d|ddqS)r4r9urldater)r;r9r<r>rr r&r r r!r)sNResultr2)rr/r1)r,r#rrr7r r r!r7ks     z"DuckDuckGoSearchAPIWrapper.results)N)__name__ __module__ __qualname____doc__r rstr__annotations__r rrintrrr model_configr classmethodrrr"rr/r1r8r7r r r r!r s\        r N) rCtypingrrrrpydanticrrrr r r r r!s