In [ ]:

%pip install mem0ai

基本用法¶

初始化 Mem0¶

In [1]:

from mem0 import Memory
m = Memory()

存储记忆¶

In [2]:

# 针对一个用户
result = m.add("周末喜欢打板球", user_id="alice", metadata={"category": "hobbies"})
print(result)

[{'id': '6954a481-413d-4025-b206-51c013515161', 'event': 'add', 'data': 'Likes playing cricket on weekends'}]

检索记忆¶

In [3]:

# 得到所有的回忆
all_memories = m.get_all()
print(all_memories)

[{'id': '6954a481-413d-4025-b206-51c013515161', 'text': 'Likes playing cricket on weekends', 'metadata': {'category': 'hobbies', 'user_id': 'alice', 'data': 'Likes playing cricket on weekends', 'created_at': 1721791053}}]

In [4]:

# 通过ID获取单个内存
specific_memory = m.get("6954a481-413d-4025-b206-51c013515161")
print(specific_memory)

{'id': '6954a481-413d-4025-b206-51c013515161', 'text': 'Likes playing cricket on weekends', 'metadata': {'category': 'hobbies', 'user_id': 'alice', 'data': 'Likes playing cricket on weekends', 'created_at': 1721791053}}

搜索记忆¶

In [5]:

related_memories = m.search(query="爱丽丝的爱好是什么？", user_id="alice")
print(related_memories)

[{'id': '6954a481-413d-4025-b206-51c013515161', 'text': 'Likes playing cricket on weekends', 'metadata': {'category': 'hobbies', 'user_id': 'alice', 'data': 'Likes playing cricket on weekends', 'created_at': 1721791053}, 'score': 0.17170298882127374}]

更新记忆¶

In [7]:

result = m.update(memory_id="6954a481-413d-4025-b206-51c013515161", data="周末喜欢打网球")
print(result)

None

记忆历史¶

In [9]:

history = m.history(memory_id="6954a481-413d-4025-b206-51c013515161")
print(history)

[{'id': 'eae5f646-ef1b-4176-98db-40e566469bcd', 'memory_id': '6954a481-413d-4025-b206-51c013515161', 'prev_value': None, 'new_value': 'Likes playing cricket on weekends', 'event': 'add', 'timestamp': '2024-07-24 03:17:33.773835', 'is_deleted': 0}, {'id': '7d4f1798-67a0-4ee3-83e8-6f99852b49fd', 'memory_id': '6954a481-413d-4025-b206-51c013515161', 'prev_value': 'Likes playing cricket on weekends', 'new_value': '周末喜欢打网球', 'event': 'update', 'timestamp': '2024-07-24 05:22:27.181007', 'is_deleted': 0}]

删除记忆¶

In [ ]:

m.delete(memory_id="23ef25f8-784d-4927-aa44-61ea95a06ad3") # Delete a memory

m.delete_all(user_id="alice") # Delete all memories

重置内存¶

In [ ]:

m.reset() # Reset all memories