fixruad

src/heart_flow/observation.py

@@ -46,7 +46,7 @@ class ChattingObservation(Observation):
         self.mid_memorys = []
         self.max_mid_memory_len = global_config.compress_length_limit
         self.mid_memory_info = ""
-        
+
         self.person_list = []
 
         self.llm_summary = LLMRequest(
@@ -157,11 +157,9 @@ class ChattingObservation(Observation):
         )
 
         self.person_list = await get_person_id_list(self.talking_message)
-        
-        # print(f"self.11111person_list: {self.person_list}")
-        
 
-        
+        # print(f"self.11111person_list: {self.person_list}")
+
         logger.trace(
             f"Chat {self.chat_id} - 压缩早期记忆：{self.mid_memory_info}\n现在聊天内容：{self.talking_message_str}"
         )

src/heart_flow/sub_mind.py

@@ -16,7 +16,6 @@ import difflib
 from src.plugins.person_info.relationship_manager import relationship_manager
 
 
-
 logger = get_logger("sub_heartflow")
 
 
@@ -60,6 +59,7 @@ def calculate_similarity(text_a: str, text_b: str) -> float:
     matcher = difflib.SequenceMatcher(None, text_a, text_b)
     return matcher.ratio()
 
+
 def calculate_replacement_probability(similarity: float) -> float:
     """
     根据相似度计算替换的概率。
@@ -138,13 +138,12 @@ class SubMind:
 
         # 获取个性化信息
         individuality = Individuality.get_instance()
-        
 
         relation_prompt = ""
         print(f"person_list: {person_list}")
         for person in person_list:
             relation_prompt += await relationship_manager.build_relationship_info(person, is_id=True)
-            
+
         print(f"relat22222ion_prompt: {relation_prompt}")
 
         # 构建个性部分
@@ -302,7 +301,7 @@ class SubMind:
             logger.warning(f"{self.log_prefix} LLM返回空结果，思考失败。")
 
         # ---------- 6. 应用概率性去重和修饰 ----------
-        new_content = content # 保存 LLM 直接输出的结果
+        new_content = content  # 保存 LLM 直接输出的结果
         try:
             similarity = calculate_similarity(previous_mind, new_content)
             replacement_prob = calculate_replacement_probability(similarity)
@@ -319,11 +318,13 @@ class SubMind:
                 if similarity == 1.0:
                     logger.debug(f"{self.log_prefix} 想法完全重复 (相似度 1.0)，执行特殊处理...")
                     # 随机截取大约一半内容
-                    if len(new_content) > 1: # 避免内容过短无法截取
-                        split_point = max(1, len(new_content) // 2 + random.randint(-len(new_content)//4, len(new_content)//4))
+                    if len(new_content) > 1:  # 避免内容过短无法截取
+                        split_point = max(
+                            1, len(new_content) // 2 + random.randint(-len(new_content) // 4, len(new_content) // 4)
+                        )
                         truncated_content = new_content[:split_point]
                     else:
-                        truncated_content = new_content # 如果只有一个字符或者为空，就不截取了
+                        truncated_content = new_content  # 如果只有一个字符或者为空，就不截取了
 
                     # 添加语气词和转折/承接词
                     yu_qi_ci = random.choice(yu_qi_ci_liebiao)
@@ -347,21 +348,21 @@ class SubMind:
                     if deduplicated_content:
                         # 根据概率决定是否添加词语
                         prefix_str = ""
-                        if random.random() < 0.3: # 30% 概率添加语气词
+                        if random.random() < 0.3:  # 30% 概率添加语气词
                             prefix_str += random.choice(yu_qi_ci_liebiao)
-                        if random.random() < 0.7: # 70% 概率添加转折/承接词
+                        if random.random() < 0.7:  # 70% 概率添加转折/承接词
                             prefix_str += random.choice(zhuan_jie_ci_liebiao)
 
                         # 组合最终结果
                         if prefix_str:
-                            content = f"{prefix_str}，{deduplicated_content}" # 更新 content
+                            content = f"{prefix_str}，{deduplicated_content}"  # 更新 content
                             logger.debug(f"{self.log_prefix} 去重并添加引导词后: {content}")
                         else:
-                            content = deduplicated_content # 更新 content
+                            content = deduplicated_content  # 更新 content
                             logger.debug(f"{self.log_prefix} 去重后 (未添加引导词): {content}")
                     else:
                         logger.warning(f"{self.log_prefix} 去重后内容为空，保留原始LLM输出: {new_content}")
-                        content = new_content # 保留原始 content
+                        content = new_content  # 保留原始 content
             else:
                 logger.debug(f"{self.log_prefix} 未执行概率性去重 (概率: {replacement_prob:.2f})")
                 # content 保持 new_content 不变

src/heart_flow/subheartflow_manager.py

@@ -266,9 +266,9 @@ class SubHeartflowManager:
 
             # --- 新增：检查是否允许进入 FOCUS 模式 --- #
             if not global_config.allow_focus_mode:
-                if int(time.time()) % 60 == 0: # 每60秒输出一次日志避免刷屏
+                if int(time.time()) % 60 == 0:  # 每60秒输出一次日志避免刷屏
                     logger.debug(f"{log_prefix} 配置不允许进入 FOCUSED 状态 (allow_focus_mode=False)")
-                return # 如果不允许，直接返回
+                return  # 如果不允许，直接返回
             # --- 结束新增 ---
 
             logger.debug(f"{log_prefix} 当前状态 ({current_state.value}) 开始尝试提升到FOCUSED状态")