use structlog instead of loguru

src/chat/utils/logger_config.py

@@ -1,88 +0,0 @@
-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

src/chat/utils/prompt_builder.py

@@ -3,14 +3,14 @@ import re
 from contextlib import asynccontextmanager
 import asyncio
 import contextvars
-from src.common.logger import get_module_logger
+from src.common.logger import get_logger
 
 # import traceback
 from rich.traceback import install
 
 install(extra_lines=3)
 
-logger = get_module_logger("prompt_build")
+logger = get_logger("prompt_build")
 
 
 class PromptContext:

src/chat/utils/statistic.py

@@ -3,14 +3,14 @@ from datetime import datetime, timedelta
 from typing import Any, Dict, Tuple, List
 
 
-from src.common.logger import get_module_logger
+from src.common.logger import get_logger
 from src.manager.async_task_manager import AsyncTask
 
 from ...common.database.database import db  # This db is the Peewee database instance
 from ...common.database.database_model import OnlineTime, LLMUsage, Messages  # Import the Peewee model
 from src.manager.local_store_manager import local_storage
 
-logger = get_module_logger("maibot_statistic")
+logger = get_logger("maibot_statistic")
 
 # 统计数据的键
 TOTAL_REQ_CNT = "total_requests"

src/chat/utils/typo_generator.py

@@ -13,9 +13,9 @@ from pathlib import Path
 import jieba
 from pypinyin import Style, pinyin
 
-from src.common.logger import get_module_logger
+from src.common.logger import get_logger
 
-logger = get_module_logger("typo_gen")
+logger = get_logger("typo_gen")
 
 
 class ChineseTypoGenerator:

src/chat/utils/utils.py

@@ -7,7 +7,7 @@ import jieba
 import numpy as np
 from maim_message import UserInfo
 
-from src.common.logger import get_module_logger
+from src.common.logger import get_logger
 from src.manager.mood_manager import mood_manager
 from ..message_receive.message import MessageRecv
 from src.llm_models.utils_model import LLMRequest
@@ -15,7 +15,7 @@ from .typo_generator import ChineseTypoGenerator
 from ...config.config import global_config
 from ...common.message_repository import find_messages, count_messages
 
-logger = get_module_logger("chat_utils")
+logger = get_logger("chat_utils")
 
 
 def is_english_letter(char: str) -> bool:

src/chat/utils/utils_image.py

@@ -13,7 +13,7 @@ from src.common.database.database_model import Images, ImageDescriptions
 from src.config.config import global_config
 from src.llm_models.utils_model import LLMRequest
 
-from src.common.logger_manager import get_logger
+from src.common.logger import get_logger
 from rich.traceback import install
 
 install(extra_lines=3)