mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
e9a9217cc8
Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai> Co-authored-by: Lucas Marandat <31749711+lucasmrdt@users.noreply.github.com> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai> Co-authored-by: Peter Evers <pevers90@gmail.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@protonmail.com> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
import os
|
|
import re
|
|
|
|
from vibe.core.paths import LOG_DIR, LOG_FILE
|
|
|
|
LOG_DIR.path.mkdir(parents=True, exist_ok=True)
|
|
|
|
logger = logging.getLogger("vibe")
|
|
|
|
|
|
class StructuredLogFormatter(logging.Formatter):
|
|
def format(self, record: logging.LogRecord) -> str:
|
|
timestamp = datetime.fromtimestamp(record.created, tz=UTC).isoformat()
|
|
ppid = os.getppid()
|
|
pid = os.getpid()
|
|
level = record.levelname
|
|
message = encode_log_message(record.getMessage())
|
|
|
|
line = f"{timestamp} {ppid} {pid} {level} {message}"
|
|
|
|
if record.exc_info:
|
|
exc_text = encode_log_message(self.formatException(record.exc_info))
|
|
line = f"{line} {exc_text}"
|
|
|
|
return line
|
|
|
|
|
|
def encode_log_message(message: str) -> str:
|
|
return message.replace("\\", "\\\\").replace("\n", "\\n")
|
|
|
|
|
|
def decode_log_message(encoded: str) -> str:
|
|
return re.sub(
|
|
r"\\(.)", lambda m: "\n" if m.group(1) == "n" else m.group(1), encoded
|
|
)
|
|
|
|
|
|
def apply_logging_config(target_logger: logging.Logger) -> None:
|
|
LOG_DIR.path.mkdir(parents=True, exist_ok=True)
|
|
|
|
max_bytes = int(os.environ.get("LOG_MAX_BYTES", 10 * 1024 * 1024))
|
|
|
|
if os.environ.get("DEBUG_MODE") == "true":
|
|
log_level_str = "DEBUG"
|
|
else:
|
|
log_level_str = os.environ.get("LOG_LEVEL", "WARNING").upper()
|
|
valid_levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
|
|
if log_level_str not in valid_levels:
|
|
log_level_str = "WARNING"
|
|
|
|
handler = RotatingFileHandler(
|
|
LOG_FILE.path, maxBytes=max_bytes, backupCount=0, encoding="utf-8"
|
|
)
|
|
handler.setFormatter(StructuredLogFormatter())
|
|
log_level = getattr(logging, log_level_str, logging.WARNING)
|
|
handler.setLevel(log_level)
|
|
|
|
# Make sure the logger is not gating logs
|
|
target_logger.setLevel(logging.DEBUG)
|
|
|
|
target_logger.addHandler(handler)
|
|
|
|
|
|
apply_logging_config(logger)
|