Get startedGet started for free

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

View Course

Exercise instructions

  • Start by importing MemorySaver from the checkpoint.memory module in LangGraph.
  • Create an instance of MemorySaver() as memory.
  • Compile the chatbot graph again by setting the checkpointer argument to memory in the .compile() method applied to graph_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=____)
Edit and Run Code