Update get_memory.py

pull/998/head
2829798842 2025-05-28 19:15:50 +08:00 committed by GitHub
parent de393be088
commit b80c8a27c1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 9 deletions

View File

@ -10,12 +10,12 @@ class GetMemoryTool(BaseTool):
"""从记忆系统中获取相关记忆的工具""" """从记忆系统中获取相关记忆的工具"""
name = "get_memory" name = "get_memory"
description = "使用工具从记忆系统中获取相关记忆" description = "Use tool to retrieve relevant memories from the memory system"
parameters = { parameters = {
"type": "object", "type": "object",
"properties": { "properties": {
"topic": {"type": "string", "description": "要查询的相关主题,用逗号隔开"}, "topic": {"type": "string", "description": "Related topics to query, separated by commas"},
"max_memory_num": {"type": "integer", "description": "最大返回记忆数量"}, "max_memory_num": {"type": "integer", "description": "Maximum number of memories to return"},
}, },
"required": ["topic"], "required": ["topic"],
} }
@ -47,17 +47,17 @@ class GetMemoryTool(BaseTool):
memory_info += memory[1] + "\n" memory_info += memory[1] + "\n"
if memory_info: if memory_info:
content = f"你记得这些事情: {memory_info}\n" content = f"You remember these things: {memory_info}\n"
content += "以上是你的回忆,不一定是目前聊天里的人说的,也不一定是现在发生的事情,请记住。\n" content += "The above are your memories, not necessarily what people in the current chat said, nor necessarily what is happening now, please remember.\n"
else: else:
content = f"{topic}的记忆,你记不太清" content = f"Memories about {topic}, you don't remember clearly"
return {"type": "memory", "id": topic_list, "content": content} return {"type": "memory", "id": topic_list, "content": content}
except Exception as e: except Exception as e:
logger.error(f"记忆获取工具执行失败: {str(e)}") logger.error(f"Memory retrieval tool execution failed: {str(e)}")
# 在失败时也保持格式一致但id可能不适用或设为None/Error # Keep format consistent on failure, but id may not apply or be set to None/Error
return {"type": "memory_error", "id": topic_list, "content": f"记忆获取失败: {str(e)}"} return {"type": "memory_error", "id": topic_list, "content": f"Memory retrieval failed: {str(e)}"}
# 注册工具 # 注册工具