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

This commit is contained in:
春河晴
2025-04-17 15:56:49 +09:00
parent 78d7ce284a
commit 9a38aeafe1

View File

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