mirror of
https://github.com/hwchase17/langchain
synced 2024-11-20 03:25:56 +00:00
21 lines
556 B
Python
21 lines
556 B
Python
from os import PathLike
|
|
from pathlib import Path
|
|
from typing import Optional, Union
|
|
|
|
|
|
class VwLogger:
|
|
"""Vowpal Wabbit custom logger."""
|
|
|
|
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)
|
|
|
|
def log(self, vw_ex: str) -> None:
|
|
if self.path:
|
|
with open(self.path, "a") as f:
|
|
f.write(f"{vw_ex}\n\n")
|
|
|
|
def logging_enabled(self) -> bool:
|
|
return bool(self.path)
|