Source code for langchain_experimental.rl_chain.vw_logger

from os import PathLike
from pathlib import Path
from typing import Optional, Union


[docs]class VwLogger:
    """Vowpal Wabbit custom logger."""

[docs]    def __init__(self, path: Optional[Union[str, PathLike]]):
        self.path = Path(path) if path else None
        if self.path:
            self.path.parent.mkdir(parents=True, exist_ok=True)

[docs]    def log(self, vw_ex: str) -> None:
        if self.path:
            with open(self.path, "a") as f:
                f.write(f"{vw_ex}\n\n")

[docs]    def logging_enabled(self) -> bool:
        return bool(self.path)