diff --git a/src/common/logger.py b/src/common/logger.py index 62dbfb051..ec2c887a5 100644 --- a/src/common/logger.py +++ b/src/common/logger.py @@ -349,6 +349,24 @@ BASE_TOOL_STYLE_CONFIG = { }, } +CHAT_STREAM_STYLE_CONFIG = { + "advanced": { + "console_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "聊天流 | " + "{message}" + ), + "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 聊天流 | {message}", + }, + "simple": { + "console_format": ( + "{time:MM-DD HH:mm} | 聊天流 | {message}" + ), + "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 聊天流 | {message}", + }, +} + PERSON_INFO_STYLE_CONFIG = { "advanced": { "console_format": ( @@ -526,6 +544,7 @@ BACKGROUND_TASKS_STYLE_CONFIG = ( BACKGROUND_TASKS_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else BACKGROUND_TASKS_STYLE_CONFIG["advanced"] ) MEMORY_STYLE_CONFIG = MEMORY_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else MEMORY_STYLE_CONFIG["advanced"] +CHAT_STREAM_STYLE_CONFIG = CHAT_STREAM_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else CHAT_STREAM_STYLE_CONFIG["advanced"] TOPIC_STYLE_CONFIG = TOPIC_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else TOPIC_STYLE_CONFIG["advanced"] SENDER_STYLE_CONFIG = SENDER_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else SENDER_STYLE_CONFIG["advanced"] LLM_STYLE_CONFIG = LLM_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else LLM_STYLE_CONFIG["advanced"] diff --git a/src/plugins/chat/chat_stream.py b/src/plugins/chat/chat_stream.py index ebeaa7c0f..e50dc3ec2 100644 --- a/src/plugins/chat/chat_stream.py +++ b/src/plugins/chat/chat_stream.py @@ -8,9 +8,14 @@ from typing import Dict, Optional from ...common.database import db from ..message.message_base import GroupInfo, UserInfo -from src.common.logger import get_module_logger +from src.common.logger import get_module_logger, LogConfig, CHAT_STREAM_STYLE_CONFIG -logger = get_module_logger("chat_stream") +chat_stream_log_config = LogConfig( + console_format=CHAT_STREAM_STYLE_CONFIG["console_format"], + file_format=CHAT_STREAM_STYLE_CONFIG["file_format"], +) + +logger = get_module_logger("chat_stream", config=chat_stream_log_config) class ChatStream: