Source code for langchain_community.document_loaders.merge

from typing import AsyncIterator, Iterator, List

from langchain_core.documents import Document

from langchain_community.document_loaders.base import BaseLoader


class MergedDataLoader(BaseLoader):
    """从加载器列表中合并文档"""

    def __init__(self, loaders: List):
        """使用加载器列表进行初始化"""
        self.loaders = loaders

    def lazy_load(self) -> Iterator[Document]:
        """从每个单独的加载器中延迟加载文档。"""
        for loader in self.loaders:
            # Check if lazy_load is implemented
            try:
                data = loader.lazy_load()
            except NotImplementedError:
                data = loader.load()
            for document in data:
                yield document

    async def alazy_load(self) -> AsyncIterator[Document]:
        """从每个单独的加载器中延迟加载文档。"""
        for loader in self.loaders:
            async for document in loader.alazy_load():
                yield document