NgX ZddlZddlmZddlZddlmZddlmZmZGddeZ dS)N)List)Document) BaseModelFieldceZdZUdZeed< eeZeed< dZ eed< dedefd Z dede e fd Z dede efd Zd S) BraveSearchWrapperz'Wrapper around the Brave search engine.api_key)default_factory search_kwargsz.https://api.search.brave.com/res/v1/web/searchbase_urlqueryreturncn||}d|D}tj|S)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 cg|]y}|d|ddtd|dg|dgdzS)titleurl N descriptionextra_snippets)rlinksnippet)getjoinfilter.0items f/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/utilities/brave_search.py z*BraveSearchWrapper.run..s    '**88txx 66XBRTV9W9WX      )_search_requestjsondumps)selfr web_search_results final_resultss rrunzBraveSearchWrapper.runsM"111>>   +    z-(((r cD||}d|DS)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. cg|]}tdtd|dg|dg|d|ddS)rNrrrr)rr) page_contentmetadata)rrrrrs rrz9BraveSearchWrapper.download_documents..5s      XXtxx 66XBRTV9W9WX $(88G#4#4dhhuooNN        r )r!)r$r resultss rdownload_documentsz%BraveSearchWrapper.download_documents+s:&&u--        r c|jdd}tj}i|j|dd}||j||jtdtj|j|}|j std|j | did 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")r$r r1reqparamsresponses rr!z"BraveSearchWrapper._search_requestAs$(L(  &((OD&O*N*NO  v... 7?KLL L<999{ B@(*>@@AA A}}""5"--11)R@@@r N)__name__ __module__ __qualname____doc__str__annotations__rdictr r r'rrr-r!r rrr s11 LLL9%555M4555EDHcDDD3))))))0  X    ,ASAT$ZAAAAAAr r) r"typingrr3langchain_core.documentsrpydanticrrrrDr rrHs ------%%%%%%%%GAGAGAGAGAGAGAGAGAGAr