Skip to content

Commit c7087f0

Browse files
committed
feature(extract-core): add document timeout to docling
1 parent 905f6ec commit c7087f0

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

‎extract-core/extract_core/docling_.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,9 @@
2323
from docling.datamodel.settings import (
2424
BatchConcurrencySettings as DoclingBatchConcurrencySettings,
2525
)
26+
from docling.datamodel.settings import DebugSettings
2627
from docling.datamodel.settings import (
27-
DebugSettings,
28-
InferenceSettings,
28+
InferenceSettings as DoclingInferenceSettings,
2929
)
3030
from icij_common.pydantic_utils import (
3131
merge_configs,
@@ -270,6 +270,10 @@ class BatchConcurrencySettings(DoclingBatchConcurrencySettings):
270270
max_page_batches: int = 2
271271

272272

273+
class InferenceSettings(DoclingInferenceSettings):
274+
document_timeout: float | None = None
275+
276+
273277
class DoclingSettings(BaseModel):
274278
perf: BatchConcurrencySettings = Field(default_factory=BatchConcurrencySettings)
275279
debug: DebugSettings = Field(default_factory=DebugSettings)

0 commit comments

Comments
 (0)