fix(logging): attach RichHandler before level validation and reduce fragile hardcoded names

Address review feedback: move RichHandler attachment before log level validation so warnings use consistent formatting, use module.__name__ instead of hardcoded logger name string in tests, and fix stale logger reference in A2A docs.

Signed-off-by: Major Hayden <major@redhat.com>

Author: major

docs/a2a_protocol.md

@@ -771,7 +771,7 @@ service:
   color_log: true
 ```
 
-Check logs for entries from `app.endpoints.health` logger.
+Check logs for entries from `app.endpoints.a2a` logger.
 
 ## Protocol Version
 

src/log.py

@@ -28,6 +28,10 @@ def get_logger(name: str) -> logging.Logger:
     if any(isinstance(h, RichHandler) for h in logger.handlers):
         return logger
 
+    # Attach RichHandler before any log calls so warnings use consistent formatting
+    logger.handlers = [RichHandler()]
+    logger.propagate = False
+
     # Read log level from environment variable with default fallback
     level_str = os.environ.get(LIGHTSPEED_STACK_LOG_LEVEL_ENV_VAR, DEFAULT_LOG_LEVEL)
 
@@ -42,6 +46,4 @@ def get_logger(name: str) -> logging.Logger:
         validated_level = getattr(logging, DEFAULT_LOG_LEVEL)
 
     logger.setLevel(validated_level)
-    logger.handlers = [RichHandler()]
-    logger.propagate = False
     return logger

tests/unit/authorization/test_azure_token_manager.py

@@ -12,6 +12,7 @@
 from pydantic import SecretStr
 from pytest_mock import MockerFixture
 
+from authorization import azure_token_manager
 from authorization.azure_token_manager import (
     AzureEntraIDManager,
     TOKEN_EXPIRATION_LEEWAY,
@@ -136,7 +137,7 @@ def test_refresh_token_failure_logs_error(
             return_value=mock_credential_instance,
         )
 
-        azure_logger = logging.getLogger("authorization.azure_token_manager")
+        azure_logger = logging.getLogger(azure_token_manager.__name__)
         azure_logger.propagate = True
         try:
             with caplog.at_level("WARNING"):