Adding graph memory
Excellent! Your chatbot can use Wikipedia to answer questions based on the the user's query. Now you can take things a step further by allowing your users to ask follow up questions! To do this, you'll first need to enable memory within your chatbot. Memory will allow your chatbot access to the full context of the conversation so it can handle follow up conversations.
This exercise is part of the course
Designing Agentic Systems with LangChain
Exercise instructions
- Start by importing
MemorySaver
from thecheckpoint.memory
module in LangGraph. - Create an instance of
MemorySaver()
asmemory
. - Compile the chatbot graph again by setting the
checkpointer
argument tomemory
in the.compile()
method applied tograph_builder
.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Import the modules for saving memory
from langgraph.____.____ import ____
# Modify the graph with memory checkpointing
memory = ____()
# Compile the graph passing in memory
graph = ____.___(checkpointer=____)