NeoAnd/Docling
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(CLI): Expose code and formula models in the CLI (#820)

Browse Source 
feat: expose code and formula models in the CLI

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi 2025-01-28 06:26:03 +01:00 committed by GitHub
parent 4d41db3f7a
commit 6882e6c38d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
docling/cli/main.py
View File

@ -211,6 +211,14 @@ def convert(
TableFormerMode, TableFormerMode,
typer.Option(..., help="The mode to use in the table structure model."), typer.Option(..., help="The mode to use in the table structure model."),
] = TableFormerMode.FAST, ] = TableFormerMode.FAST,
enrich_code: Annotated[
bool,
typer.Option(..., help="Enable the code enrichment model in the pipeline."),
] = False,
enrich_formula: Annotated[
bool,
typer.Option(..., help="Enable the formula enrichment model in the pipeline."),
] = False,
artifacts_path: Annotated[ artifacts_path: Annotated[
Optional[Path], Optional[Path],
typer.Option(..., help="If provided, the location of the model artifacts."), typer.Option(..., help="If provided, the location of the model artifacts."),
@ -365,6 +373,8 @@ def convert(
do_ocr=ocr, do_ocr=ocr,
ocr_options=ocr_options, ocr_options=ocr_options,
do_table_structure=True, do_table_structure=True,
do_code_enrichment=enrich_code,
do_formula_enrichment=enrich_formula,
document_timeout=document_timeout, document_timeout=document_timeout,
) )
pipeline_options.table_structure_options.do_cell_matching = ( pipeline_options.table_structure_options.do_cell_matching = (