From b9503b11e6ae7e66dd9579faec6c8b14178e2b1c Mon Sep 17 00:00:00 2001
From: Derek Higgins <derekh@redhat.com>
Date: Tue, 2 Dec 2025 11:03:19 +0000
Subject: [PATCH] fix: call setup_logging early to apply category-specific log
 levels

Category-specific log levels from LLAMA_STACK_LOGGING were not applied to
loggers created before setup_logging() was called. This fix moves the
setup_logging() call earlier in the initialization sequence to ensure all
loggers respect their configured levels regardless of initialization timing.

Closes: #4252

Signed-off-by: Derek Higgins <derekh@redhat.com>
---
 src/llama_stack/cli/llama.py          | 6 +++---
 src/llama_stack/core/server/server.py | 5 +----
 tests/integration/fixtures/common.py  | 5 +++++
 3 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/llama_stack/cli/llama.py b/src/llama_stack/cli/llama.py
index aa8893bc0..b0b753ae3 100644
--- a/src/llama_stack/cli/llama.py
+++ b/src/llama_stack/cli/llama.py
@@ -8,6 +8,9 @@ import argparse
 
 from llama_stack.log import setup_logging
 
+# Initialize logging early before any loggers get created
+setup_logging()
+
 from .stack import StackParser
 from .stack.utils import print_subcommand_description
 
@@ -44,9 +47,6 @@ class LlamaCLIParser:
 
 
 def main():
-    # Initialize logging from environment variables before any other operations
-    setup_logging()
-
     parser = LlamaCLIParser()
     args = parser.parse_args()
     parser.run(args)
diff --git a/src/llama_stack/core/server/server.py b/src/llama_stack/core/server/server.py
index 0d3513980..6a8c7b006 100644
--- a/src/llama_stack/core/server/server.py
+++ b/src/llama_stack/core/server/server.py
@@ -55,7 +55,7 @@ from llama_stack.core.telemetry.tracing import CURRENT_TRACE_CONTEXT, setup_logg
 from llama_stack.core.utils.config import redact_sensitive_fields
 from llama_stack.core.utils.config_resolution import Mode, resolve_config_or_distro
 from llama_stack.core.utils.context import preserve_contexts_async_generator
-from llama_stack.log import LoggingConfig, get_logger, setup_logging
+from llama_stack.log import LoggingConfig, get_logger
 from llama_stack_api import Api, ConflictError, PaginatedResponse, ResourceNotFoundError
 
 from .auth import AuthenticationMiddleware
@@ -367,9 +367,6 @@ def create_app() -> StackApp:
     Returns:
         Configured StackApp instance.
     """
-    # Initialize logging from environment variables first
-    setup_logging()
-
     config_file = os.getenv("LLAMA_STACK_CONFIG")
     if config_file is None:
         raise ValueError("LLAMA_STACK_CONFIG environment variable is required")
diff --git a/tests/integration/fixtures/common.py b/tests/integration/fixtures/common.py
index 817180cfe..f90486cc2 100644
--- a/tests/integration/fixtures/common.py
+++ b/tests/integration/fixtures/common.py
@@ -14,6 +14,11 @@ import tempfile
 import time
 from urllib.parse import urlparse
 
+# Initialize logging early before any loggers get created
+from llama_stack.log import setup_logging
+
+setup_logging()
+
 import pytest
 import requests
 import yaml