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

fix

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

1
config/bot_config_template.toml
View File

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

6
src/plugins/memory_system/memory.py
View File

@@ -672,7 +672,7 @@ class Hippocampus:
first_layer, _ = self.memory_graph.get_related_item(topic, depth=1) first_layer, _ = self.memory_graph.get_related_item(topic, depth=1)
if first_layer: 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) first_layer = random.sample(first_layer, max_memory_num)
# 为每条记忆添加来源主题和相似度信息 # 为每条记忆添加来源主题和相似度信息
for memory in first_layer: for memory in first_layer:
@@ -682,9 +682,13 @@ class Hippocampus:
'content': memory 'content': memory
}) })
# 如果记忆数量超过5个,随机选择5个
# 按相似度排序 # 按相似度排序
relevant_memories.sort(key=lambda x: x['similarity'], reverse=True) 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 return relevant_memories