1
This commit is contained in:
SengokuCola
2025-03-07 00:34:58 +08:00
parent 8ef00ee571
commit 71e851fbd4
2 changed files with 7 additions and 2 deletions

View File

@@ -60,6 +60,7 @@ ban_user_id = [] #禁止回复消息的QQ号
[model.llm_reasoning] #R1
name = "Pro/deepseek-ai/DeepSeek-R1"
# name = "Qwen/QwQ-32B"
base_url = "SILICONFLOW_BASE_URL"
key = "SILICONFLOW_KEY"

View File

@@ -672,7 +672,7 @@ class Hippocampus:
first_layer, _ = self.memory_graph.get_related_item(topic, depth=1)
if first_layer:
# 如果记忆条数超过限制,随机选择指定数量的记忆
if len(first_layer) > max_memory_num:
if len(first_layer) > max_memory_num/2:
first_layer = random.sample(first_layer, max_memory_num)
# 为每条记忆添加来源主题和相似度信息
for memory in first_layer:
@@ -681,10 +681,14 @@ class Hippocampus:
'similarity': score,
'content': memory
})
# 如果记忆数量超过5个,随机选择5个
# 按相似度排序
relevant_memories.sort(key=lambda x: x['similarity'], reverse=True)
if len(relevant_memories) > max_memory_num:
relevant_memories = random.sample(relevant_memories, max_memory_num)
return relevant_memories