U
    ™ÀÂhç  ã                   @   s@   d dl mZmZmZ d dlmZ d dlmZ G dd„ deƒZdS )é    )ÚAsyncIteratorÚIteratorÚList)ÚDocument)Ú
BaseLoaderc                   @   sB   e Zd ZdZedœdd„Zee dœdd„Ze	e dœdd	„Z
d
S )ÚMergedDataLoaderz&Merge documents from a list of loaders©Úloadersc                 C   s
   || _ dS )z!Initialize with a list of loadersNr   )Úselfr	   © r   úN/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/document_loaders/merge.pyÚ__init__   s    zMergedDataLoader.__init__)Úreturnc              	   c   sJ   | j D ]>}z| ¡ }W n tk
r2   | ¡ }Y nX |D ]
}|V  q8qdS ©z+Lazy load docs from each individual loader.N)r	   Ú	lazy_loadÚNotImplementedErrorÚload)r
   ÚloaderÚdataÚdocumentr   r   r   r      s    
zMergedDataLoader.lazy_loadc                 C  s.   | j D ]"}| ¡ 2 z3 dH W }|V  q6 qdS r   )r	   Ú
alazy_load)r
   r   r   r   r   r   r      s    
zMergedDataLoader.alazy_loadN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   r   r   r   r   r      s   r   N)	Útypingr   r   r   Zlangchain_core.documentsr   Z)langchain_community.document_loaders.baser   r   r   r   r   r   Ú<module>   s   