ruff

src/chat/message_receive/bot.py

@@ -561,7 +561,11 @@ class ChatBot:
 
                 # 将兴趣度结果同步回原始消息，便于后续流程使用
                 message.interest_value = getattr(db_message, "interest_value", getattr(message, "interest_value", 0.0))
-                setattr(message, "should_reply", getattr(db_message, "should_reply", getattr(message, "should_reply", False)))
+                setattr(
+                    message,
+                    "should_reply",
+                    getattr(db_message, "should_reply", getattr(message, "should_reply", False)),
+                )
                 setattr(message, "should_act", getattr(db_message, "should_act", getattr(message, "should_act", False)))
 
                 # 存储消息到数据库，只进行一次写入

src/chat/message_receive/chat_stream.py

@@ -298,8 +298,10 @@ class ChatStream:
                     db_message.should_reply = result.should_reply
                     db_message.should_act = result.should_act
 
-                    logger.debug(f"消息 {db_message.message_id} 兴趣值已更新: {result.interest_value:.3f}, "
-                                f"should_reply: {result.should_reply}, should_act: {result.should_act}")
+                    logger.debug(
+                        f"消息 {db_message.message_id} 兴趣值已更新: {result.interest_value:.3f}, "
+                        f"should_reply: {result.should_reply}, should_act: {result.should_act}"
+                    )
                 else:
                     logger.warning(f"消息 {db_message.message_id} 兴趣值计算失败: {result.error_message}")
                     # 使用默认值
@@ -521,18 +523,17 @@ class ChatManager:
             # 优先使用缓存管理器（优化版本）
             try:
                 from src.chat.message_manager.stream_cache_manager import get_stream_cache_manager
+
                 cache_manager = get_stream_cache_manager()
 
                 if cache_manager.is_running:
                     optimized_stream = await cache_manager.get_or_create_stream(
-                        stream_id=stream_id,
-                        platform=platform,
-                        user_info=user_info,
-                        group_info=group_info
+                        stream_id=stream_id, platform=platform, user_info=user_info, group_info=group_info
                     )
 
                     # 设置消息上下文
                     from .message import MessageRecv
+
                     if stream_id in self.last_messages and isinstance(self.last_messages[stream_id], MessageRecv):
                         optimized_stream.set_context(self.last_messages[stream_id])
 
@@ -715,7 +716,7 @@ class ChatManager:
                 success = await batch_writer.schedule_stream_update(
                     stream_id=stream_data_dict["stream_id"],
                     update_data=ChatManager._prepare_stream_data(stream_data_dict),
-                    priority=1  # 流更新的优先级
+                    priority=1,  # 流更新的优先级
                 )
                 if success:
                     stream.saved = True
@@ -738,7 +739,7 @@ class ChatManager:
                 result = await batch_update(
                     model_class=ChatStreams,
                     conditions={"stream_id": stream_data_dict["stream_id"]},
-                    data=ChatManager._prepare_stream_data(stream_data_dict)
+                    data=ChatManager._prepare_stream_data(stream_data_dict),
                 )
                 if result and result > 0:
                     stream.saved = True
@@ -874,43 +875,43 @@ chat_manager = None
 
 
 def _convert_to_original_stream(self, optimized_stream) -> "ChatStream":
-        """将OptimizedChatStream转换为原始ChatStream以保持兼容性"""
-        try:
-            # 创建原始ChatStream实例
-            original_stream = ChatStream(
-                stream_id=optimized_stream.stream_id,
-                platform=optimized_stream.platform,
-                user_info=optimized_stream._get_effective_user_info(),
-                group_info=optimized_stream._get_effective_group_info()
-            )
+    """将OptimizedChatStream转换为原始ChatStream以保持兼容性"""
+    try:
+        # 创建原始ChatStream实例
+        original_stream = ChatStream(
+            stream_id=optimized_stream.stream_id,
+            platform=optimized_stream.platform,
+            user_info=optimized_stream._get_effective_user_info(),
+            group_info=optimized_stream._get_effective_group_info(),
+        )
 
-            # 复制状态
-            original_stream.create_time = optimized_stream.create_time
-            original_stream.last_active_time = optimized_stream.last_active_time
-            original_stream.sleep_pressure = optimized_stream.sleep_pressure
-            original_stream.base_interest_energy = optimized_stream.base_interest_energy
-            original_stream._focus_energy = optimized_stream._focus_energy
-            original_stream.no_reply_consecutive = optimized_stream.no_reply_consecutive
-            original_stream.saved = optimized_stream.saved
+        # 复制状态
+        original_stream.create_time = optimized_stream.create_time
+        original_stream.last_active_time = optimized_stream.last_active_time
+        original_stream.sleep_pressure = optimized_stream.sleep_pressure
+        original_stream.base_interest_energy = optimized_stream.base_interest_energy
+        original_stream._focus_energy = optimized_stream._focus_energy
+        original_stream.no_reply_consecutive = optimized_stream.no_reply_consecutive
+        original_stream.saved = optimized_stream.saved
 
-            # 复制上下文信息（如果存在）
-            if hasattr(optimized_stream, "_stream_context") and optimized_stream._stream_context:
-                original_stream.stream_context = optimized_stream._stream_context
+        # 复制上下文信息（如果存在）
+        if hasattr(optimized_stream, "_stream_context") and optimized_stream._stream_context:
+            original_stream.stream_context = optimized_stream._stream_context
 
-            if hasattr(optimized_stream, "_context_manager") and optimized_stream._context_manager:
-                original_stream.context_manager = optimized_stream._context_manager
+        if hasattr(optimized_stream, "_context_manager") and optimized_stream._context_manager:
+            original_stream.context_manager = optimized_stream._context_manager
 
-            return original_stream
+        return original_stream
 
-        except Exception as e:
-            logger.error(f"转换OptimizedChatStream失败: {e}")
-            # 如果转换失败，创建一个新的原始流
-            return ChatStream(
-                stream_id=optimized_stream.stream_id,
-                platform=optimized_stream.platform,
-                user_info=optimized_stream._get_effective_user_info(),
-                group_info=optimized_stream._get_effective_group_info()
-            )
+    except Exception as e:
+        logger.error(f"转换OptimizedChatStream失败: {e}")
+        # 如果转换失败，创建一个新的原始流
+        return ChatStream(
+            stream_id=optimized_stream.stream_id,
+            platform=optimized_stream.platform,
+            user_info=optimized_stream._get_effective_user_info(),
+            group_info=optimized_stream._get_effective_group_info(),
+        )
 
 
 def get_chat_manager():

src/chat/message_receive/optimized_chat_stream.py

@@ -80,10 +80,7 @@ class OptimizedChatStream:
     ):
         # 共享的只读数据
         self._shared_context = SharedContext(
-            stream_id=stream_id,
-            platform=platform,
-            user_info=user_info,
-            group_info=group_info
+            stream_id=stream_id, platform=platform, user_info=user_info, group_info=group_info
         )
 
         # 本地修改数据
@@ -269,14 +266,13 @@ class OptimizedChatStream:
         self._stream_context = StreamContext(
             stream_id=self.stream_id,
             chat_type=ChatType.GROUP if self.group_info else ChatType.PRIVATE,
-            chat_mode=ChatMode.NORMAL
+            chat_mode=ChatMode.NORMAL,
         )
 
         # 创建单流上下文管理器
         from src.chat.message_manager.context_manager import SingleStreamContextManager
-        self._context_manager = SingleStreamContextManager(
-            stream_id=self.stream_id, context=self._stream_context
-        )
+
+        self._context_manager = SingleStreamContextManager(stream_id=self.stream_id, context=self._stream_context)
 
     @property
     def stream_context(self):
@@ -331,9 +327,11 @@ class OptimizedChatStream:
         # 恢复stream_context信息
         if "stream_context_chat_type" in data:
             from src.plugin_system.base.component_types import ChatMode, ChatType
+
             instance.stream_context.chat_type = ChatType(data["stream_context_chat_type"])
         if "stream_context_chat_mode" in data:
             from src.plugin_system.base.component_types import ChatMode, ChatType
+
             instance.stream_context.chat_mode = ChatMode(data["stream_context_chat_mode"])
 
         # 恢复interruption_count信息
@@ -352,6 +350,7 @@ class OptimizedChatStream:
             if isinstance(actions, str):
                 try:
                     import json
+
                     actions = json.loads(actions)
                 except json.JSONDecodeError:
                     logger.warning(f"无法解析actions JSON字符串: {actions}")
@@ -458,7 +457,7 @@ class OptimizedChatStream:
             stream_id=self.stream_id,
             platform=self.platform,
             user_info=self._get_effective_user_info(),
-            group_info=self._get_effective_group_info()
+            group_info=self._get_effective_group_info(),
         )
 
         # 复制本地修改（但不触发写时复制）
@@ -482,9 +481,5 @@ def create_optimized_chat_stream(
 ) -> OptimizedChatStream:
     """创建优化版聊天流实例"""
     return OptimizedChatStream(
-        stream_id=stream_id,
-        platform=platform,
-        user_info=user_info,
-        group_info=group_info,
-        data=data
+        stream_id=stream_id, platform=platform, user_info=user_info, group_info=group_info, data=data
     )