diff --git a/document-parser/infra/settings.py b/document-parser/infra/settings.py index 1d2a952..4f15384 100644 --- a/document-parser/infra/settings.py +++ b/document-parser/infra/settings.py @@ -10,6 +10,7 @@ from dataclasses import dataclass, field class Settings: app_version: str = "dev" conversion_engine: str = "local" # "local" or "remote" + deployment_mode: str = "self-hosted" # "self-hosted" or "huggingface" docling_serve_url: str = "http://localhost:5001" docling_serve_api_key: str | None = None conversion_timeout: int = 600 @@ -27,6 +28,7 @@ class Settings: return cls( app_version=os.environ.get("APP_VERSION", "dev"), conversion_engine=os.environ.get("CONVERSION_ENGINE", "local"), + deployment_mode=os.environ.get("DEPLOYMENT_MODE", "self-hosted"), docling_serve_url=os.environ.get("DOCLING_SERVE_URL", "http://localhost:5001"), docling_serve_api_key=os.environ.get("DOCLING_SERVE_API_KEY"), conversion_timeout=int(os.environ.get("CONVERSION_TIMEOUT", "600")), diff --git a/document-parser/main.py b/document-parser/main.py index 654c88e..957cada 100644 --- a/document-parser/main.py +++ b/document-parser/main.py @@ -117,5 +117,6 @@ async def health() -> dict[str, str]: "status": status, "version": settings.app_version, "engine": settings.conversion_engine, + "deploymentMode": settings.deployment_mode, "database": db_status, } diff --git a/document-parser/tests/test_settings.py b/document-parser/tests/test_settings.py index 1326901..ffe6a2a 100644 --- a/document-parser/tests/test_settings.py +++ b/document-parser/tests/test_settings.py @@ -10,6 +10,7 @@ class TestSettingsDefaults: s = Settings() assert s.app_version == "dev" assert s.conversion_engine == "local" + assert s.deployment_mode == "self-hosted" assert s.docling_serve_url == "http://localhost:5001" assert s.docling_serve_api_key is None assert s.conversion_timeout == 600 @@ -30,6 +31,7 @@ class TestSettingsFromEnv: def test_reads_env_vars(self, monkeypatch): monkeypatch.setenv("APP_VERSION", "1.2.3") monkeypatch.setenv("CONVERSION_ENGINE", "remote") + monkeypatch.setenv("DEPLOYMENT_MODE", "huggingface") monkeypatch.setenv("DOCLING_SERVE_URL", "http://serve:9000") monkeypatch.setenv("DOCLING_SERVE_API_KEY", "secret-key") monkeypatch.setenv("CONVERSION_TIMEOUT", "120") @@ -41,6 +43,7 @@ class TestSettingsFromEnv: assert s.app_version == "1.2.3" assert s.conversion_engine == "remote" + assert s.deployment_mode == "huggingface" assert s.docling_serve_url == "http://serve:9000" assert s.docling_serve_api_key == "secret-key" assert s.conversion_timeout == 120 @@ -53,6 +56,7 @@ class TestSettingsFromEnv: for key in ( "APP_VERSION", "CONVERSION_ENGINE", + "DEPLOYMENT_MODE", "DOCLING_SERVE_URL", "DOCLING_SERVE_API_KEY", "CONVERSION_TIMEOUT", diff --git a/frontend/src/app/App.vue b/frontend/src/app/App.vue index f8b29d2..a75d8be 100644 --- a/frontend/src/app/App.vue +++ b/frontend/src/app/App.vue @@ -29,6 +29,13 @@ + +
@@ -39,11 +46,12 @@