为图添加记忆功能
做得很好!您的聊天机器人已经可以使用 Wikipedia 根据用户的查询来回答问题。现在再进一步,让用户可以追问后续问题!为此,您需要先在聊天机器人中启用记忆功能。记忆功能会让聊天机器人访问完整的会话上下文,从而更好地处理连续对话与追问。
本练习是课程的一部分
使用 LangChain 设计 Agentic 系统
练习说明
- 先从 LangGraph 的
checkpoint.memory模块中导入MemorySaver。 - 创建一个
MemorySaver()实例并命名为memory。 - 重新编译聊天机器人的图:在对
graph_builder调用.compile()时,将checkpointer参数设置为memory。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Import the modules for saving memory
from langgraph.____.____ import ____
# Modify the graph with memory checkpointing
memory = ____()
# Compile the graph passing in memory
graph = ____.___(checkpointer=____)