From f3d1bae425e9cd20bcd5b0c86e23c59516fa3d98 Mon Sep 17 00:00:00 2001 From: Jack Jiang Date: Mon, 1 Jul 2024 17:39:52 +0800 Subject: [PATCH] feat(ark): support all logging levels --- volcenginesdkarkruntime/_utils/_logs.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/volcenginesdkarkruntime/_utils/_logs.py b/volcenginesdkarkruntime/_utils/_logs.py index bd0cf1b9..eada4de4 100644 --- a/volcenginesdkarkruntime/_utils/_logs.py +++ b/volcenginesdkarkruntime/_utils/_logs.py @@ -13,12 +13,8 @@ def _basic_config() -> None: def setup_logging() -> None: - env = os.environ.get("ARK_LOG") - if env == "debug": + level = logging.getLevelNamesMapping().get(os.environ.get("ARK_LOG", "").upper()) + if level is not None: _basic_config() - logger.setLevel(logging.DEBUG) - httpx_logger.setLevel(logging.DEBUG) - elif env == "info": - _basic_config() - logger.setLevel(logging.INFO) - httpx_logger.setLevel(logging.INFO) + logger.setLevel(level) + httpx_logger.setLevel(level)