ruff

src/chat/heart_flow/observation/chatting_observation.py

@@ -140,22 +140,22 @@ class ChattingObservation(Observation):
                 return None
 
         # logger.debug(f"找到的锚定消息：find_msg: {find_msg}")
-        
+
         # 创建所需的user_info字段
         user_info = {
             "platform": find_msg.get("user_platform", ""),
             "user_id": find_msg.get("user_id", ""),
             "user_nickname": find_msg.get("user_nickname", ""),
-            "user_cardname": find_msg.get("user_cardname", "")
+            "user_cardname": find_msg.get("user_cardname", ""),
         }
-        
+
         # 创建所需的group_info字段，如果是群聊的话
         group_info = {}
         if find_msg.get("chat_info_group_id"):
             group_info = {
                 "platform": find_msg.get("chat_info_group_platform", ""),
                 "group_id": find_msg.get("chat_info_group_id", ""),
-                "group_name": find_msg.get("chat_info_group_name", "")
+                "group_name": find_msg.get("chat_info_group_name", ""),
             }
 
         content_format = ""
@@ -196,7 +196,7 @@ class ChattingObservation(Observation):
             limit=self.max_now_obs_len,
             limit_mode="latest",
         )
-        
+
         # print(f"new_messages_list: {new_messages_list}")
 
         last_obs_time_mark = self.last_observe_time

src/chat/memory_system/Hippocampus.py

@@ -861,9 +861,7 @@ class EntorhinalCortex:
                         # 确保在更新前获取最新的 memorized_times
                         current_memorized_times = message.get("memorized_times", 0)
                         # 使用 Peewee 更新记录
-                        Messages.update(
-                            memorized_times=current_memorized_times + 1
-                        ).where(
+                        Messages.update(memorized_times=current_memorized_times + 1).where(
                             Messages.message_id == message["message_id"]
                         ).execute()
                     return messages  # 直接返回原始的消息列表
@@ -983,9 +981,7 @@ class EntorhinalCortex:
                 if not node.last_modified:
                     update_data["last_modified"] = current_time
 
-                GraphNodes.update(
-                    **update_data
-                ).where(GraphNodes.concept == concept).execute()
+                GraphNodes.update(**update_data).where(GraphNodes.concept == concept).execute()
                 logger.info(f"[时间更新] 节点 {concept} 添加缺失的时间字段")
 
             # 获取时间信息(如果不存在则使用当前时间)
@@ -1014,9 +1010,7 @@ class EntorhinalCortex:
                 if not edge.last_modified:
                     update_data["last_modified"] = current_time
 
-                GraphEdges.update(
-                    **update_data
-                ).where(
+                GraphEdges.update(**update_data).where(
                     (GraphEdges.source == source) & (GraphEdges.target == target)
                 ).execute()
                 logger.info(f"[时间更新] 边 {source} - {target} 添加缺失的时间字段")

src/chat/utils/chat_message_builder.py

@@ -175,15 +175,15 @@ async def _build_readable_messages_internal(
     # 1 & 2: 获取发送者信息并提取消息组件
     for msg in messages:
         # 检查并修复缺少的user_info字段
-        if 'user_info' not in msg:
+        if "user_info" not in msg:
             # 创建user_info字段
-            msg['user_info'] = {
-                'platform': msg.get('user_platform', ''),
-                'user_id': msg.get('user_id', ''),
-                'user_nickname': msg.get('user_nickname', ''),
-                'user_cardname': msg.get('user_cardname', '')
+            msg["user_info"] = {
+                "platform": msg.get("user_platform", ""),
+                "user_id": msg.get("user_id", ""),
+                "user_nickname": msg.get("user_nickname", ""),
+                "user_cardname": msg.get("user_cardname", ""),
             }
-        
+
         user_info = msg.get("user_info", {})
         platform = user_info.get("platform")
         user_id = user_info.get("user_id")

src/common/database/database_model.py

@@ -279,6 +279,7 @@ class GraphNodes(BaseModel):
     """
     用于存储记忆图节点的模型
     """
+
     concept = TextField(unique=True, index=True)  # 节点概念
     memory_items = TextField()  # JSON格式存储的记忆列表
     hash = TextField()  # 节点哈希值
@@ -293,6 +294,7 @@ class GraphEdges(BaseModel):
     """
     用于存储记忆图边的模型
     """
+
     source = TextField(index=True)  # 源节点
     target = TextField(index=True)  # 目标节点
     strength = IntegerField()  # 连接强度