feat(trainer): add structured and configurable logging support

kubeflow/__init__.py

@@ -12,4 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import logging
+
+# Configure NullHandler for the kubeflow package to avoid logging noise
+# when users haven't configured logging. Users can override this by setting
+# their own logging configuration.
+logger = logging.getLogger(__name__)
+logger.addHandler(logging.NullHandler())
+
 __version__ = "0.1.0"

kubeflow/trainer/__init__.py

@@ -23,6 +23,9 @@
 # Import the Kubeflow Trainer constants.
 from kubeflow.trainer.constants.constants import DATASET_PATH, MODEL_PATH  # noqa: F401
 
+# Import the Kubeflow Trainer logging utilities.
+from kubeflow.trainer.logging import get_logger, setup_logging  # noqa: F401
+
 # Import the Kubeflow Trainer types.
 from kubeflow.trainer.types.types import (
     BuiltinTrainer,
@@ -59,4 +62,6 @@
     "TrainerType",
     "LocalProcessBackendConfig",
     "KubernetesBackendConfig",
+    "get_logger",
+    "setup_logging",
 ]

kubeflow/trainer/api/trainer_client.py

@@ -44,15 +44,21 @@ def __init__(
             ValueError: Invalid backend configuration.
 
         """
+        logger.debug("Initializing TrainerClient with backend_config=%s", backend_config)
+
         # initialize training backend
         if not backend_config:
             backend_config = KubernetesBackendConfig()
+            logger.debug("Using default KubernetesBackendConfig")
 
         if isinstance(backend_config, KubernetesBackendConfig):
             self.backend = KubernetesBackend(backend_config)
+            logger.debug("Initialized Kubernetes backend")
         elif isinstance(backend_config, LocalProcessBackendConfig):
             self.backend = LocalProcessBackend(backend_config)
+            logger.debug("Initialized LocalProcess backend")
         else:
+            logger.error("Invalid backend config type: %s", type(backend_config))
             raise ValueError(f"Invalid backend config '{backend_config}'")
 
     def list_runtimes(self) -> list[types.Runtime]:
@@ -119,7 +125,17 @@ def train(
             TimeoutError: Timeout to create TrainJobs.
             RuntimeError: Failed to create TrainJobs.
         """
-        return self.backend.train(runtime=runtime, initializer=initializer, trainer=trainer)
+        logger.debug(
+            "Creating TrainJob with runtime=%s, initializer=%s, trainer=%s",
+            runtime,
+            initializer,
+            trainer,
+        )
+
+        job_id = self.backend.train(runtime=runtime, initializer=initializer, trainer=trainer)
+        logger.debug("Successfully created TrainJob with ID: %s", job_id)
+
+        return job_id
 
     def list_jobs(self, runtime: Optional[types.Runtime] = None) -> list[types.TrainJob]:
         """List of the created TrainJobs. If a runtime is specified, only TrainJobs associated with

kubeflow/trainer/logging/__init__.py

@@ -0,0 +1,24 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Kubeflow SDK logging module.
+
+This module provides structured and configurable logging support for the Kubeflow SDK.
+It includes centralized logger configuration, structured log messages, and context-aware logging.
+"""
+
+from .config import get_logger, setup_logging
+from .formatters import StructuredFormatter
+
+__all__ = ["get_logger", "setup_logging", "StructuredFormatter"]

kubeflow/trainer/logging/config.py

@@ -0,0 +1,123 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Logging configuration for Kubeflow SDK."""
+
+import logging
+import logging.config
+import os
+from typing import Optional, Union
+
+
+def setup_logging(
+    level: Union[str, int] = "INFO",
+    format_type: str = "console",
+    log_file: Optional[str] = None,
+) -> None:
+    """Setup logging configuration for Kubeflow SDK.
+
+    Args:
+        level: Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
+        format_type: Output format type ('console', 'json', 'detailed')
+        log_file: Optional log file path for file output
+    """
+    # Convert string level to logging constant
+    if isinstance(level, str):
+        level = getattr(logging, level.upper(), logging.INFO)
+
+    # Base configuration
+    config = {
+        "version": 1,
+        "disable_existing_loggers": False,
+        "formatters": {
+            "console": {
+                "format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+                "datefmt": "%Y-%m-%d %H:%M:%S",
+            },
+            "detailed": {
+                "format": (
+                    "%(asctime)s - %(name)s - %(levelname)s - %(filename)s:%(lineno)d - %(message)s"
+                ),
+                "datefmt": "%Y-%m-%d %H:%M:%S",
+            },
+            "json": {
+                "()": "kubeflow.trainer.logging.formatters.StructuredFormatter",
+            },
+        },
+        "handlers": {
+            "console": {
+                "class": "logging.StreamHandler",
+                "level": level,
+                "formatter": format_type,
+                "stream": "ext://sys.stdout",
+            },
+        },
+        "loggers": {
+            "kubeflow": {
+                "level": level,
+                "handlers": ["console"],
+                "propagate": False,
+            },
+        },
+        "root": {
+            "level": level,
+            "handlers": ["console"],
+        },
+    }
+
+    # Add file handler if log_file is specified
+    if log_file:
+        config["handlers"]["file"] = {
+            "class": "logging.FileHandler",
+            "level": level,
+            "formatter": format_type,
+            "filename": log_file,
+            "mode": "a",
+        }
+        config["loggers"]["kubeflow"]["handlers"].append("file")
+        config["root"]["handlers"].append("file")
+
+    # Apply configuration
+    logging.config.dictConfig(config)
+
+
+def get_logger(name: str) -> logging.Logger:
+    """Get a logger instance for the given name.
+
+    Args:
+        name: Logger name, typically __name__ of the calling module
+
+    Returns:
+        Logger instance configured for Kubeflow SDK
+    """
+    # Ensure the logger name starts with 'kubeflow'
+    if not name.startswith("kubeflow"):
+        name = f"kubeflow.{name}"
+
+    return logging.getLogger(name)
+
+
+def configure_from_env() -> None:
+    """Configure logging from environment variables.
+
+    Environment variables:
+        KUBEFLOW_LOG_LEVEL: Logging level (default: INFO)
+        KUBEFLOW_LOG_FORMAT: Output format (default: console)
+        KUBEFLOW_LOG_FILE: Log file path (optional)
+    """
+    level = os.getenv("KUBEFLOW_LOG_LEVEL", "INFO")
+    format_type = os.getenv("KUBEFLOW_LOG_FORMAT", "console")
+    log_file = os.getenv("KUBEFLOW_LOG_FILE")
+
+    setup_logging(level=level, format_type=format_type, log_file=log_file)

kubeflow/trainer/logging/formatters.py

@@ -0,0 +1,52 @@
+# Copyright 2025 The Kubeflow Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Custom log formatters for Kubeflow SDK."""
+
+from datetime import datetime, timezone
+import json
+import logging
+
+
+class StructuredFormatter(logging.Formatter):
+    """JSON structured formatter for Kubeflow SDK logs.
+
+    This formatter outputs logs in JSON format, making them suitable for
+    log aggregation systems like ELK stack, Fluentd, etc.
+    """
+
+    def format(self, record: logging.LogRecord) -> str:
+        """Format log record as JSON.
+
+        Args:
+            record: Log record to format
+
+        Returns:
+            JSON formatted log string
+        """
+        log_entry = {
+            "timestamp": datetime.now(timezone.utc).isoformat(),
+            "level": record.levelname,
+            "logger": record.name,
+            "message": record.getMessage(),
+            "module": record.module,
+            "function": record.funcName,
+            "line": record.lineno,
+        }
+
+        # Add exception info if present
+        if record.exc_info:
+            log_entry["exception"] = self.formatException(record.exc_info)
+
+        return json.dumps(log_entry, ensure_ascii=False)