mirror of
https://github.com/hwchase17/langchain
synced 2024-11-04 06:00:26 +00:00
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
|
from typing import List
|
||
|
|
|
||
|
|
from langchain_community.document_loaders.unstructured import (
|
||
|
|
UnstructuredFileLoader,
|
||
|
|
satisfies_min_unstructured_version,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class UnstructuredEPubLoader(UnstructuredFileLoader):
|
||
|
|
"""Load `EPub` files using `Unstructured`.
|
||
|
|
|
||
|
|
You can run the loader in one of two modes: "single" and "elements".
|
||
|
|
If you use "single" mode, the document will be returned as a single
|
||
|
|
langchain Document object. If you use "elements" mode, the unstructured
|
||
|
|
library will split the document into elements such as Title and NarrativeText.
|
||
|
|
You can pass in additional unstructured kwargs after mode to apply
|
||
|
|
different unstructured settings.
|
||
|
|
|
||
|
|
Examples
|
||
|
|
--------
|
||
|
|
from langchain_community.document_loaders import UnstructuredEPubLoader
|
||
|
|
|
||
|
|
loader = UnstructuredEPubLoader(
|
||
|
|
"example.epub", mode="elements", strategy="fast",
|
||
|
|
)
|
||
|
|
docs = loader.load()
|
||
|
|
|
||
|
|
References
|
||
|
|
----------
|
||
|
|
https://unstructured-io.github.io/unstructured/bricks.html#partition-epub
|
||
|
|
"""
|
||
|
|
|
||
|
|
def _get_elements(self) -> List:
|
||
|
|
min_unstructured_version = "0.5.4"
|
||
|
|
if not satisfies_min_unstructured_version(min_unstructured_version):
|
||
|
|
raise ValueError(
|
||
|
|
"Partitioning epub files is only supported in "
|
||
|
|
f"unstructured>={min_unstructured_version}."
|
||
|
|
)
|
||
|
|
from unstructured.partition.epub import partition_epub
|
||
|
|
|
||
|
|
return partition_epub(filename=self.file_path, **self.unstructured_kwargs)
|