96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
|
"""日志工具"""
|
|||
|
|
"""
|
|||
|
|
utils/logger.py
|
|||
|
|
统一日志模块，支持彩色终端输出与文件记录
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
import sys
|
|||
|
|
from datetime import datetime
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ── ANSI 颜色常量 ──────────────────────────────────────────────
|
|||
|
|
class Color:
|
|||
|
|
RESET = "\033[0m"
|
|||
|
|
BOLD = "\033[1m"
|
|||
|
|
CYAN = "\033[96m"
|
|||
|
|
GREEN = "\033[92m"
|
|||
|
|
YELLOW = "\033[93m"
|
|||
|
|
RED = "\033[91m"
|
|||
|
|
MAGENTA = "\033[95m"
|
|||
|
|
BLUE = "\033[94m"
|
|||
|
|
GREY = "\033[90m"
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ── 自定义彩色 Formatter ───────────────────────────────────────
|
|||
|
|
class ColorFormatter(logging.Formatter):
|
|||
|
|
LEVEL_COLORS = {
|
|||
|
|
logging.DEBUG: Color.GREY,
|
|||
|
|
logging.INFO: Color.CYAN,
|
|||
|
|
logging.WARNING: Color.YELLOW,
|
|||
|
|
logging.ERROR: Color.RED,
|
|||
|
|
logging.CRITICAL: Color.MAGENTA,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
COMPONENT_COLORS = {
|
|||
|
|
"CLIENT": Color.BLUE,
|
|||
|
|
"LLM": Color.GREEN,
|
|||
|
|
"MCP": Color.YELLOW,
|
|||
|
|
"TOOL": Color.MAGENTA,
|
|||
|
|
"MEMORY": Color.CYAN,
|
|||
|
|
"SYSTEM": Color.GREY,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def format(self, record: logging.LogRecord) -> str:
|
|||
|
|
level_color = self.LEVEL_COLORS.get(record.levelno, Color.RESET)
|
|||
|
|
time_str = datetime.now().strftime("%H:%M:%S.%f")[:-3]
|
|||
|
|
|
|||
|
|
# 从 logger 名称提取组件标签，例如 "agent.CLIENT"
|
|||
|
|
component = record.name.split(".")[-1].upper()
|
|||
|
|
comp_color = self.COMPONENT_COLORS.get(component, Color.RESET)
|
|||
|
|
|
|||
|
|
prefix = (
|
|||
|
|
f"{Color.GREY}[{time_str}]{Color.RESET} "
|
|||
|
|
f"{comp_color}{Color.BOLD}[{component:6s}]{Color.RESET} "
|
|||
|
|
f"{level_color}{record.getMessage()}{Color.RESET}"
|
|||
|
|
)
|
|||
|
|
return prefix
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ── Logger 工厂函数 ────────────────────────────────────────────
|
|||
|
|
def get_logger(component: str, level: int = logging.DEBUG) -> logging.Logger:
|
|||
|
|
"""
|
|||
|
|
获取指定组件的 Logger 实例。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
component: 组件名称，如 "CLIENT"、"LLM"、"MCP"
|
|||
|
|
level: 日志级别，默认 DEBUG
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
配置好的 Logger 实例
|
|||
|
|
"""
|
|||
|
|
logger = logging.getLogger(f"agent.{component}")
|
|||
|
|
logger.setLevel(level)
|
|||
|
|
|
|||
|
|
# 避免重复添加 Handler
|
|||
|
|
if logger.handlers:
|
|||
|
|
return logger
|
|||
|
|
|
|||
|
|
# 终端 Handler（彩色）
|
|||
|
|
console_handler = logging.StreamHandler(sys.stdout)
|
|||
|
|
console_handler.setFormatter(ColorFormatter())
|
|||
|
|
logger.addHandler(console_handler)
|
|||
|
|
|
|||
|
|
# 文件 Handler（纯文本）
|
|||
|
|
log_dir = Path("logs")
|
|||
|
|
log_dir.mkdir(exist_ok=True)
|
|||
|
|
file_handler = logging.FileHandler(log_dir / "agent.log", encoding="utf-8")
|
|||
|
|
file_handler.setFormatter(
|
|||
|
|
logging.Formatter("[%(asctime)s] [%(name)s] %(levelname)s: %(message)s")
|
|||
|
|
)
|
|||
|
|
logger.addHandler(file_handler)
|
|||
|
|
|
|||
|
|
# 防止日志向上传播到 root logger
|
|||
|
|
logger.propagate = False
|
|||
|
|
return logger
|