phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-15 06:00:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 8e78478f9a into 8ed69978f9

Browse source
This commit is contained in:
Charlie Doern 2025-08-14 17:17:51 -07:00 • committed by GitHub
commit 022e2338fe
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 11 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/log.py
View file

@ -19,6 +19,7 @@ from llama_stack.core.datatypes import LoggingConfig
# Default log level # Default log level
DEFAULT_LOG_LEVEL = logging.INFO DEFAULT_LOG_LEVEL = logging.INFO
DEFAULT_TELEMETRY_LEVEL = logging.WARN
# Predefined categories # Predefined categories
CATEGORIES = [ CATEGORIES = [
@ -35,8 +36,10 @@ CATEGORIES = [
"openai_responses", "openai_responses",
] ]
# Initialize category levels with default level # Initialize category levels with default level, except for telemetry which gets WARN
_category_levels: dict[str, int] = dict.fromkeys(CATEGORIES, DEFAULT_LOG_LEVEL) _category_levels: dict[str, int] = {
category: DEFAULT_TELEMETRY_LEVEL if category == "telemetry" else DEFAULT_LOG_LEVEL for category in CATEGORIES
}
def config_to_category_levels(category: str, level: str): def config_to_category_levels(category: str, level: str):
@ -99,6 +102,7 @@ def parse_environment_config(env_config: str) -> dict[str, int]:
Returns: Returns:
Dict[str, int]: A dictionary mapping categories to their log levels. Dict[str, int]: A dictionary mapping categories to their log levels.
""" """
category_levels = {} category_levels = {}
delimiter = "," delimiter = ","
for pair in env_config.split(delimiter): for pair in env_config.split(delimiter):
@ -193,6 +197,7 @@ def setup_logging(category_levels: dict[str, int], log_file: str | None) -> None
"filename": log_file, "filename": log_file,
"mode": "a", "mode": "a",
"encoding": "utf-8", "encoding": "utf-8",
"filters": ["category_filter"],
} }
logging_config = { logging_config = {
@ -213,7 +218,9 @@ def setup_logging(category_levels: dict[str, int], log_file: str | None) -> None
"loggers": { "loggers": {
category: { category: {
"handlers": list(handlers.keys()), # Apply all handlers "handlers": list(handlers.keys()), # Apply all handlers
"level": category_levels.get(category, DEFAULT_LOG_LEVEL), "level": category_levels.get(
category, (DEFAULT_LOG_LEVEL if category != "telemetry" else DEFAULT_TELEMETRY_LEVEL)
),
"propagate": False, # Disable propagation to root logger "propagate": False, # Disable propagation to root logger
} }
for category in CATEGORIES for category in CATEGORIES
@ -237,6 +244,7 @@ def get_logger(
""" """
Returns a logger with the specified name and category. Returns a logger with the specified name and category.
If no category is provided, defaults to 'uncategorized'. If no category is provided, defaults to 'uncategorized'.
Note: telemetry category defaults to WARN as the default level
Parameters: Parameters:
name (str): The name of the logger (e.g., module or filename). name (str): The name of the logger (e.g., module or filename).

3
llama_stack/providers/inline/telemetry/meta_reference/console_span_processor.py
View file

@ -21,9 +21,6 @@ class ConsoleSpanProcessor(SpanProcessor):
self.print_attributes = print_attributes self.print_attributes = print_attributes
def on_start(self, span: ReadableSpan, parent_context=None) -> None: def on_start(self, span: ReadableSpan, parent_context=None) -> None:
if span.attributes and span.attributes.get("__autotraced__"):
return
timestamp = datetime.fromtimestamp(span.start_time / 1e9, tz=UTC).strftime("%H:%M:%S.%f")[:-3] timestamp = datetime.fromtimestamp(span.start_time / 1e9, tz=UTC).strftime("%H:%M:%S.%f")[:-3]
logger.info(f"[dim]{timestamp}[/dim] [bold magenta][START][/bold magenta] [dim]{span.name}[/dim]") logger.info(f"[dim]{timestamp}[/dim] [bold magenta][START][/bold magenta] [dim]{span.name}[/dim]")