diff --git a/config/bot_config_template.toml b/config/bot_config_template.toml index 28ffb0ce3..bc4ac18e3 100644 --- a/config/bot_config_template.toml +++ b/config/bot_config_template.toml @@ -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" diff --git a/src/plugins/memory_system/memory.py b/src/plugins/memory_system/memory.py index 840980783..c121ea1e3 100644 --- a/src/plugins/memory_system/memory.py +++ b/src/plugins/memory_system/memory.py @@ -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