一个能给大模型加上“记忆”的工具，让 AI 助手更懂你，Star 26K ！ - 链载Ai

ingFang SC", "Hiragino Sans GB", "Microsoft YaHei UI", "Microsoft YaHei", Arial, sans-serif;letter-spacing: 0.544px;visibility: visible;">? 原理流程

用户输入文本：用户向系统输入一段文本。
文本嵌入：将文本转换为向量表示，以便进行相似性搜索和记忆存储。
记忆提取：从文本中提取出有价值的记忆信息，如用户的偏好和事实。
记忆存储：将提取的记忆信息存储在向量数据库中。
LLM处理：利用大语言模型（LLM）生成回答。
记忆更新：将新的对话内容作为记忆写入系统。
记忆检索：在后续对话中检索相关记忆，为LLM提供上下文信息。

—

Mem0安装

安装 Mem0 非常简单，可以通过 pip 命令进行安装：

pipinstallmem0ai

—

Mem0使用

下面是一个使用 Mem0 的 Python 代码示例，展示了如何在对话中利用 Mem0 记忆功能增强 AI 的交互能力。

一、示例代码：

importrequestsfrommem0importMemory
# 初始化记忆对象memory = Memory()
defchat_with_memories(message:str, user_id:str="default_user") ->str: """  利用 Mem0 和 Ollama 部署的 DeepSeek 实现带有记忆功能的对话
  Args:    message (str): 用户输入的消息    user_id (str, optional): 用户的唯一标识符，默认为 "default_user"
  Returns:    str: AI 的回答  """ # 搜索与当前消息相关的记忆  relevant_memories = memory.search(query=message, user_id=user_id, limit=3)
 # 将记忆格式化为字符串，以便在提示中使用  memories_str ="\n".join(f"-{entry['memory']}"forentryinrelevant_memories)
 # 定义系统提示，包含记忆信息  system_prompt =f"你是一个乐于助人的AI。请根据用户的问题和相关记忆来回答。\n用户记忆：\n{memories_str}"
 # 构建提示内容，包含系统提示和用户消息  prompt =f"{system_prompt}\n\n用户问题：{message}"
 # 调用 Ollama 部署的 DeepSeek API 获取回答  response = requests.post(   "http://xxx.xxx.xxx.xxx:11434/api/generate",    json={     "model":"deepseek-r1:1.5b",     "prompt": prompt,     "stream": false    }  ).json()
 # 提取助手的回答内容  assistant_response = response.get('response','很抱歉，我无法理解你的问题。')
 # 将对话内容作为记忆写入  memory.add([    {"role":"user","content": message},    {"role":"assistant","content": assistant_response}  ], user_id=user_id)
 returnassistant_response
defmain(): """主函数，用于与 AI 进行对话""" print("与AI聊天（输入'exit'退出）") whileTrue:    user_input =input("您: ").strip()   ifuser_input.lower() =='exit':     print("再见！")     break   print(f"AI:{chat_with_memories(user_input)}")
if__name__ =="__main__":  main()

代码说明

初始化记忆对象：

memory=Memory()

创建一个记忆对象，用于存储和检索用户记忆。

定义对话函数chat_with_memories：

参数：

message：用户输入的消息。
user_id：用户的唯一标识符，用于区分不同用户的记忆。

功能：

搜索与当前消息相关的记忆。
格式化记忆信息，构建系统提示。
调用 Ollama 部署的 DeepSeek API 获取 AI 的回答。
将对话内容作为记忆写入 Mem0

主函数main：

提供一个简单的命令行界面，可以与 AI 进行对话。
循环获取用户输入，调用chat_with_memories函数获取 AI 回答，并打印结果。

二、输出结果展示

与AI聊天（输入'exit'退出）您:你好，今天天气怎么样？AI:你好！今天天气不错，适合出去走走。你最近喜欢什么样的天气呢？您:我喜欢晴天。AI:晴天确实是个好天气，阳光明媚，心情也会变好呢！那你会在晴天做些什么呢？您:我会在晴天去公园散步。AI:听起来很惬意呢！下次晴天的时候，你可以去公园散散步，放松一下心情。希望你有个美好的一天！您:exit再见！

—

最后

综上所述，Mem0 是一个强大的 AI 记忆层工具，能够为 AI 助手和智能体提供高效的记忆管理功能，可以根据用户交互不断改进，可以显著提升 AI 应用的交互体验和智能化水平，让 AI 更加懂你。