diff --git a/llama_stack/log.py b/llama_stack/log.py index cc4c9d4cf..2a11516fa 100644 --- a/llama_stack/log.py +++ b/llama_stack/log.py @@ -247,7 +247,16 @@ def get_logger( _category_levels.update(parse_yaml_config(config)) logger = logging.getLogger(name) - logger.setLevel(_category_levels.get(category, DEFAULT_LOG_LEVEL)) + if category in _category_levels: + log_level = _category_levels[category] + else: + root_category = category.split("::")[0] + if root_category in _category_levels: + log_level = _category_levels[root_category] + else: + log_level = _category_levels.get("root", DEFAULT_LOG_LEVEL) + logging.warning(f"Unknown logging category: {category}. Falling back to default 'root' level: {log_level}") + logger.setLevel(log_level) return logging.LoggerAdapter(logger, {"category": category})