fix: 为maim_message添加logger

src/common/logger.py

@@ -835,6 +835,23 @@ API_SERVER_STYLE_CONFIG = {
     },
 }
 
+# maim_message 消息服务样式配置
+MAIM_MESSAGE_STYLE_CONFIG = {
+    "advanced": {
+        "console_format": (
+            "<white>{time:YYYY-MM-DD HH:mm:ss}</white> | "
+            "<level>{level: <8}</level> | "
+            "<fg #00B2FF>消息服务</fg #00B2FF> | "
+            "<level>{message}</level>"
+        ),
+        "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 消息服务 | {message}",
+    },
+    "simple": {
+        "console_format": "<level>{time:HH:mm:ss}</level> | <fg #00B2FF>消息服务</fg #00B2FF> | {message}",
+        "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 消息服务 | {message}",
+    },
+}
+
 
 # 根据SIMPLE_OUTPUT选择配置
 MAIN_STYLE_CONFIG = MAIN_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else MAIN_STYLE_CONFIG["advanced"]
@@ -912,6 +929,9 @@ CHAT_MESSAGE_STYLE_CONFIG = (
 CHAT_IMAGE_STYLE_CONFIG = CHAT_IMAGE_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else CHAT_IMAGE_STYLE_CONFIG["advanced"]
 INIT_STYLE_CONFIG = INIT_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else INIT_STYLE_CONFIG["advanced"]
 API_SERVER_STYLE_CONFIG = API_SERVER_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else API_SERVER_STYLE_CONFIG["advanced"]
+MAIM_MESSAGE_STYLE_CONFIG = (
+    MAIM_MESSAGE_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else MAIM_MESSAGE_STYLE_CONFIG["advanced"]
+)
 INTEREST_CHAT_STYLE_CONFIG = (
     INTEREST_CHAT_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else INTEREST_CHAT_STYLE_CONFIG["advanced"]
 )

src/common/logger_manager.py

@@ -45,6 +45,7 @@ from src.common.logger import (
     INIT_STYLE_CONFIG,
     INTEREST_CHAT_STYLE_CONFIG,
     API_SERVER_STYLE_CONFIG,
+    MAIM_MESSAGE_STYLE_CONFIG,
 )
 
 # 可根据实际需要补充更多模块配置
@@ -94,6 +95,7 @@ MODULE_LOGGER_CONFIGS = {
     "init": INIT_STYLE_CONFIG,  # 初始化
     "interest_chat": INTEREST_CHAT_STYLE_CONFIG,  # 兴趣
     "api": API_SERVER_STYLE_CONFIG,  # API服务器
+    "maim_message": MAIM_MESSAGE_STYLE_CONFIG,  # 消息服务
     # ...如有更多模块，继续添加...
 }
 

src/common/message/api.py

@@ -1,6 +1,27 @@
 from src.common.server import global_server
 import os
+import importlib.metadata
 from maim_message import MessageServer
+from src.common.logger_manager import get_logger
 
 
-global_api = MessageServer(host=os.environ["HOST"], port=int(os.environ["PORT"]), app=global_server.get_app())
+# 检查maim_message版本
+try:
+    maim_message_version = importlib.metadata.version("maim_message")
+    version_compatible = [int(x) for x in maim_message_version.split(".")] >= [0, 3, 0]
+except (importlib.metadata.PackageNotFoundError, ValueError):
+    version_compatible = False
+
+# 根据版本决定是否使用自定义logger
+kwargs = {
+    "host": os.environ["HOST"],
+    "port": int(os.environ["PORT"]),
+    "app": global_server.get_app(),
+}
+
+# 只有在版本 >= 0.3.0 时才使用自定义logger
+if version_compatible:
+    maim_message_logger = get_logger("maim_message")
+    kwargs["custom_logger"] = maim_message_logger
+
+global_api = MessageServer(**kwargs)