简介

相比于那些更繁杂的ai本地化记忆agent，ezmem提供了一个超轻量级，无需额外挂载数据库，上手即用的简单记忆逻辑

使用 pip 安装

pip install ezmem

功能示例

架构示例

效果展示

简易上手

from ezmem import EasyMemory #导入包

config = OpenAI(
    api_key="填入您自己的api",
    base_url="填入您自己的url",
)#api_key 以及 base_url配置

#快速上下文聊天，单条指令即可完成，速度较慢
res = m.chat_with_context(user_id=_.user_id,text=text,base_prompt=user_promet,memory_round=5)

#以下为单轮快速聊天
m = EasyMemory(config) #创建memory实例

m.add(text,user_id)#将记忆信息记录进user_id中

memory = m.query(user_id)#根据user_id查询记忆内容

res = m.chat(text,rule)#text：用户聊天信息，rule：ai遵循规则

#删除记忆信息
m.delete_all(user_id)#删除user_id的所有记忆信息