LangChain入门(九)-使用Memory实现一个带记忆的对话机器人

目录

一、说明

二、案例

---

 

一、说明

在第五章中我们使用的是通过自定义一个列表来存储对话的方式来保存历史的。

当然，你也可以使用自带的 memory 对象来实现这一点。

二、案例

import os

from langchain.memory import ChatMessageHistory
from langchain.chat_models import ChatOpenAI

# openAI的Key
os.environ["OPENAI_API_KEY"] = '*************'
chat = ChatOpenAI(temperature=0)

# 初始化 MessageHistory 对象
history = ChatMessageHistory()

# 给 MessageHistory 对象添加对话内容
history.add_ai_message("你好！")
history.add_user_message("中国的首都是哪里？")
history.add_user_message("他的经纬度是多少？")

# 执行对话
ai_response = chat(history.messages)
print(ai_response)