o ȳgX @sLddlZddlmZddlZddlmZddlmZmZGdddeZ dS)N)List)Document) BaseModelFieldc@s|eZdZUdZeed< eedZeed< dZ eed< dedefd d Z dede e fd d Z dede efd dZdS)BraveSearchWrapperz'Wrapper around the Brave search engine.api_key)default_factory search_kwargsz.https://api.search.brave.com/res/v1/web/searchbase_urlqueryreturncCs$|j|d}dd|D}t|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 c SsDg|]}|d|ddtd|dg|dgdqS)titleurl N descriptionextra_snippets)r linksnippet)getjoinfilter.0itemrf/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/brave_search.py s z*BraveSearchWrapper.run..)_search_requestjsondumps)selfr web_search_results final_resultsrrrruns   zBraveSearchWrapper.runcCs||}dd|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. c SsJg|]!}tdtd|dg|dg|d|dddqS)rNrrr r)r r) page_contentmetadata)rrrrrrrrr5s z9BraveSearchWrapper.download_documents..)r)r r resultsrrrdownload_documents+s  z%BraveSearchWrapper.download_documentscCs|jdd}t}i|j|dd}||j||jdur$tdtj|j|d}|j s7t d|j | did gS) Nzapplication/json)zX-Subscription-TokenAcceptT)qrz,prepared url is None, this should not happen)headersz HTTP error webr&) rrequestsPreparedRequestr prepare_urlr r ValueErrorrok Exception status_coder)r r r*reqparamsresponserrrrAs z"BraveSearchWrapper._search_requestN)__name__ __module__ __qualname____doc__str__annotations__rdictr r r#rrr'rrrrrr s  r) rtypingrr,langchain_core.documentsrpydanticrrrrrrrs