refactor: split websocket flows and add brain action decision pipeline

core/websocket_logger_setup.py

@@ -0,0 +1,86 @@
+import logging
+import os
+from datetime import datetime
+
+
+class _AnsiColorFormatter(logging.Formatter):
+    RESET = "\033[0m"
+    MESSAGE_TEXT_REPLACEMENTS = (
+        ("[PROMPT->AI 前置提示词]", "[AI提示词]"),
+        ("[PROMPT->AI", "[AI提示词"),
+        ("[THINK/TOOL_CALL]", "[AI思考-工具调用]"),
+        ("[THINK/TOOL_RETURN]", "[AI思考-工具返回]"),
+        ("[THINK/RAW_OUTPUT]", "[AI思考-原始输出]"),
+        ("[REPLY->CUSTOMER]", "[AI回复客户]"),
+        ("[ACTIVITY]", "[活动日志]"),
+        ("[AI质检]", "[AI质检]"),
+    )
+    MESSAGE_COLOR_RULES = (
+        ("[PROMPT->AI", "\033[94m"),
+        ("[THINK/", "\033[96m"),
+        ("[REPLY->CUSTOMER]", "\033[92m"),
+        ("Agent 回复", "\033[92m"),
+        ("[ACTIVITY]", "\033[95m"),
+        ("[AI质检]", "\033[97m"),
+        ("收到新消息", "\033[36m"),
+        ("发送成功", "\033[32m"),
+        ("防抖等待", "\033[93m"),
+    )
+    COLORS = {
+        logging.DEBUG: "\033[36m",
+        logging.INFO: "\033[32m",
+        logging.WARNING: "\033[33m",
+        logging.ERROR: "\033[31m",
+        logging.CRITICAL: "\033[35m",
+    }
+
+    def __init__(self, fmt: str, datefmt: str | None = None, use_color: bool = True):
+        super().__init__(fmt=fmt, datefmt=datefmt)
+        self.use_color = use_color
+
+    def format(self, record: logging.LogRecord) -> str:
+        msg = super().format(record)
+        if not self.use_color:
+            for old, new in self.MESSAGE_TEXT_REPLACEMENTS:
+                msg = msg.replace(old, new)
+            return msg
+        raw_msg = record.getMessage()
+        for old, new in self.MESSAGE_TEXT_REPLACEMENTS:
+            msg = msg.replace(old, new)
+        for key, color in self.MESSAGE_COLOR_RULES:
+            if key in raw_msg:
+                return f"{color}{msg}{self.RESET}"
+        color = self.COLORS.get(record.levelno, "")
+        if not color:
+            return msg
+        return f"{color}{msg}{self.RESET}"
+
+
+def setup_logger():
+    from logging.handlers import RotatingFileHandler
+    from config.config import LOG_DIR, LOG_MAX_BYTES, LOG_BACKUP_COUNT
+
+    logger = logging.getLogger("cs_agent")
+    if getattr(logger, "_cs_logger_configured", False):
+        return logger
+    logger.setLevel(logging.INFO)
+    logger.propagate = False
+    fmt = logging.Formatter("[%(asctime)s] %(message)s", datefmt="%H:%M:%S")
+    use_color = (os.getenv("LOG_COLOR", "1").lower() in ("1", "true", "yes")) and not bool(os.getenv("NO_COLOR"))
+
+    ch = logging.StreamHandler()
+    ch.setFormatter(_AnsiColorFormatter("[%(asctime)s] %(message)s", datefmt="%H:%M:%S", use_color=use_color))
+    logger.addHandler(ch)
+
+    LOG_DIR.mkdir(exist_ok=True)
+    today = datetime.now().strftime("%Y-%m-%d")
+    fh = RotatingFileHandler(
+        LOG_DIR / f"chat_{today}.log",
+        maxBytes=LOG_MAX_BYTES,
+        backupCount=LOG_BACKUP_COUNT,
+        encoding="utf-8",
+    )
+    fh.setFormatter(fmt)
+    logger.addHandler(fh)
+    logger._cs_logger_configured = True
+    return logger