automatically configure logging/log level when the module loads (#1115)

Signed-off-by: Peter Jausovec <[email protected]>

Author: peterj

python/packages/kagent-adk/src/kagent/adk/cli.py

@@ -9,7 +9,7 @@
 from a2a.types import AgentCard
 from google.adk.cli.utils.agent_loader import AgentLoader
 
-from kagent.core import KAgentConfig, configure_tracing
+from kagent.core import KAgentConfig, configure_logging, configure_tracing
 
 from . import AgentConfig, KAgentApp
 from .skill_fetcher import fetch_skill
@@ -134,16 +134,6 @@ def test(
     asyncio.run(test_agent(agent_config, agent_card, task))
 
 
-# --- Configure Logging ---
-def configure_logging() -> None:
-    """Configure logging based on LOG_LEVEL environment variable."""
-    log_level = os.getenv("LOG_LEVEL", "INFO").upper()
-    logging.basicConfig(
-        level=log_level,
-    )
-    logging.info(f"Logging configured with level: {log_level}")
-
-
 def run_cli():
     configure_logging()
     logger.info("Starting KAgent")

python/packages/kagent-core/src/kagent/core/__init__.py

@@ -1,4 +1,7 @@
 from ._config import KAgentConfig
+from ._logging import configure_logging
 from .tracing import configure as configure_tracing
 
-__all__ = ["KAgentConfig", "configure_tracing"]
+configure_logging()
+
+__all__ = ["KAgentConfig", "configure_tracing", "configure_logging"]

python/packages/kagent-core/src/kagent/core/_logging.py

@@ -0,0 +1,28 @@
+import logging
+import os
+
+_logging_configured = False
+
+
+def configure_logging() -> None:
+    """Configure logging based on LOG_LEVEL environment variable."""
+    global _logging_configured
+
+    log_level = os.getenv("LOG_LEVEL", "INFO").upper()
+
+    # Only configure if not already configured (avoid duplicate handlers)
+    if not logging.root.handlers:
+        logging.basicConfig(
+            level=log_level,
+            format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+        )
+        _logging_configured = True
+        logging.info(f"Logging configured with level: {log_level}")
+    elif not _logging_configured:
+        # Update level if already configured but we haven't logged yet
+        logging.root.setLevel(log_level)
+        _logging_configured = True
+        logging.info(f"Logging level updated to: {log_level}")
+    else:
+        # Already configured and logged, just update the level silently
+        logging.root.setLevel(log_level)