annotations/Azaion.Inference/constants_inf.pyx

import sys

from loguru import logger

cdef str CONFIG_FILE = "config.yaml"  # Port for the zmq

cdef str QUEUE_CONFIG_FILENAME = "secured-config.json"
cdef str AI_ONNX_MODEL_FILE = "azaion.onnx"

cdef str CDN_CONFIG = "cdn.yaml"
cdef str MODELS_FOLDER = "models"

cdef int SMALL_SIZE_KB = 3

logger.remove()
log_format = "[{time:HH:mm:ss} {level}] {message}"
logger.add(
    sink="Logs/log_inference_{time:YYYYMMDD}.txt",
    level="INFO",
    format=log_format,
    enqueue=True,
    rotation="1 day",
    retention="30 days",
)
logger.add(
    sys.stdout,
    level="DEBUG",
    format=log_format,
    filter=lambda record: record["level"].name in ("INFO", "DEBUG", "SUCCESS"),
    colorize=True
)
logger.add(
    sys.stderr,
    level="WARNING",
    format=log_format,
    colorize=True
)

cdef log(str log_message):
    logger.info(log_message)

cdef logerror(str error):
    logger.error(error)