之前介绍了一下概览,今天来看下快速开始

很简单,基本上就是CRUD


安装

pip install mem0ai

基本使用

初始化

基础
from mem0 import Memory
m = Memory()
高级

如果是在生产环境使用,如下

运行 qdrant 服务

docker pull qdrant/qdrant

docker run -p 6333:6333 -p 6334:6334 \
    -v $(pwd)/qdrant_storage:/qdrant/storage:z \
    qdrant/qdrant

初始化

from mem0 import Memory

config = {
    "vector_store": {
        "provider": "qdrant",
        "config": {
            "host": "localhost",
            "port": 6333,
        }
    },
}

m = Memory.from_config(config)

添加

# For a user
result = m.add("Likes to play cricket on weekends", user_id="alice", metadata={"category": "hobbies"})
print(result)

输出

[
  {
    'id': 'm1',
    'event': 'add',
    'data': 'Likes to play cricket on weekends'
  }
]

获取

# Get all memories
all_memories = m.get_all()
print(all_memories)

输出

[
  {
    'id': 'm1',
    'text': 'Likes to play cricket on weekends',
    'metadata': {
      'data': 'Likes to play cricket on weekends',
      'category': 'hobbies'
    }
  },
  # ... other memories ...
]
# Get a single memory by ID
specific_memory = m.get("m1")
print(specific_memory)

输出

{
  'id': 'm1',
  'text': 'Likes to play cricket on weekends',
  'metadata': {
    'data': 'Likes to play cricket on weekends',
    'category': 'hobbies'
  }
}

搜索

related_memories = m.search(query="What are Alice's hobbies?", user_id="alice")
print(related_memories)

输出

[
  {
    'id': 'm1',
    'text': 'Likes to play cricket on weekends',
    'metadata': {
      'data': 'Likes to play cricket on weekends',
      'category': 'hobbies'
    },
    'score': 0.85  # Similarity score
  },
  # ... other related memories ...
]

更新

result = m.update(memory_id="m1", data="Likes to play tennis on weekends")
print(result)

输出

{
  'id': 'm1',
  'event': 'update',
  'data': 'Likes to play tennis on weekends'
}

历史

history = m.history(memory_id="m1")
print(history)

输出

[
  {
    'id': 'h1',
    'memory_id': 'm1',
    'prev_value': None,
    'new_value': 'Likes to play cricket on weekends',
    'event': 'add',
    'timestamp': '2024-07-14 10:00:54.466687',
    'is_deleted': 0
  },
  {
    'id': 'h2',
    'memory_id': 'm1',
    'prev_value': 'Likes to play cricket on weekends',
    'new_value': 'Likes to play tennis on weekends',
    'event': 'update',
    'timestamp': '2024-07-14 10:15:17.230943',
    'is_deleted': 0
  }
]

删除

m.delete(memory_id="m1") # Delete a memory

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

重置

m.reset() # Reset all memories