OpenClaw
357cd0958f
18个插件: class Plugin -> class Liugin 10个插件: 测试块 Plugin() -> Liugin() 移除 ai_image.py 中多余的 Liugin = Plugin 别名 所有自研插件现统一使用 class Liugin 作为入口类名， 加载器保留 Plugin 兼容以支持第三方插件。
186 lines
7.4 KiB
Python
186 lines
7.4 KiB
Python
"""
|
||
记忆系统插件 — 包装 xcli_core/memory.py 中的 MemoryManager
|
||
使 AI 可以通过工具接口读写日记、搜索记忆、管理聊天记录
|
||
"""
|
||
import os
|
||
import sys
|
||
|
||
# 添加项目根目录到 path
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
project_root = os.path.dirname(current_dir)
|
||
sys.path.insert(0, project_root)
|
||
|
||
from xcli_core.memory import MemoryManager
|
||
|
||
|
||
class Liugin:
|
||
"""持久化记忆系统 — AI 可写日记、搜索记忆、管理聊天记录"""
|
||
|
||
def __init__(self):
|
||
self.cli = None
|
||
self.memory = None
|
||
self.usage = """持久化记忆系统工具
|
||
|
||
操作:
|
||
diary <内容> - 写当日日记
|
||
diary_ai <内容> - AI 写观察日记（标记为 AI 观察）
|
||
remember <内容> - 记录到长期记忆 (MEMORY.md)
|
||
search <关键词> - 全局搜索（日记+记忆+聊天记录）
|
||
context - 查看当前注入的记忆上下文
|
||
diary_list - 列出所有日记文件
|
||
diary_read <日期> - 读取指定日期的日记（如 2026-04-27）
|
||
chat_list - 列出最近的聊天记录
|
||
chat_search <关键词> - 在聊天记录中搜索
|
||
memory_read - 读取 MEMORY.md 长期记忆
|
||
memory_write <内容> - 写入 MEMORY.md（追加）
|
||
|
||
JSON 格式示例:
|
||
{"action": "use_tool", "tool": "memory_plugin", "args": "diary 今天完成了API对接"}
|
||
{"action": "use_tool", "tool": "memory_plugin", "args": "search 定时任务"}
|
||
{"action": "use_tool", "tool": "memory_plugin", "args": "remember 用户偏好深色主题"}
|
||
"""
|
||
|
||
def set_cli(self, cli):
|
||
self.cli = cli
|
||
# 初始化 MemoryManager
|
||
project_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
self.memory = MemoryManager(project_dir)
|
||
# 挂载到 cli 实例上，让 cli_core.py 能访问
|
||
cli.memory_manager = self.memory
|
||
|
||
def get_tool_info(self):
|
||
return {
|
||
"name": "memory_plugin",
|
||
"description": "持久化记忆系统 — AI 可写日记、记录长期记忆、搜索历史对话和日记。支持自动压缩上下文。",
|
||
"keywords": ["记忆", "memory", "日记", "diary", "搜索", "search", "历史", "history",
|
||
"remember", "聊天记录", "chat", "上下文", "context", "压缩", "compress"],
|
||
"usage": self.usage
|
||
}
|
||
|
||
def handle(self, args: str) -> str:
|
||
if not self.memory:
|
||
return "错误：记忆系统未初始化"
|
||
|
||
if not args or not args.strip():
|
||
return self._help()
|
||
|
||
parts = args.strip().split(None, 1)
|
||
action = parts[0].lower()
|
||
rest = parts[1].strip() if len(parts) > 1 else ""
|
||
|
||
handlers = {
|
||
"diary": self._diary,
|
||
"diary_ai": self._diary_ai,
|
||
"remember": self._remember,
|
||
"search": self._search,
|
||
"context": self._context,
|
||
"diary_list": self._diary_list,
|
||
"diary_read": self._diary_read,
|
||
"chat_list": self._chat_list,
|
||
"chat_search": self._chat_search,
|
||
"memory_read": self._memory_read,
|
||
"memory_write": self._memory_write,
|
||
"help": lambda _: self._help(),
|
||
}
|
||
|
||
handler = handlers.get(action)
|
||
if not handler:
|
||
return f"未知操作: {action}\n" + self._help()
|
||
|
||
return handler(rest)
|
||
|
||
def _diary(self, content: str) -> str:
|
||
if not content:
|
||
return "错误：请提供日记内容\n用法: diary <内容>"
|
||
self.memory.write_diary(content)
|
||
return f" 日记已写入 {self.memory._today_file()}"
|
||
|
||
def _diary_ai(self, content: str) -> str:
|
||
if not content:
|
||
return "错误：请提供 AI 观察内容\n用法: diary_ai <内容>"
|
||
self.memory.write_ai_diary(content)
|
||
return f" AI 观察日记已写入"
|
||
|
||
def _remember(self, content: str) -> str:
|
||
if not content:
|
||
return "错误：请提供要记住的内容\n用法: remember <内容>"
|
||
self.memory.append_memory("备忘", f"- {content}")
|
||
return f" 已记录到 MEMORY.md 长期记忆"
|
||
|
||
def _search(self, keyword: str) -> str:
|
||
if not keyword:
|
||
return "错误：请提供搜索关键词\n用法: search <关键词>"
|
||
results = self.memory.search_memory(keyword)
|
||
if not results:
|
||
return f" 未找到与 '{keyword}' 相关的记忆"
|
||
lines = [f" 搜索 '{keyword}' 的结果 ({len(results)} 条):"]
|
||
for r in results:
|
||
source = r.get("source", "")
|
||
snippet = r.get("snippet", "")
|
||
role = r.get("role", "")
|
||
line_num = r.get("line", "")
|
||
prefix = f"[{role}] " if role else ""
|
||
line_info = f" (L{line_num})" if line_num else ""
|
||
lines.append(f" {source}{line_info}: {prefix}{snippet}")
|
||
return "\n".join(lines)
|
||
|
||
def _context(self, _: str) -> str:
|
||
ctx = self.memory.build_memory_context()
|
||
return ctx if ctx else " 当前没有记忆上下文"
|
||
|
||
def _diary_list(self, _: str) -> str:
|
||
files = self.memory.list_diary_files()
|
||
if not files:
|
||
return " 还没有日记文件"
|
||
return " 日记文件:\n" + "\n".join(f" {f}" for f in files[:20])
|
||
|
||
def _diary_read(self, date_str: str) -> str:
|
||
if not date_str:
|
||
return "错误：请提供日期\n用法: diary_read <日期，如 2026-04-27>"
|
||
content = self.memory.load_diary(date_str)
|
||
return content if content else f" 没有找到 {date_str} 的日记"
|
||
|
||
def _chat_list(self, _: str) -> str:
|
||
chats = self.memory.list_chats()
|
||
if not chats:
|
||
return " 没有保存的聊天记录"
|
||
lines = [" 最近的聊天记录:"]
|
||
for c in chats:
|
||
lines.append(f" {c['name']} ({c['time']}, {c['size_kb']}KB)")
|
||
return "\n".join(lines)
|
||
|
||
def _chat_search(self, keyword: str) -> str:
|
||
if not keyword:
|
||
return "错误：请提供关键词\n用法: chat_search <关键词>"
|
||
results = self.memory.search_chats(keyword)
|
||
if not results:
|
||
return f" 未在聊天记录中找到 '{keyword}'"
|
||
lines = [f" 聊天记录搜索 '{keyword}':"]
|
||
for r in results:
|
||
lines.append(f" {r['file']} [{r['role']}]: {r['snippet']}")
|
||
return "\n".join(lines)
|
||
|
||
def _memory_read(self, _: str) -> str:
|
||
content = self.memory.load_memory()
|
||
return content if content else " MEMORY.md 为空"
|
||
|
||
def _memory_write(self, content: str) -> str:
|
||
if not content:
|
||
return "错误：请提供内容\n用法: memory_write <内容>"
|
||
self.memory.append_memory("备忘", f"- {content}")
|
||
return f" 已写入 MEMORY.md"
|
||
|
||
def _help(self) -> str:
|
||
return """ 记忆系统用法:
|
||
diary <内容> - 写当日日记
|
||
diary_ai <内容> - AI 写观察日记
|
||
remember <内容> - 记到长期记忆 (MEMORY.md)
|
||
search <关键词> - 全局搜索记忆
|
||
context - 查看记忆上下文
|
||
diary_list - 列出日记
|
||
diary_read <日期> - 读取日记
|
||
chat_list - 列出聊天记录
|
||
chat_search <关键词> - 搜索聊天记录
|
||
memory_read - 读取 MEMORY.md
|
||
memory_write <内容> - 写入 MEMORY.md"""
|