【AI】Langchain框架学习03

消息存储，使用redis（用docker运行redis）保存聊天记录

• 安装redis相关依赖包

pip install -q langchain-redis redis

• 构建问答

from langchain redis import RedischatMessageHistory

#这也是BasechatMessageHistory的子类。本地需启动Redis服务
history = RedisChatMessageHistory(session_id="test", redis_url="redis://localhost:6379/0")
#第一轮聊天!
# history.add_user_message("你是谁?")
# aimessage = llm.invoke(history.messages)
# print(aimessage.content)
# history.add_message(aimessage)
#第二轮聊天
history.add_user_message("请重复一次")
aimessage2 = llm.invoke(history.messages)
print(aimessage2.content)
history.add_message(aimessage2)

# 结果就是大模型会根据redis存储的历史记录回答问题