From a7278a37c77787ab24a466f201a8b6b2d3277325 Mon Sep 17 00:00:00 2001
From: SengokuCola <1026294844@qq.com>
Date: Fri, 21 Mar 2025 17:59:13 +0800
Subject: [PATCH] =?UTF-8?q?better=20cmd=E6=B8=85=E7=90=86=E5=A4=A7?=
 =?UTF-8?q?=E5=B8=88?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/plugins/chat/utils.py           |  2 +-
 src/plugins/chat/utils_image.py     |  2 +-
 src/plugins/memory_system/memory.py | 16 ++++++++++++----
 3 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/src/plugins/chat/utils.py b/src/plugins/chat/utils.py
index fd940a645..cc53db623 100644
--- a/src/plugins/chat/utils.py
+++ b/src/plugins/chat/utils.py
@@ -314,7 +314,7 @@ def split_into_sentences_w_remove_punctuation(text: str) -> List[str]:
             sentence = sentence.replace("，", " ").replace(",", " ")
         sentences_done.append(sentence)
 
-    logger.info(f"处理后的句子: {sentences_done}")
+    logger.debug(f"处理后的句子: {sentences_done}")
     return sentences_done
 
 
diff --git a/src/plugins/chat/utils_image.py b/src/plugins/chat/utils_image.py
index 521795024..7e20b35db 100644
--- a/src/plugins/chat/utils_image.py
+++ b/src/plugins/chat/utils_image.py
@@ -184,7 +184,7 @@ class ImageManager:
                 logger.warning(f"虽然生成了描述，但是找到缓存图片描述 {cached_description}")
                 return f"[图片：{cached_description}]"
 
-            logger.info(f"描述是{description}")
+            logger.debug(f"描述是{description}")
 
             if description is None:
                 logger.warning("AI未能生成图片描述")
diff --git a/src/plugins/memory_system/memory.py b/src/plugins/memory_system/memory.py
index ece8de748..6efbddd56 100644
--- a/src/plugins/memory_system/memory.py
+++ b/src/plugins/memory_system/memory.py
@@ -379,6 +379,7 @@ class Hippocampus:
         start_time = time.time()
         memory_samples = self.get_memory_sample()
         all_added_nodes = []
+        all_connected_nodes = []
         all_added_edges = []
         for i, messages in enumerate(memory_samples, 1):
             all_topics = []
@@ -396,6 +397,7 @@ class Hippocampus:
             current_time = datetime.datetime.now().timestamp()
             logger.debug(f"添加节点: {', '.join(topic for topic, _ in compressed_memory)}")
             all_added_nodes.extend(topic for topic, _ in compressed_memory)
+            # all_connected_nodes.extend(topic for topic, _ in similar_topics_dict)
             
             for topic, memory in compressed_memory:
                 self.memory_graph.add_dot(topic, memory)
@@ -407,8 +409,13 @@ class Hippocampus:
                     for similar_topic, similarity in similar_topics:
                         if topic != similar_topic:
                             strength = int(similarity * 10)
+                            
                             logger.debug(f"连接相似节点: {topic} 和 {similar_topic} (强度: {strength})")
                             all_added_edges.append(f"{topic}-{similar_topic}")
+                            
+                            all_connected_nodes.append(topic)
+                            all_connected_nodes.append(similar_topic)
+                            
                             self.memory_graph.G.add_edge(
                                 topic,
                                 similar_topic,
@@ -425,7 +432,8 @@ class Hippocampus:
                     self.memory_graph.connect_dot(all_topics[i], all_topics[j])
 
         logger.success(f"更新记忆: {', '.join(all_added_nodes)}")
-        logger.success(f"强化连接: {', '.join(all_added_edges)}")
+        logger.debug(f"强化连接: {', '.join(all_added_edges)}")
+        logger.info(f"强化连接节点: {', '.join(all_connected_nodes)}")
         # logger.success(f"强化连接: {', '.join(all_added_edges)}")
         self.sync_memory_to_db()
         
@@ -860,10 +868,9 @@ class Hippocampus:
 
     async def memory_activate_value(self, text: str, max_topics: int = 5, similarity_threshold: float = 0.3) -> int:
         """计算输入文本对记忆的激活程度"""
-        logger.info(f"识别主题: {await self._identify_topics(text)}")
-
         # 识别主题
         identified_topics = await self._identify_topics(text)
+        
         if not identified_topics:
             return 0
 
@@ -924,7 +931,8 @@ class Hippocampus:
 
         # 计算最终激活值
         activation = int((topic_match + average_similarities) / 2 * 100)
-        logger.info(f"匹配率: {topic_match:.3f}, 平均相似度: {average_similarities:.3f}, 激活值: {activation}")
+        
+        logger.info(f"识别主题: {identified_topics}, 匹配率: {topic_match:.3f}, 激活值: {activation}")
 
         return activation