使用Memory实现一个带记忆的对话机器人

上一个例子我们使用的是通过自定义一个列表来存储对话的方式来保存历史的。

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

  1. from langchain.memory import ChatMessageHistory
  2. from langchain.chat_models import ChatOpenAI
  4. chat = ChatOpenAI(temperature=0)
  6. # 初始化 MessageHistory 对象
  7. history = ChatMessageHistory()
  9. # 给 MessageHistory 对象添加对话内容
  10. history.add_ai_message("你好！")
  11. history.add_user_message("中国的首都是哪里？")
  13. # 执行对话
  14. ai_response = chat(history.messages)
  15. print(ai_response)