o ®©Zh¿ã@s<dZddlmZddlmZddlmZGdd„deƒZdS)zWrapper around wikipedia API.é)ÚUnion)ÚDocument)ÚDocstorec@s4eZdZdZd dd„Zdedeeeffdd„ZdS) Ú WikipediazWikipedia API.ÚreturnNcCs&zddl}WdStytdƒ‚w)z*Check that wikipedia package is installed.rNzZCould not import wikipedia python package. Please install it with `pip install wikipedia`.)Ú wikipediaÚ ImportError)Úselfr©r ú]/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/docstore/wikipedia.pyÚ__init__ s ÿÿzWikipedia.__init__Úsearchc CsŽddl}z| |¡j}| |¡j}t|d|id}W|S|jy1d|›d| |¡›}Y|S|jyFd|›d| |¡›}Y|Sw)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. rNÚpage)Ú page_contentÚmetadatazCould not find [z ]. Similar: )rrÚcontentÚurlrZ PageErrorr ZDisambiguationError)r r rrrÚresultr r r r s   ÿüþþzWikipedia.search)rN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r Ústrrrr r r r r r s  rN)rÚtypingrZlangchain_core.documentsrZ!langchain_community.docstore.baserrr r r r Ús