Memobase 使用教程

1. 项目介绍

Memobase 是一个基于用户档案的记忆系统，旨在为生成式人工智能（GenAI）应用带来长期用户记忆。无论是构建虚拟伴侣、教育工具还是个性化助手，Memobase 都能赋予您的 AI 记忆、理解和随着用户成长的能力。

Memobase 提供了结构化的用户档案，能够记录用户的语言、婚姻状况、教育背景、兴趣、心理特征和职业等信息，从而让 AI 更好地理解和服务于用户。

2. 项目快速启动

环境准备

在开始之前，确保您已经安装了 Python 环境。

安装 Memobase

通过以下命令安装 Memobase Python SDK：

pip install memobase

启动 Memobase 后端

您需要准备以下两个东西来继续：

• 项目 URL。默认为 http://localhost:8019
• 项目令牌。默认为 secret

初始化 Memobase 客户端

在您的 Python 代码中，初始化 Memobase 客户端并连接到后端：

from memobase import MemoBaseClient, ChatBlob

mb = MemoBaseClient(
    project_url='http://localhost:8019',
    api_key='secret'
)

assert mb.ping()

管理用户

创建、更新和获取用户信息：

# 添加用户
uid = mb.add_user({"any_key": "any_value"})

# 更新用户信息
mb.update_user(uid, {"any_key": "any_value2"})

# 获取用户信息
user = mb.get_user(uid)
print(user)

插入数据

在 Memobase 中，所有类型的数据都是以 blob 的形式存在的，您可以插入、获取和删除这些数据：

# 插入数据
messages = [
    {"role": "user", "content": "Hello, I'm Gus"},
    {"role": "assistant", "content": "Hi, nice to meet you, Gus!"}
]

bid = user.insert(ChatBlob(messages=messages))
print(user.get(bid))

# 默认情况下，Memobase 处理完数据后会移除 blobs。
# user.delete(bid)

获取用户记忆

# 刷新并获取记忆
user.flush()

# 获取用户档案
print(user.profile(need_json=True))

3. 应用案例和最佳实践

用户记忆集成

将用户档案集成到您的 AI 提示中，以便更好地理解用户：

context = user.context(max_token_size=500, prefer_topics=["basic_info"])
print(context)

用户分析和跟踪

通过分析用户对话，记录用户偏好和行为：

profiles = user.profile()

def under_age_30(p):
    return p.sub_topic == "age" and int(p.content) < 30

def love_cat(p):
    return p.topic == "interest" and p.sub_topic == "pets" and "cat" in p.content

is_user_under_30 = len([p for p in profiles if under_age_30(p)]) > 0
is_user_love_cat = len([p for p in profiles if love_cat(p)]) > 0

推荐产品

根据用户职业推荐相应的产品：

def pick_an_ad(profiles):
    work_titles = [p for p in profiles if p.topic == "work" and p.sub_topic == "title"]
    if not work_titles:
        return None
    wt = work_titles[0].content
    if wt == "Software Engineer":
        return "Deep Learning Stuff"
    elif wt == "some job":
        return "some ads"

4. 典型生态项目

Memobase 可以与各种 AI 应用集成，例如聊天机器人、个性化教育工具等，形成一个强大的生态系统。通过记录和分析用户数据，这些项目可以更好地适应用户需求，提供更个性化的服务。