From ce8a390f8ab5e602748de8e7c22183a4bb4fb9ae Mon Sep 17 00:00:00 2001
From: "CNMr.Sunshine" <61444298+CNMrSunshine@users.noreply.github.com>
Date: Sat, 30 Aug 2025 17:32:22 +0800
Subject: [PATCH] Update heartflow_message_processor.py

---
 .../heart_flow/heartflow_message_processor.py | 34 ++++++++++++++-----
 1 file changed, 25 insertions(+), 9 deletions(-)

diff --git a/src/chat/heart_flow/heartflow_message_processor.py b/src/chat/heart_flow/heartflow_message_processor.py
index b1dccdaf..bb95337f 100644
--- a/src/chat/heart_flow/heartflow_message_processor.py
+++ b/src/chat/heart_flow/heartflow_message_processor.py
@@ -37,16 +37,32 @@ async def _calculate_interest(message: MessageRecv) -> Tuple[float, list[str]]:
     
     is_mentioned, _ = is_mentioned_bot_in_message(message)
     interested_rate = 0.0
+    keywords = []
+    keywords_lite = []
 
-    with Timer("记忆激活"):
-        interested_rate, keywords,keywords_lite = await hippocampus_manager.get_activate_from_text(
-            message.processed_plain_text,
-            max_depth= 4,
-            fast_retrieval=global_config.chat.interest_rate_mode == "fast",
-        )
-        message.key_words = keywords
-        message.key_words_lite = keywords_lite
-        logger.debug(f"记忆激活率: {interested_rate:.2f}, 关键词: {keywords}")
+    # 只有在记忆系统启用时才进行记忆激活
+    if global_config.memory.enable_memory:
+        with Timer("记忆激活"):
+            try:
+                interested_rate, keywords, keywords_lite = await hippocampus_manager.get_activate_from_text(
+                    message.processed_plain_text,
+                    max_depth= 4,
+                    fast_retrieval=global_config.chat.interest_rate_mode == "fast",
+                )
+                message.key_words = keywords
+                message.key_words_lite = keywords_lite
+                logger.debug(f"记忆激活率: {interested_rate:.2f}, 关键词: {keywords}")
+            except Exception as e:
+                logger.warning(f"记忆激活失败，跳过记忆功能: {e}")
+                interested_rate = 0.0
+                keywords = []
+                keywords_lite = []
+                message.key_words = keywords
+                message.key_words_lite = keywords_lite
+    else:
+        # 记忆系统被禁用时，设置默认值
+        message.key_words = []
+        message.key_words_lite = []
 
     text_len = len(message.processed_plain_text)
     # 根据文本长度分布调整兴趣度，采用分段函数实现更精确的兴趣度计算