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

feat(cli): add option for downloading all models, refine help messages (#1061)

Browse Source 
* chore(cli): update download help messages

Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com>

* add `--all` flag to model download CLI

Signed-off-by: Panos Vagenas <pva@zurich.ibm.com>

---------

Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com>
Signed-off-by: Panos Vagenas <pva@zurich.ibm.com>
This commit is contained in:
Panos Vagenas 2025-02-26 13:27:29 +01:00 committed by GitHub
parent e197225739
commit ab683e4fb6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
docling/cli/models.py
View File

@ -53,18 +53,27 @@ def download(
..., ...,
"-o", "-o",
"--output-dir", "--output-dir",
help="The directory where all the models are downloaded.", help="The directory where to download the models.",
), ),
] = (settings.cache_dir / "models"), ] = (settings.cache_dir / "models"),
force: Annotated[ force: Annotated[
bool, typer.Option(..., help="If true, the download will be forced") bool, typer.Option(..., help="If true, the download will be forced.")
] = False, ] = False,
models: Annotated[ models: Annotated[
Optional[list[_AvailableModels]], Optional[list[_AvailableModels]],
typer.Argument( typer.Argument(
help=f"Models to download (default behavior: all will be downloaded)", help=f"Models to download (default behavior: a predefined set of models will be downloaded).",
), ),
] = None, ] = None,
all: Annotated[
bool,
typer.Option(
...,
"--all",
help="If true, all available models will be downloaded (mutually exclusive with passing specific models).",
show_default=True,
),
] = False,
quiet: Annotated[ quiet: Annotated[
bool, bool,
typer.Option( typer.Option(
@ -75,6 +84,10 @@ def download(
), ),
] = False, ] = False,
): ):
if models and all:
raise typer.BadParameter(
"Cannot simultaneously set 'all' parameter and specify models to download."
)
if not quiet: if not quiet:
FORMAT = "%(message)s" FORMAT = "%(message)s"
logging.basicConfig( logging.basicConfig(
@ -83,7 +96,7 @@ def download(
datefmt="[%X]", datefmt="[%X]",
handlers=[RichHandler(show_level=False, show_time=False, markup=True)], handlers=[RichHandler(show_level=False, show_time=False, markup=True)],
) )
to_download = models or _default_models to_download = models or ([m for m in _AvailableModels] if all else _default_models)
output_dir = download_models( output_dir = download_models(
output_dir=output_dir, output_dir=output_dir,
force=force, force=force,