diff --git a/llama_stack/core/conversations/conversations.py b/llama_stack/core/conversations/conversations.py index bef138e69..612b2f68e 100644 --- a/llama_stack/core/conversations/conversations.py +++ b/llama_stack/core/conversations/conversations.py @@ -32,7 +32,7 @@ from llama_stack.providers.utils.sqlstore.sqlstore import ( sqlstore_impl, ) -logger = get_logger(name=__name__, category="openai::conversations") +logger = get_logger(name=__name__, category="openai_conversations") class ConversationServiceConfig(BaseModel): diff --git a/llama_stack/log.py b/llama_stack/log.py index 6f751b21d..6ccc200fc 100644 --- a/llama_stack/log.py +++ b/llama_stack/log.py @@ -31,12 +31,17 @@ CATEGORIES = [ "client", "telemetry", "openai_responses", + "openai_conversations", "testing", "providers", "models", "files", "vector_io", "tool_runtime", + "cli", + "post_training", + "scoring", + "tests", ] UNCATEGORIZED = "uncategorized" @@ -261,11 +266,10 @@ def get_logger( if root_category in _category_levels: log_level = _category_levels[root_category] else: + assert category == UNCATEGORIZED, ( + "Unknown logging category: {category}. To resolve, choose a valid category from the CATEGORIES list or add it to the CATEGORIES list." + ) log_level = _category_levels.get("root", DEFAULT_LOG_LEVEL) - if category != UNCATEGORIZED: - 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}) diff --git a/llama_stack/models/llama/tokenizer_utils.py b/llama_stack/models/llama/tokenizer_utils.py index 9830bb61b..05da410a1 100644 --- a/llama_stack/models/llama/tokenizer_utils.py +++ b/llama_stack/models/llama/tokenizer_utils.py @@ -9,7 +9,7 @@ from pathlib import Path from llama_stack.log import get_logger -logger = get_logger(__name__, "tokenizer_utils") +logger = get_logger(__name__, "models") def load_bpe_file(model_path: Path) -> dict[bytes, int]: