fix: 为maim_message添加logger

This commit is contained in:
tcmofashi
2025-05-15 07:36:07 +08:00
parent ac6f96f805
commit cb0cbfe84d
3 changed files with 44 additions and 1 deletions

View File

@@ -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)