7|hX RddlZddlmZddlZddlmZddlmZmZGddeZ y)N)List)Document) BaseModelFieldceZdZUdZeed< eeZeed< dZ eed< dedefd Z dede e fd Z dede efd Zy ) BraveSearchWrapperz'Wrapper around the Brave search engine.api_key)default_factory search_kwargsz.https://api.search.brave.com/res/v1/web/searchbase_urlqueryreturnc2|j|}|Dcgc]a}|jd|jddjtd|jdg|jdgdc}}t j |Scc}w) zQuery the Brave search engine and return the results as a JSON string. Args: query: The query to search for. Returns: The results as a JSON string. )r titleurl N descriptionextra_snippets)rlinksnippet)_search_requestgetjoinfilterjsondumps)selfr web_search_resultsitem final_resultss i/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/utilities/brave_search.pyrunzBraveSearchWrapper.runs"111>+  '*88txx 6XBRTV9WX    zz-((  sA&Bc|j|}|Dcgc]k}tdjtd|j dg|j dg|j d|j ddmc}Scc}w) zQuery the Brave search engine and return the results as a list of Documents. Args: query: The query to search for. Returns: The results as a list of Documents. rNrrrr)rr) page_contentmetadata)rrrrr)rr resultsrs r!download_documentsz%BraveSearchWrapper.download_documents+s&&u-     XXtxx 6XBRTV9WX $(88G#4dhhuoN     sA0B c|jdd}tj}i|j|dd}|j |j ||j tdtj|j |}|jstd|j|jjdijd gS) Nzapplication/json)zX-Subscription-TokenAcceptT)qrz,prepared url is None, this should not happen)headersz HTTP error webr&) r requestsPreparedRequestr prepare_urlr r ValueErrorrok Exception status_coder)rr r+reqparamsresponses r!rz"BraveSearchWrapper._search_requestAs$(LL( &&(OD&&O*NO  v. 77?KL L<<9{{k(*>*>)?@A A}}""5"-11)R@@N)__name__ __module__ __qualname____doc__str__annotations__rdictr r r"rrr'rr7r!rr sp1 L95M45EDHcD3)))0  X ,ASAT$ZAr7r) rtypingrr-langchain_core.documentsrpydanticrrrr?r7r!rCs$ -%GAGAr7