File size: 1,111 Bytes
f440f03
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
"""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"}