fix：调整目录结构，优化hfc prompt，移除日程，移除动态和llm判断willing模式，

2025-05-13 18:37:55 +08:00
parent 6376da0682
commit fed71bccad
131 changed files with 422 additions and 1500 deletions
--- a/src/chat/utils/logger_config.py
+++ b/src/chat/utils/logger_config.py
@@ -0,0 +1,88 @@
+import sys
+import loguru
+from enum import Enum
+
+
+class LogClassification(Enum):
+    BASE = "base"
+    MEMORY = "memory"
+    EMOJI = "emoji"
+    CHAT = "chat"
+    PBUILDER = "promptbuilder"
+
+
+class LogModule:
+    logger = loguru.logger.opt()
+
+    def __init__(self):
+        pass
+
+    def setup_logger(self, log_type: LogClassification):
+        """配置日志格式
+
+        Args:
+            log_type: 日志类型，可选值：BASE(基础日志)、MEMORY(记忆系统日志)、EMOJI(表情包系统日志)
+        """
+        # 移除默认日志处理器
+        self.logger.remove()
+
+        # 基础日志格式
+        base_format = (
+            "<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | "
+            " d<cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
+        )
+
+        chat_format = (
+            "<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | "
+            "<cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
+        )
+
+        # 记忆系统日志格式
+        memory_format = (
+            "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | "
+            "<light-magenta>海马体</light-magenta> | <level>{message}</level>"
+        )
+
+        # 表情包系统日志格式
+        emoji_format = (
+            "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <yellow>表情包</yellow> | "
+            "<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
+        )
+
+        promptbuilder_format = (
+            "<green>{time:HH:mm}</green> | <level>{level: <8}</level> | <yellow>Prompt</yellow> | "
+            "<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>"
+        )
+
+        # 根据日志类型选择日志格式和输出
+        if log_type == LogClassification.CHAT:
+            self.logger.add(
+                sys.stderr,
+                format=chat_format,
+                # level="INFO"
+            )
+        elif log_type == LogClassification.PBUILDER:
+            self.logger.add(
+                sys.stderr,
+                format=promptbuilder_format,
+                # level="INFO"
+            )
+        elif log_type == LogClassification.MEMORY:
+            # 同时输出到控制台和文件
+            self.logger.add(
+                sys.stderr,
+                format=memory_format,
+                # level="INFO"
+            )
+            self.logger.add("logs/memory.log", format=memory_format, level="INFO", rotation="1 day", retention="7 days")
+        elif log_type == LogClassification.EMOJI:
+            self.logger.add(
+                sys.stderr,
+                format=emoji_format,
+                # level="INFO"
+            )
+            self.logger.add("logs/emoji.log", format=emoji_format, level="INFO", rotation="1 day", retention="7 days")
+        else:  # BASE
+            self.logger.add(sys.stderr, format=base_format, level="INFO")
+
+        return self.logger