a bg@s<dZddlmZddlmZddlmZGdddeZdS)zWrapper around wikipedia API.)Union)Document)Docstorec@s6eZdZdZddddZeeeefdddZdS) WikipediazWikipedia API.N)returncCs,z ddl}Wnty&tdYn0dS)z*Check that wikipedia package is installed.rNzZCould not import wikipedia python package. Please install it with `pip install wikipedia`.) wikipedia ImportError)selfrr t/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/docstore/wikipedia.py__init__ s   zWikipedia.__init__)searchrc Csddl}z,||j}||j}t|d|id}WnR|jy^d|d||}Yn*|jyd|d||}Yn0|S)aTry to search for wiki page. If page exists, return the page summary, and a PageWithLookups object. If page does not exist, return similar entries. Args: search: search string. Returns: a Document object or error message. rNpage) page_contentmetadatazCould not find [z ]. Similar: )rrcontenturlrZ PageErrorr ZDisambiguationError)r r rrrresultr r r r s    zWikipedia.search) __name__ __module__ __qualname____doc__r strrrr r r r r r s rN)rtypingrZlangchain_core.documentsrZ!langchain_community.docstore.baserrr r r r s