一个专为大语言模型设计的记忆管理系统,支持基于重要性的记忆筛选和持久化存储。
使用以下命令安装:
pip install llm-memory
from llm_memory import CombinedMemory
# 初始化记忆系统
memory = CombinedMemory(
max_size=100, # 最大记忆容量
system_prompt="你是一个助手", # 系统提示词
uid="user123", # 用户ID,用于数据库存储
recent_size=20 # 保留最近对话轮数
)
# 添加消息
memory.add_message("user", "你好", importance=1)
memory.add_message("assistant", "你好!有什么我可以帮你的吗?", importance=1)
# 获取上下文
context = memory.get_context()
# 保存到数据库
memory.save_memory_to_db()
- system_prompt: 系统提示词,可选
- uid: 用户唯一标识,用于数据库存储,可选
- max_size: 最大记忆容量,必需
- recent_size: 保留最近对话轮数,默认为20轮
- 基于重要性的记忆管理
- 自动保持最近的对话(可配置保留轮数)
- 支持数据库持久化
- 可配置的系统提示词
当提供 uid 参数时,会自动将对话历史保存到本地数据库(使用 TinyDB)。数据库文件默认保存为 memory.json。
Python >= 3.7 tinydb >= 4.7.0
MIT License