LLM/docling-serve
1
0
Fork 0
mirror of https://github.com/docling-project/docling-serve.git synced 2025-12-03 02:23:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: set DOCLING_SERVE_ARTIFACTS_PATH in images (#53)

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi
2025-02-19 22:03:56 +01:00
committed by GitHub
parent ec33a61faa
commit 4877248368
5 changed files with 68 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
docling_serve/__main__.py
View File

@@ -3,7 +3,8 @@ import logging
import platform
import sys
import warnings
from typing import Annotated, Any, Union
from pathlib import Path
from typing import Annotated, Any, Optional, Union
import typer
import uvicorn
@@ -161,6 +162,15 @@ def dev(
),
] = uvicorn_settings.proxy_headers,
# docling options
artifacts_path: Annotated[
Optional[Path],
typer.Option(
help=(
"If set to a valid directory, "
"the model weights will be loaded from this path."
)
),
] = docling_serve_settings.artifacts_path,
enable_ui: Annotated[bool, typer.Option(help="Enable the development UI.")] = True,
) -> Any:
"""
@@ -179,6 +189,7 @@ def dev(
uvicorn_settings.root_path = root_path
uvicorn_settings.proxy_headers = proxy_headers
docling_serve_settings.artifacts_path = artifacts_path
docling_serve_settings.enable_ui = enable_ui
_run(
@@ -243,6 +254,15 @@ def run(
),
] = uvicorn_settings.proxy_headers,
# docling options
artifacts_path: Annotated[
Optional[Path],
typer.Option(
help=(
"If set to a valid directory, "
"the model weights will be loaded from this path."
)
),
] = docling_serve_settings.artifacts_path,
enable_ui: Annotated[
bool, typer.Option(help="Enable the development UI.")
] = docling_serve_settings.enable_ui,
@@ -264,6 +284,7 @@ def run(
uvicorn_settings.root_path = root_path
uvicorn_settings.proxy_headers = proxy_headers
docling_serve_settings.artifacts_path = artifacts_path
docling_serve_settings.enable_ui = enable_ui
_run(