Skip to content

datarobot_genai.core.utils.logging

logging

setup_logging

setup_logging() -> None

Setup uniform logging for the application.

Source code in datarobot_genai/core/utils/logging.py
def setup_logging() -> None:
    """Setup uniform logging for the application."""  # noqa: D401
    current_log_level = logging.getLogger().getEffectiveLevel()
    logger.info(f"Setting up logging, log level: {logging._levelToName[current_log_level]}")

    for name in ("LiteLLM", "LiteLLM Router", "LiteLLM Proxy"):
        lg = logging.getLogger(name)
        if lg:
            logger.debug(f"Resetting logger {name}")
            lg.handlers.clear()
            lg.propagate = True