From 5172f2d22c6b9db09827f942d7a3a7337f2e6f08 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A2=A8=E6=A2=93=E6=9F=92?= <1787882683@qq.com>
Date: Wed, 23 Apr 2025 23:30:59 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AB=9F=E7=84=B6=E8=BF=98=E6=BC=8F=E4=BA=86?=
 =?UTF-8?q?=E4=B8=80=E4=B8=AA=EF=BC=9F=E6=9D=80=EF=BC=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/common/logger.py                   | 21 +++++++++++++++++++++
 src/heart_flow/subheartflow_manager.py |  9 +++++++--
 2 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/src/common/logger.py b/src/common/logger.py
index 67a38594f..4347fd97e 100644
--- a/src/common/logger.py
+++ b/src/common/logger.py
@@ -385,6 +385,24 @@ BACKGROUND_TASKS_STYLE_CONFIG = {
     },
 }
 
+SUBHEARTFLOW_MANAGER_STYLE_CONFIG = {
+    "advanced": {
+        "console_format": (
+            "<white>{time:YYYY-MM-DD HH:mm:ss}</white> | "
+            "<level>{level: <8}</level> | "
+            "<light-blue>小脑袋管理</light-blue> | "
+            "<level>{message}</level>"
+        ),
+        "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 小脑袋管理 | {message}",
+    },
+    "simple": {
+        "console_format": (
+            "<level>{time:MM-DD HH:mm}</level> | <light-blue>小脑袋管理</light-blue> | <light-blue>{message}</light-blue>"
+        ),  # noqa: E501
+        "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 小脑袋管理 | {message}",
+    },
+}
+
 WILLING_STYLE_CONFIG = {
     "advanced": {
         "console_format": (
@@ -482,6 +500,9 @@ EMOJI_STYLE_CONFIG = EMOJI_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else EMOJI_ST
 REMOTE_STYLE_CONFIG = REMOTE_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else REMOTE_STYLE_CONFIG["advanced"]
 BASE_TOOL_STYLE_CONFIG = BASE_TOOL_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else BASE_TOOL_STYLE_CONFIG["advanced"]
 PERSON_INFO_STYLE_CONFIG = PERSON_INFO_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else PERSON_INFO_STYLE_CONFIG["advanced"]
+SUBHEARTFLOW_MANAGER_STYLE_CONFIG = (
+    SUBHEARTFLOW_MANAGER_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else SUBHEARTFLOW_MANAGER_STYLE_CONFIG["advanced"]
+)
 BACKGROUND_TASKS_STYLE_CONFIG = (
     BACKGROUND_TASKS_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else BACKGROUND_TASKS_STYLE_CONFIG["advanced"]
 )
diff --git a/src/heart_flow/subheartflow_manager.py b/src/heart_flow/subheartflow_manager.py
index 4a207f935..02f104bfa 100644
--- a/src/heart_flow/subheartflow_manager.py
+++ b/src/heart_flow/subheartflow_manager.py
@@ -4,7 +4,7 @@ import random
 from typing import Dict, Any, Optional, List
 
 # 导入日志模块
-from src.common.logger import get_module_logger
+from src.common.logger import get_module_logger, LogConfig, SUBHEARTFLOW_MANAGER_STYLE_CONFIG
 
 # 导入聊天流管理模块
 from src.plugins.chat.chat_stream import chat_manager
@@ -15,7 +15,12 @@ from src.heart_flow.mai_state_manager import MaiState, MaiStateInfo
 from .observation import ChattingObservation
 
 # 初始化日志记录器
-logger = get_module_logger("subheartflow_manager")
+
+subheartflow_manager_log_config = LogConfig(
+    console_format=SUBHEARTFLOW_MANAGER_STYLE_CONFIG["console_format"],
+    file_format=SUBHEARTFLOW_MANAGER_STYLE_CONFIG["file_format"], 
+)
+logger = get_module_logger("subheartflow_manager", config=subheartflow_manager_log_config)
 
 # 子心流管理相关常量
 INACTIVE_THRESHOLD_SECONDS = 1200  # 子心流不活跃超时时间(秒)