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

better：优化中期记忆调用

Browse Source
This commit is contained in:
SengokuCola
2025-04-19 18:54:03 +08:00
parent 85dd3e83f6
commit 560fb738fc
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/do_tool/tool_can_use/mid_chat_mem.py
View File

@@ -9,11 +9,11 @@ class GetMidMemoryTool(BaseTool):
"""从记忆系统中获取相关记忆的工具""" """从记忆系统中获取相关记忆的工具"""
name = "mid_chat_mem" name = "mid_chat_mem"
description = "之前的聊天内容中获取具体信息，当最新消息提到，或者你需要回复的消息中提到，你可以使用这个工具" description = "之前的聊天内容概述id中获取具体信息，如果没有聊天内容概述id，就不要使用"
parameters = { parameters = {
"type": "object", "type": "object",
"properties": { "properties": {
"id": {"type": "integer", "description": "要查询的聊天记录id"}, "id": {"type": "integer", "description": "要查询的聊天记录概述id"},
}, },
"required": ["id"], "required": ["id"],
} }

2
src/heart_flow/observation.py
View File

@@ -161,7 +161,7 @@ class ChattingObservation(Observation):
if len(self.mid_memorys) > self.max_mid_memory_len: if len(self.mid_memorys) > self.max_mid_memory_len:
self.mid_memorys.pop(0) # 移除最旧的 self.mid_memorys.pop(0) # 移除最旧的
mid_memory_str = "之前聊天的内容概括是：\n" mid_memory_str = "之前聊天的内容概述是：\n"
for mid_memory_item in self.mid_memorys: # 重命名循环变量以示区分 for mid_memory_item in self.mid_memorys: # 重命名循环变量以示区分
time_diff = int((datetime.now().timestamp() - mid_memory_item["created_at"]) / 60) time_diff = int((datetime.now().timestamp() - mid_memory_item["created_at"]) / 60)
mid_memory_str += f"距离现在{time_diff}分钟前(聊天记录id:{mid_memory_item['id']})：{mid_memory_item['theme']}\n" mid_memory_str += f"距离现在{time_diff}分钟前(聊天记录id:{mid_memory_item['id']})：{mid_memory_item['theme']}\n"