
    dh                     B    S r SSKJr  SSKJr  SSKJr   " S S\5      rg)zWrapper around wikipedia API.    )Union)Document)Docstorec                   >    \ rS rSrSrS	S jrS\S\\\4   4S jr	Sr
g)
	Wikipedia
   zWikipedia API.returnNc                 @     SSK ng! [         a    [        S5      ef = f)z*Check that wikipedia package is installed.r   NzZCould not import wikipedia python package. Please install it with `pip install wikipedia`.)	wikipediaImportError)selfr   s     ^/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/docstore/wikipedia.py__init__Wikipedia.__init__   s,    	 	B 	s    searchc                 B   SSK n UR                  U5      R                  nUR                  U5      R                  n[	        USU0S9nU$ ! UR
                   a    SU SUR                  U5       3n U$ UR                   a    SU SUR                  U5       3n U$ f = f)zTry 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.
r   Npage)page_contentmetadatazCould not find [z]. Similar: )r   r   contenturlr   	PageErrorr   DisambiguationError)r   r   r   r   r   results         r   r   Wikipedia.search   s     			W$>>&199L..(,,C+3)VSM,F 	 "" 	W'x|I<L<LV<T;UVF  ,, 	W'x|I<L<LV<T;UVF	Ws   AA
 
'B4&BB )r	   N)__name__
__module____qualname____firstlineno____doc__r   strr   r   r   __static_attributes__r       r   r   r   
   s&    S U3=%9 r$   r   N)r!   typingr   langchain_core.documentsr   !langchain_community.docstore.baser   r   r   r$   r   <module>r(      s    #  - 6$ $r$   