NexQuant/rdagent/log/conf.py

from datetime import datetime, timezone
from pathlib import Path
from typing import Any

from pydantic_settings import SettingsConfigDict

from rdagent.core.conf import ExtendedBaseSettings


class LogSettings(ExtendedBaseSettings):
    model_config = SettingsConfigDict(env_prefix="LOG_", protected_namespaces=())

    trace_path: str = str(Path.cwd() / "log" / datetime.now(timezone.utc).strftime("%Y-%m-%d_%H-%M-%S-%f"))

    format_console: str | None = None
    """"If it is None, leave it as the default"""

    ui_server_port: int | None = None

    storages: dict[str, list[int | str]] = {}

    def set_ui_server_port(self, port: int | None) -> None:
        self.ui_server_port = port
        if port is None:
            self.storages.pop("rdagent.log.ui.storage.WebStorage", None)
            return

        self.storages["rdagent.log.ui.storage.WebStorage"] = [port, self.trace_path]

    def model_post_init(self, _context: Any, /) -> None:
        self.set_ui_server_port(self.ui_server_port)


LOG_SETTINGS = LogSettings()