__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 log(self, row):
        return _get_default_logger().log(row)

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


default_logger: DatasetLogger = _LazyLogger()