昺h4 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|]A}|d|d|ddBS)titleurl description)rlinksnippet)get.0items r/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/utilities/brave_search.py z*BraveSearchWrapper.run..s\    '**88M22     )_search_requestjsondumps)selfr web_search_results final_resultss rrunzBraveSearchWrapper.runsM"111>>  +    z-(((rcD||}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. c g|]O}t|d|d|ddPS)rrr)rr) page_contentmetadata)rrrs rrz9BraveSearchWrapper.download_documents..1si    !XXm44#'88G#4#4dhhuooNN      r)r)r r resultss rdownload_documentsz%BraveSearchWrapper.download_documents's:&&u--       rc|jdd}tj}i|jd|i}||j||jtdtj|j|}|j std|j | didgS) Nzapplication/json)zX-Subscription-TokenAcceptqz,prepared url is None, this should not happen)headersz HTTP error webr() r requestsPreparedRequestr prepare_urlr r ValueErrorrok Exception status_coder)r r r-reqparamsresponses rrz"BraveSearchWrapper._search_request9s$(L(  &((7D&73,7  v... 7?KLL L<999{ B@(*>@@AA A}}""5"--11)R@@@rN)__name__ __module__ __qualname____doc__str__annotations__rdictr r r#rrr)rrrrr s11 LLL9%555M4555EDHcDDD3))))))(  X    $ASAT$ZAAAAAArr) rtypingrr/langchain_core.documentsrlangchain_core.pydantic_v1rrrr@rrrDs ------77777777?A?A?A?A?A?A?A?A?A?Ar