mirror of
https://github.com/DS4SD/docling.git
synced 2025-07-28 13:04:25 +00:00
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
import logging
|
|
|
|
from docling.backend.abstract_backend import (
|
|
AbstractDocumentBackend,
|
|
DeclarativeDocumentBackend,
|
|
)
|
|
from docling.datamodel.base_models import ConversionStatus
|
|
from docling.datamodel.document import ConversionResult, InputDocument
|
|
from docling.datamodel.pipeline_options import PdfPipelineOptions, PipelineOptions
|
|
from docling.pipeline.base_model_pipeline import BaseModelPipeline
|
|
|
|
_log = logging.getLogger(__name__)
|
|
|
|
|
|
class SimpleModelPipeline(BaseModelPipeline):
|
|
"""SimpleModelPipeline.
|
|
|
|
This class is used at the moment for formats / backends
|
|
which produce straight DoclingDocument output.
|
|
"""
|
|
|
|
def __init__(self, pipeline_options: PdfPipelineOptions):
|
|
super().__init__(pipeline_options)
|
|
|
|
def execute(self, in_doc: InputDocument) -> ConversionResult:
|
|
conv_res = ConversionResult(input=in_doc)
|
|
|
|
_log.info(f"Processing document {in_doc.file.name}")
|
|
|
|
if not in_doc.valid:
|
|
conv_res.status = ConversionStatus.FAILURE
|
|
return conv_res
|
|
|
|
if not isinstance(in_doc._backend, DeclarativeDocumentBackend):
|
|
raise RuntimeError(
|
|
f"The selected backend {type(in_doc._backend).__name__} for {in_doc.file} is not a declarative backend. "
|
|
f"Can not convert this with simple pipeline. "
|
|
f"Please check your format configuration on DocumentConverter."
|
|
)
|
|
# conv_res.status = ConversionStatus.FAILURE
|
|
# return conv_res
|
|
|
|
# Instead of running a page-level pipeline to build up the document structure,
|
|
# the backend is expected to be of type DeclarativeDocumentBackend, which can output
|
|
# a DoclingDocument straight.
|
|
|
|
conv_res.output = in_doc._backend.convert()
|
|
|
|
# Do other stuff with conv_res.experimental
|
|
|
|
conv_res = self.assemble_document(in_doc, conv_res)
|
|
|
|
conv_res.status = ConversionStatus.SUCCESS
|
|
|
|
return conv_res
|
|
|
|
def assemble_document(
|
|
self, in_doc: InputDocument, conv_res: ConversionResult
|
|
) -> ConversionResult:
|
|
return conv_res
|
|
|
|
@classmethod
|
|
def get_default_options(cls) -> PipelineOptions:
|
|
return PipelineOptions()
|
|
|
|
@classmethod
|
|
def is_backend_supported(cls, backend: AbstractDocumentBackend):
|
|
return isinstance(backend, DeclarativeDocumentBackend)
|