core-python/maris_core/runtime.py

"""Runtime palīgfunkcijas izvietošanai un lokālai palaišanai."""

from __future__ import annotations

import os

from maris_core.utils.env import get_hf_token


def configure_huggingface_environment() -> None:
    os.environ.setdefault("HF_HUB_DISABLE_XET", "1")

    token = get_hf_token()
    if token is None:
        return

    for variable_name in (
        "HF_TOKEN",
        "HUGGING_FACE_HUB_TOKEN",
        "HUGGINGFACEHUB_API_TOKEN",
    ):
        os.environ.setdefault(variable_name, token)


def resolve_port(default: int = 8000) -> int:
    value = os.getenv("PORT")
    if not value:
        return default

    try:
        return int(value)
    except ValueError as exc:
        raise ValueError(f"Invalid PORT value: {value!r}") from exc


def resolve_host(default: str = "0.0.0.0") -> str:
    value = os.getenv("HOST")
    if value:
        return value.strip()

    return default


def is_reload_enabled(default: bool = False) -> bool:
    value = os.getenv("MARIS_RELOAD")
    if value is None:
        return default

    return value.strip().lower() in {"1", "true", "yes", "on"}