BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor: 使用itertools.combinations简化节点连接逻辑

pull/784/head
春河晴 2025-04-17 15:56:49 +09:00
parent 78d7ce284a
commit 9a38aeafe1
No known key found for this signature in database
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/plugins/memory_system/Hippocampus.py
View File

@ -4,6 +4,8 @@ import math
import random import random
import time import time
import re import re
from itertools import combinations
import jieba import jieba
import networkx as nx import networkx as nx
import numpy as np import numpy as np
@ -1553,11 +1555,10 @@ class ParahippocampalGyrus:
last_modified=current_time, last_modified=current_time,
) )
for i in range(len(all_topics)): for topic1, topic2 in combinations(all_topics, 2):
for j in range(i + 1, len(all_topics)): logger.debug(f"连接同批次节点: {topic1} 和 {topic2}")
logger.debug(f"连接同批次节点: {all_topics[i]} 和 {all_topics[j]}") all_added_edges.append(f"{topic1}-{topic2}")
all_added_edges.append(f"{all_topics[i]}-{all_topics[j]}") self.memory_graph.connect_dot(topic1, topic2)
self.memory_graph.connect_dot(all_topics[i], all_topics[j])
logger.success(f"更新记忆: {', '.join(all_added_nodes)}") logger.success(f"更新记忆: {', '.join(all_added_nodes)}")
logger.debug(f"强化连接: {', '.join(all_added_edges)}") logger.debug(f"强化连接: {', '.join(all_added_edges)}")