ES-Memory — 个人版 AI 记忆引擎
> MIT License · 纯本地 · 零联网 · 单文件存储
30 秒快速开始
git clone <this-repo>
cd es-memory
pip install jieba # 可选,装上中文分词更准
python3 example.py # 一键跑通所有功能
1 分钟上手
from es_memory.engine import ESMemory1
mem = ESMemory1(data_dir="./my_data")
# 加记忆
n1 = mem.add("客户A是大客户", keywords=["客户", "VIP"], tags=["业务"])
# 查记忆
for r in mem.search("客户", top_k=3):
print(f"{r.weight:.0f}% | {r.content}")
# 钉住重要记忆(免衰减)
mem.pin(n1.id)
# 看状态
print(mem.stats())
CLI 用法
python -m es_memory.cli add "客户A是大客户" --kw 客户 --tag 业务
python -m es_memory.cli search "客户" --top 3
python -m es_memory.cli list
python -m es_memory.cli pin mem-xxxxxxxxxxxx
python -m es_memory.cli decay --dry-run
python -m es_memory.cli stats
核心特性
- 艾宾浩斯衰减:低权重记忆自动清理,释放本地空间
- 钉住免衰减:重要记忆用
pin()锁定,永不删除 - 中文分词:支持 jieba 精确模式,缺失时自动回退
- 纯本地:数据只存在你传的目录,绝不联网
数据格式
- 存储:
<data_dir>/nodes.json,明文 JSON - 备份:整个
data_dir拷走即可 - 迁移:不同机器拷过去,重启自动加载
License
MIT