__init__.py

import os

from eval_protocol.dataset_logger.dataset_logger import DatasetLogger
from eval_protocol.dataset_logger.sqlite_dataset_logger_adapter import SqliteDatasetLoggerAdapter


# Allow disabling sqlite logger to avoid environment-specific constraints in simple CLI runs.
def _get_default_logger():
    if os.getenv("DISABLE_EP_SQLITE_LOG", "0").strip() != "1":
        return SqliteDatasetLoggerAdapter()
    else:

        class _NoOpLogger(DatasetLogger):
            def log(self, row):
                return None

            def read(self, rollout_id=None):
                return []

        return _NoOpLogger()


# Lazy property that creates the logger only when accessed
class _LazyLogger(DatasetLogger):
    def __init__(self):
        self._logger: DatasetLogger | None = None

    def _get_logger(self):
        if self._logger is None:
            self._logger = _get_default_logger()
        return self._logger

    def log(self, row):
        return self._get_logger().log(row)

    def read(self, rollout_id=None):
        return self._get_logger().read(rollout_id)


default_logger: DatasetLogger = _LazyLogger()