Skip to content

pdf_loader

PdfLoader

Bases: BaseFileLoader

Source code in griptape/loaders/pdf_loader.py 
12
13
14
15
16
17
18
19
20
21
22
23
24
@define
class PdfLoader(BaseFileLoader):
    def try_parse(
        self,
        data: bytes,
        *,
        password: str | None = None,
    ) -> ListArtifact:
        pypdf = import_optional_dependency("pypdf")
        reader = pypdf.PdfReader(BytesIO(data), strict=True, password=password)
        pages = [TextArtifact(p.extract_text()) for p in reader.pages]

        return ListArtifact(pages)

try_parse(data, *, password=None)

Source code in griptape/loaders/pdf_loader.py 
14
15
16
17
18
19
20
21
22
23
24
def try_parse(
    self,
    data: bytes,
    *,
    password: str | None = None,
) -> ListArtifact:
    pypdf = import_optional_dependency("pypdf")
    reader = pypdf.PdfReader(BytesIO(data), strict=True, password=password)
    pages = [TextArtifact(p.extract_text()) for p in reader.pages]

    return ListArtifact(pages)