🤖 自动格式化代码 [skip ci]

src/chat/focus_chat/info_processors/working_memory_processor.py

@@ -8,7 +8,6 @@ from src.common.logger import get_logger
 from src.chat.utils.prompt_builder import Prompt, global_prompt_manager
 from src.chat.message_receive.chat_stream import get_chat_manager
 from .base_processor import BaseProcessor
-from src.chat.focus_chat.info.mind_info import MindInfo
 from typing import List
 from src.chat.heart_flow.observation.working_observation import WorkingMemoryObservation
 from src.chat.focus_chat.working_memory.working_memory import WorkingMemory
@@ -138,7 +137,9 @@ class WorkingMemoryProcessor(BaseProcessor):
                 logger.error(traceback.format_exc())
                 return []
 
-            logger.debug(f"{self.log_prefix} 解析LLM返回的JSON,selected_memory_ids: {selected_memory_ids}, merge_memory: {merge_memory}")
+            logger.debug(
+                f"{self.log_prefix} 解析LLM返回的JSON,selected_memory_ids: {selected_memory_ids}, merge_memory: {merge_memory}"
+            )
 
             # 根据selected_memory_ids，调取记忆
             memory_str = ""
@@ -208,11 +209,7 @@ class WorkingMemoryProcessor(BaseProcessor):
                     return
 
                 # 创建新记忆
-                await working_memory.add_memory(
-                    from_source="chat_compress",
-                    summary=content,
-                    brief=theme
-                )
+                await working_memory.add_memory(from_source="chat_compress", summary=content, brief=theme)
 
                 logger.debug(f"{self.log_prefix} 压缩聊天记忆成功: {theme} - {content}")
 

src/chat/focus_chat/working_memory/memory_item.py

@@ -1,4 +1,4 @@
-from typing import Dict, Any, Tuple
+from typing import Tuple
 import time
 import random
 import string

src/chat/focus_chat/working_memory/memory_manager.py

@@ -1,4 +1,4 @@
-from typing import Dict, Any, Type, TypeVar, List, Optional
+from typing import Dict, TypeVar, List, Optional
 import traceback
 from json_repair import repair_json
 from rich.traceback import install
@@ -263,7 +263,7 @@ class MemoryManager:
             return False
 
         # 获取要删除的项
-        item = self._id_map[memory_id]
+        self._id_map[memory_id]
 
         # 从内存中删除
         self._memories = [i for i in self._memories if i.id != memory_id]
@@ -376,7 +376,9 @@ class MemoryManager:
         )
 
         # 创建新的记忆项
-        merged_memory = MemoryItem(summary=merged_data["summary"], from_source=merged_source, brief=merged_data["brief"])
+        merged_memory = MemoryItem(
+            summary=merged_data["summary"], from_source=merged_source, brief=merged_data["brief"]
+        )
 
         # 记忆强度取两者最大值
         merged_memory.memory_strength = max(memory_item1.memory_strength, memory_item2.memory_strength)

src/chat/focus_chat/working_memory/working_memory.py

@@ -54,7 +54,7 @@ class WorkingMemory:
             except Exception as e:
                 print(f"自动衰减记忆时出错: {str(e)}")
 
-    async def add_memory(self, summary: Any, from_source: str = "",brief: str = ""):
+    async def add_memory(self, summary: Any, from_source: str = "", brief: str = ""):
         """
         添加一段记忆到指定聊天
 

src/chat/heart_flow/observation/chatting_observation.py

@@ -1,6 +1,5 @@
 from datetime import datetime
 from src.config.config import global_config
-import traceback
 from src.chat.utils.chat_message_builder import (
     get_raw_msg_before_timestamp_with_chat,
     build_readable_messages,
@@ -50,6 +49,7 @@ Prompt(
     "chat_summary_private_prompt",  # Template for private chat
 )
 
+
 class ChattingObservation(Observation):
     def __init__(self, chat_id):
         super().__init__(chat_id)
@@ -192,19 +192,13 @@ class ChattingObservation(Observation):
 
             # 构建压缩提示
             oldest_messages_str = build_readable_messages(
-                messages=oldest_messages, 
-                timestamp_mode="normal_no_YMD", 
-                read_mark=0, 
-                show_actions=True
+                messages=oldest_messages, timestamp_mode="normal_no_YMD", read_mark=0, show_actions=True
             )
 
             # 根据聊天类型选择提示模板
             if self.is_group_chat:
                 prompt_template_name = "chat_summary_group_prompt"
-                prompt = await global_prompt_manager.format_prompt(
-                    prompt_template_name, 
-                    chat_logs=oldest_messages_str
-                )
+                prompt = await global_prompt_manager.format_prompt(prompt_template_name, chat_logs=oldest_messages_str)
             else:
                 prompt_template_name = "chat_summary_private_prompt"
                 chat_target_name = "对方"