28fe56cfc1
MyPy compliance for the whole codebase - systematic way of executing tools (`executor.py`) - support for using pydantic models in tool inputs and outputs - mypy compliance (most of the changes) - removal of unused code (from previous iterations) Co-authored-by: Nate Barbettini <nate@arcade-ai.com>
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from typing import TYPE_CHECKING
|
|
|
|
from loguru import logger
|
|
|
|
from arcade.actor.core.conf import settings
|
|
|
|
actor_log_path = os.path.join(settings.WORK_DIR, "actor_logs")
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
import loguru
|
|
|
|
|
|
class Logger:
|
|
"""Logger for the Actor server"""
|
|
|
|
def __init__(self) -> None:
|
|
self.log_path = actor_log_path
|
|
|
|
def log(self) -> loguru.Logger:
|
|
if not os.path.exists(self.log_path):
|
|
os.makedirs(self.log_path, exist_ok=True)
|
|
|
|
log_stdout_file = os.path.join(self.log_path, settings.LOG_STDOUT_FILENAME)
|
|
log_stderr_file = os.path.join(self.log_path, settings.LOG_STDERR_FILENAME)
|
|
|
|
log_config = {
|
|
"rotation": "10 MB",
|
|
"retention": "15 days",
|
|
"compression": "tar.gz",
|
|
"enqueue": True,
|
|
}
|
|
# stdout
|
|
logger.add(
|
|
log_stdout_file,
|
|
level="INFO",
|
|
filter=lambda record: record["level"].name == "INFO" or record["level"].no <= 25, # type: ignore[call-overload]
|
|
backtrace=False,
|
|
diagnose=False,
|
|
**log_config,
|
|
)
|
|
# stderr
|
|
logger.add(
|
|
log_stderr_file,
|
|
level="ERROR",
|
|
filter=lambda record: record["level"].name == "ERROR" or record["level"].no >= 30, # type: ignore[call-overload]
|
|
backtrace=True,
|
|
diagnose=True,
|
|
**log_config,
|
|
)
|
|
|
|
return logger
|
|
|
|
|
|
log = Logger().log()
|