[AI Mem0] 快速开始：智能记忆管理，让你的数据活起来！

2024-07-22 09:52:12 浏览数 (1)
  • [AI Mem0] 概览，智能自我改进记忆层

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

很简单，基本上就是CRUD

安装

代码语言：bash复制
pip install mem0ai

基本使用

初始化

基础
代码语言：python代码运行次数：0复制
from mem0 import Memory
m = Memory()
高级

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

运行 qdrant 服务

代码语言：bash复制
docker pull qdrant/qdrant

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

初始化

代码语言：python代码运行次数：0复制
from mem0 import Memory

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

m = Memory.from_config(config)

添加

代码语言：python代码运行次数：0复制
# For a user
result = m.add("Likes to play cricket on weekends", user_id="alice", metadata={"category": "hobbies"})
print(result)

输出

代码语言：json复制
[
  {
    'id': 'm1',
    'event': 'add',
    'data': 'Likes to play cricket on weekends'
  }
]

获取

代码语言：python代码运行次数：0复制
# Get all memories
all_memories = m.get_all()
print(all_memories)

输出

代码语言：json复制
[
  {
    'id': 'm1',
    'text': 'Likes to play cricket on weekends',
    'metadata': {
      'data': 'Likes to play cricket on weekends',
      'category': 'hobbies'
    }
  },
  # ... other memories ...
]
代码语言：python代码运行次数：0复制
# Get a single memory by ID
specific_memory = m.get("m1")
print(specific_memory)

输出

代码语言：json复制
{
  'id': 'm1',
  'text': 'Likes to play cricket on weekends',
  'metadata': {
    'data': 'Likes to play cricket on weekends',
    'category': 'hobbies'
  }
}

搜索

代码语言：python代码运行次数：0复制
related_memories = m.search(query="What are Alice's hobbies?", user_id="alice")
print(related_memories)

输出

代码语言：json复制
[
  {
    '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 ...
]

更新

代码语言：python代码运行次数：0复制
result = m.update(memory_id="m1", data="Likes to play tennis on weekends")
print(result)

输出

代码语言：json复制
{
  'id': 'm1',
  'event': 'update',
  'data': 'Likes to play tennis on weekends'
}

历史

代码语言：python代码运行次数：0复制
history = m.history(memory_id="m1")
print(history)

输出

代码语言：json复制
[
  {
    '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
  }
]

删除

代码语言：python代码运行次数：0复制
m.delete(memory_id="m1") # Delete a memory

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

重置

代码语言：python代码运行次数：0复制
m.reset() # Reset all memories
  • github
  • doc
  • AI 博客 - 从零开始学AI
  • 公众号 - 从零开始学AI
人工智能

0 人点赞