Source code for langchain_community.document_loaders.org_mode

from pathlib import Path
from typing import Any, List, Union

from langchain_community.document_loaders.unstructured import (
    UnstructuredFileLoader,
    validate_unstructured_version,
)


class UnstructuredOrgModeLoader(UnstructuredFileLoader):
    """使用`Unstructured`加载`Org-Mode`文件。

    您可以以两种模式之一运行加载程序："single"和"elements"。
    如果使用"single"模式，文档将作为单个langchain Document对象返回。
    如果使用"elements"模式，unstructured库将文档拆分为诸如Title和NarrativeText之类的元素。
    您可以在模式之后传递额外的unstructured kwargs以应用不同的unstructured设置。

    示例
    --------
    from langchain_community.document_loaders import UnstructuredOrgModeLoader

    loader = UnstructuredOrgModeLoader(
        "example.org", mode="elements", strategy="fast",
    )
    docs = loader.load()

    参考资料
    ----------
    https://unstructured-io.github.io/unstructured/bricks.html#partition-org"""

    def __init__(
        self,
        file_path: Union[str, Path],
        mode: str = "single",
        **unstructured_kwargs: Any,
    ):
        """参数：
    file_path：要加载的文件路径。
    mode：从中加载文件的模式。默认为"single"。
    **unstructured_kwargs：要传递给非结构化部分的任何额外关键字参数。
"""
        validate_unstructured_version(min_unstructured_version="0.7.9")
        super().__init__(file_path=file_path, mode=mode, **unstructured_kwargs)

    def _get_elements(self) -> List:
        from unstructured.partition.org import partition_org

        return partition_org(filename=self.file_path, **self.unstructured_kwargs)