gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
春河晴
2025-04-17 15:56:49 +09:00
parent 78d7ce284a
commit 9a38aeafe1
1 changed files with 6 additions and 5 deletions
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)}")