2c254382c8
Reject invalid configuration at startup: negative timeouts, zero concurrency, bad table mode. Reports all errors at once.
72 lines
3.4 KiB
Python
72 lines
3.4 KiB
Python
"""Centralized application settings — loaded from environment variables."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from dataclasses import dataclass, field
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
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 = 900
|
|
document_timeout: float = 120.0 # Docling-level per-document timeout (seconds)
|
|
max_concurrent_analyses: int = 3
|
|
default_table_mode: str = "accurate" # "accurate" or "fast"
|
|
max_page_count: int = 0 # 0 = unlimited (upload validation)
|
|
max_file_size: int = 0 # 0 = unlimited (Docling-level, bytes)
|
|
upload_dir: str = "./uploads"
|
|
db_path: str = "./data/docling_studio.db"
|
|
cors_origins: list[str] = field(
|
|
default_factory=lambda: ["http://localhost:3000", "http://localhost:5173"]
|
|
)
|
|
|
|
def __post_init__(self) -> None:
|
|
errors: list[str] = []
|
|
if self.document_timeout <= 0:
|
|
errors.append(f"document_timeout must be > 0 (got {self.document_timeout})")
|
|
if self.conversion_timeout <= 0:
|
|
errors.append(f"conversion_timeout must be > 0 (got {self.conversion_timeout})")
|
|
if self.max_concurrent_analyses < 1:
|
|
errors.append(
|
|
f"max_concurrent_analyses must be >= 1 (got {self.max_concurrent_analyses})"
|
|
)
|
|
if self.max_page_count < 0:
|
|
errors.append(f"max_page_count must be >= 0 (got {self.max_page_count})")
|
|
if self.max_file_size < 0:
|
|
errors.append(f"max_file_size must be >= 0 (got {self.max_file_size})")
|
|
if self.default_table_mode not in ("accurate", "fast"):
|
|
errors.append(
|
|
f"default_table_mode must be 'accurate' or 'fast' (got '{self.default_table_mode}')"
|
|
)
|
|
if errors:
|
|
raise ValueError("Invalid settings:\n " + "\n ".join(errors))
|
|
|
|
@classmethod
|
|
def from_env(cls) -> Settings:
|
|
"""Build a Settings instance from environment variables."""
|
|
cors_raw = os.environ.get("CORS_ORIGINS", "http://localhost:3000,http://localhost:5173")
|
|
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", "900")),
|
|
document_timeout=float(os.environ.get("DOCUMENT_TIMEOUT", "120.0")),
|
|
max_concurrent_analyses=int(os.environ.get("MAX_CONCURRENT_ANALYSES", "3")),
|
|
default_table_mode=os.environ.get("DEFAULT_TABLE_MODE", "accurate"),
|
|
max_page_count=int(os.environ.get("MAX_PAGE_COUNT", "0")),
|
|
max_file_size=int(os.environ.get("MAX_FILE_SIZE", "0")),
|
|
upload_dir=os.environ.get("UPLOAD_DIR", "./uploads"),
|
|
db_path=os.environ.get("DB_PATH", "./data/docling_studio.db"),
|
|
cors_origins=[o.strip() for o in cors_raw.split(",")],
|
|
)
|
|
|
|
|
|
# Module-level singleton — import this from other modules.
|
|
settings = Settings.from_env()
|