From 80ff6e81549413b3fe366fb4ba06bb51552831bb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A2=A8=E6=A2=93=E6=9F=92?= <1787882683@qq.com>
Date: Sun, 4 May 2025 01:43:44 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0API=E6=9C=8D=E5=8A=A1?=
=?UTF-8?q?=E5=99=A8=E6=97=A5=E5=BF=97=E6=A0=B7=E5=BC=8F=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
src/common/logger.py | 20 +++++++++++++++++++-
src/common/logger_manager.py | 2 ++
2 files changed, 21 insertions(+), 1 deletion(-)
diff --git a/src/common/logger.py b/src/common/logger.py
index a82c6d883..432b1bdca 100644
--- a/src/common/logger.py
+++ b/src/common/logger.py
@@ -808,6 +808,22 @@ INIT_STYLE_CONFIG = {
},
}
+API_SERVER_STYLE_CONFIG = {
+ "advanced": {
+ "console_format": (
+ "{time:YYYY-MM-DD HH:mm:ss} | "
+ "{level: <8} | "
+ "API服务 | "
+ "{message}"
+ ),
+ "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | API服务 | {message}",
+ },
+ "simple": {
+ "console_format": "{time:MM-DD HH:mm} | API服务 | {message}",
+ "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | API服务 | {message}",
+ },
+}
+
# 根据SIMPLE_OUTPUT选择配置
MAIN_STYLE_CONFIG = MAIN_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else MAIN_STYLE_CONFIG["advanced"]
@@ -878,7 +894,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"]
+)
def is_registered_module(record: dict) -> bool:
"""检查是否为已注册的模块"""
diff --git a/src/common/logger_manager.py b/src/common/logger_manager.py
index 5c5538385..4c28f82f8 100644
--- a/src/common/logger_manager.py
+++ b/src/common/logger_manager.py
@@ -41,6 +41,7 @@ from src.common.logger import (
CHAT_MESSAGE_STYLE_CONFIG,
CHAT_IMAGE_STYLE_CONFIG,
INIT_STYLE_CONFIG,
+ API_SERVER_STYLE_CONFIG,
)
# 可根据实际需要补充更多模块配置
@@ -86,6 +87,7 @@ MODULE_LOGGER_CONFIGS = {
"chat_message": CHAT_MESSAGE_STYLE_CONFIG, # 聊天消息
"chat_image": CHAT_IMAGE_STYLE_CONFIG, # 聊天图片
"init": INIT_STYLE_CONFIG, # 初始化
+ "api": API_SERVER_STYLE_CONFIG, # API服务器
# ...如有更多模块,继续添加...
}