o ȳg@sTddlZddlmZddlZddlmZddlmZddlm Z GdddeZ dS)N)List)Document) BaseModel)URLc@s^eZdZUdZdZeed< dedefddZdedee fdd Z dedee fd d Z d S) JinaSearchAPIWrapperz&Wrapper around the Jina search engine.zhttps://s.jina.ai/base_urlqueryreturncCs$|j|d}dd|D}t|S)zQuery the Jina search engine and return the results as a JSON string. Args: query: The query to search for. Returns: The results as a JSON string. )rcSs2g|]}|d|d|d|ddqS)titleurl descriptioncontent)r linksnippetr )get.0itemre/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/jina_search.py sz,JinaSearchAPIWrapper.run..)_search_requestjsondumps)selfrweb_search_results final_resultsrrrruns  zJinaSearchAPIWrapper.runcCs||}dd|DS)zQuery the Jina 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 Ss8g|]}t|d|d|d|dddqS)r r r r )r rr ) page_contentmetadata)rrrrrrr/s z;JinaSearchAPIWrapper.download_documents..)r)rrresultsrrrdownload_documents%s  z'JinaSearchAPIWrapper.download_documentscCsNddi}tt|j|}tj||d}|jstd|j|dgS)NAcceptzapplication/json)headersz HTTP error data) strrrrequestsrok Exception status_coder)rrr#r responserrrr;sz$JinaSearchAPIWrapper._search_requestN) __name__ __module__ __qualname____doc__rr%__annotations__rrrr!dictrrrrrr s  r) rtypingrr&langchain_core.documentsrpydanticryarlrrrrrrs