o
    Zh                     @   sR   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 G dd dee	Z
dS )	    )List)CallbackManagerForRetrieverRun)Document)BaseRetriever)PubMedAPIWrapperc                   @   s*   e Zd ZdZdededee fddZdS )PubMedRetrieverz`PubMed API` retriever.

    It wraps load() to get_relevant_documents().
    It uses all PubMedAPIWrapper arguments without any change.
    queryrun_managerreturnc                C   s   | j |dS )N)r   )Z	load_docs)selfr   r	    r   \/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/retrievers/pubmed.py_get_relevant_documents   s   z'PubMedRetriever._get_relevant_documentsN)	__name__
__module____qualname____doc__strr   r   r   r   r   r   r   r   r   
   s    r   N)typingr   Zlangchain_core.callbacksr   Zlangchain_core.documentsr   Zlangchain_core.retrieversr   Z$langchain_community.utilities.pubmedr   r   r   r   r   r   <module>   s    