a bg@sTddlZddlmZddlZddlmZddlmZddlm Z GdddeZ dS)N)List)Document) BaseModel)URLc@sVeZdZUdZdZeed<eedddZeee dddZ eee dd d Z d S) JinaSearchAPIWrapperz&Wrapper around the Jina search engine.zhttps://s.jina.ai/base_url)queryreturncCs$|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 linkZsnippetr )get.0itemrw/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/jina_search.py s z,JinaSearchAPIWrapper.run..)_search_requestjsondumps)selfrZweb_search_resultsZ 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|]0}t|d|d|d|dddqS)r r r r )r rr )Z page_contentmetadata)rrrrrrr/s z;JinaSearchAPIWrapper.download_documents..)r)rrresultsrrrdownload_documents%s  z'JinaSearchAPIWrapper.download_documentscCsNddi}tt|j|}tj||d}|js>td|j|dgS)NAcceptzapplication/json)headersz HTTP error data) strrrrequestsrok Exception status_coder)rrrr responserrrr;sz$JinaSearchAPIWrapper._search_requestN) __name__ __module__ __qualname____doc__rr!__annotations__rrrrdictrrrrrr s  r) rtypingrr"Zlangchain_core.documentsrZpydanticrZyarlrrrrrrs