LangGraph 通过 检查点支持时间旅行:
- 重放:从之前的检查点重试。
- 分叉:从之前的检查点分支出来,并修改状态以探索替代路径。
两者都是通过从之前的检查点恢复来工作的。检查点之前的节点不会重新执行(结果已经保存)。检查点之后的节点会重新执行,包括任何 LLM 调用、API 请求和中断(这些可能会产生不同的结果)。
使用之前检查点的配置调用图,即可从该点开始重放。
重放会重新执行节点——它不只是从缓存中读取。LLM 调用、API 请求和中断会再次触发,并且可能返回不同的结果。从最终检查点(没有 next 节点)重放不会产生任何操作。
使用 get_state_history 找到你想从中重放的检查点,然后使用该检查点的配置调用 invoke:
from langgraph.graph import StateGraph, START
from langgraph.checkpoint.memory import InMemorySaver
from typing_extensions import TypedDict, NotRequired
from langchain_core.utils.uuid import uuid7
class State(TypedDict):
topic: NotRequired[str]
joke: NotRequired[str]
def generate_topic(state: State):
return {"topic": "socks in the dryer"}
def write_joke(state: State):
return {"joke": f"Why do {state['topic']} disappear? They elope!"}
checkpointer = InMemorySaver()
graph = (
StateGraph(State)
.add_node("generate_topic", generate_topic)
.add_node("write_joke", write_joke)
.add_edge(START, "generate_topic")
.add_edge("generate_topic", "write_joke")
.compile(checkpointer=checkpointer)
)
# Step 1: Run the graph
config = {"configurable": {"thread_id": str(uuid7())}}
result = graph.invoke({}, config)
# Step 2: Find a checkpoint to replay from
history = list(graph.get_state_history(config))
# History is in reverse chronological order
for state in history:
print(f"next={state.next}, checkpoint_id={state.config['configurable']['checkpoint_id']}")
# Step 3: Replay from a specific checkpoint
# Find the checkpoint before write_joke
before_joke = next(s for s in history if s.next == ("write_joke",))
replay_result = graph.invoke(None, before_joke.config)
# write_joke re-executes (runs again), generate_topic does not
分叉会基于过去的检查点创建一个带有修改后状态的新分支。对之前的检查点调用 update_state 来创建分叉,然后使用 None 调用 invoke 继续执行。
update_state 不会回滚线程。它会创建一个从指定点分支出来的新检查点。原始执行历史会保持不变。
# Find checkpoint before write_joke
history = list(graph.get_state_history(config))
before_joke = next(s for s in history if s.next == ("write_joke",))
# Fork: update state to change the topic
fork_config = graph.update_state(
before_joke.config,
values={"topic": "chickens"},
)
# Resume from the fork — write_joke re-executes with the new topic
fork_result = graph.invoke(None, fork_config)
print(fork_result["joke"]) # A joke about chickens, not socks
从特定节点开始
当你调用 update_state 时,值会使用指定节点的写入器进行应用(包括 reducers)。检查点会记录该节点生成了此次更新,并且执行会从该节点的后继节点继续。
默认情况下,LangGraph 会根据检查点的版本历史推断 as_node。从特定检查点分叉时,这种推断几乎总是正确的。
在以下情况下需要显式指定 as_node:
- 并行分支:多个节点在同一步中更新了状态,而 LangGraph 无法确定哪个节点是最后更新的(
InvalidUpdateError)。
- 没有执行历史:在一个新线程上设置状态(常见于测试)。
- 跳过节点:将
as_node 设置为更靠后的节点,让图认为该节点已经运行过。
# graph: generate_topic -> write_joke
# Treat this update as if generate_topic produced it.
# Execution resumes at write_joke (the successor of generate_topic).
fork_config = graph.update_state(
before_joke.config,
values={"topic": "chickens"},
as_node="generate_topic",
)
如果你的图使用 interrupt 来实现人在回路中工作流,那么在时间旅行期间,中断总是会被重新触发。包含该中断的节点会重新执行,而 interrupt() 会暂停并等待新的 Command(resume=...)。
from langgraph.types import interrupt, Command
class State(TypedDict):
value: list[str]
def ask_human(state: State):
answer = interrupt("What is your name?")
return {"value": [f"Hello, {answer}!"]}
def final_step(state: State):
return {"value": ["Done"]}
graph = (
StateGraph(State)
.add_node("ask_human", ask_human)
.add_node("final_step", final_step)
.add_edge(START, "ask_human")
.add_edge("ask_human", "final_step")
.compile(checkpointer=InMemorySaver())
)
config = {"configurable": {"thread_id": "1"}}
# First run: hits interrupt
graph.invoke({"value": []}, config)
# Resume with answer
graph.invoke(Command(resume="Alice"), config)
# Replay from before ask_human
history = list(graph.get_state_history(config))
before_ask = [s for s in history if s.next == ("ask_human",)][-1]
replay_result = graph.invoke(None, before_ask.config)
# Pauses at interrupt — waiting for new Command(resume=...)
# Fork from before ask_human
fork_config = graph.update_state(before_ask.config, {"value": ["forked"]})
fork_result = graph.invoke(None, fork_config)
# Pauses at interrupt — waiting for new Command(resume=...)
# Resume the forked interrupt with a different answer
graph.invoke(Command(resume="Bob"), fork_config)
# Result: {"value": ["forked", "Hello, Bob!", "Done"]}
多个中断
如果你的图在多个位置收集输入(例如,多步骤表单),你可以从这些中断之间的位置进行分叉,从而修改后面的回答,而不必重新询问前面的问题。
def ask_name(state):
name = interrupt("What is your name?")
return {"value": [f"name:{name}"]}
def ask_age(state):
age = interrupt("How old are you?")
return {"value": [f"age:{age}"]}
# Graph: ask_name -> ask_age -> final
# After completing both interrupts:
# Fork from BETWEEN the two interrupts (after ask_name, before ask_age)
history = list(graph.get_state_history(config))
between = [s for s in history if s.next == ("ask_age",)][-1]
fork_config = graph.update_state(between.config, {"value": ["modified"]})
result = graph.invoke(None, fork_config)
# ask_name result preserved ("name:Alice")
# ask_age pauses at interrupt — waiting for new answer
使用子图进行时间旅行时,具体行为取决于子图是否有自己的检查点保存器。这决定了你可以从多细的检查点粒度开始时间旅行。
默认情况下,子图会继承父图的检查点保存器。父图会将整个子图视为一个单一超步——整个子图执行过程中,在父级别只有一个检查点。从子图之前的位置进行时间旅行时,会从头重新执行该子图。你无法时间旅行到默认子图中节点之间的某个位置——你只能从父级别进行时间旅行。# Subgraph without its own checkpointer (default)
subgraph = (
StateGraph(State)
.add_node("step_a", step_a) # Has interrupt()
.add_node("step_b", step_b) # Has interrupt()
.add_edge(START, "step_a")
.add_edge("step_a", "step_b")
.compile() # No checkpointer — inherits from parent
)
graph = (
StateGraph(State)
.add_node("subgraph_node", subgraph)
.add_edge(START, "subgraph_node")
.compile(checkpointer=InMemorySaver())
)
config = {"configurable": {"thread_id": "1"}}
# Complete both interrupts
graph.invoke({"value": []}, config) # Hits step_a interrupt
graph.invoke(Command(resume="Alice"), config) # Hits step_b interrupt
graph.invoke(Command(resume="30"), config) # Completes
# Time travel from before the subgraph
history = list(graph.get_state_history(config))
before_sub = [s for s in history if s.next == ("subgraph_node",)][-1]
fork_config = graph.update_state(before_sub.config, {"value": ["forked"]})
result = graph.invoke(None, fork_config)
# The entire subgraph re-executes from scratch
# You cannot time travel to a point between step_a and step_b
在子图上设置 checkpointer=True,为它提供自己的检查点历史。这会在子图内部的每一步创建检查点,让你可以从子图内部的特定位置开始时间旅行——例如,在两个中断之间。使用带有 subgraphs=True 的 get_state 来访问子图自己的检查点配置,然后从它进行分叉:# Subgraph with its own checkpointer
subgraph = (
StateGraph(State)
.add_node("step_a", step_a) # Has interrupt()
.add_node("step_b", step_b) # Has interrupt()
.add_edge(START, "step_a")
.add_edge("step_a", "step_b")
.compile(checkpointer=True) # Own checkpoint history
)
graph = (
StateGraph(State)
.add_node("subgraph_node", subgraph)
.add_edge(START, "subgraph_node")
.compile(checkpointer=InMemorySaver())
)
config = {"configurable": {"thread_id": "1"}}
# Run until step_a interrupt
graph.invoke({"value": []}, config)
# Resume step_a -> hits step_b interrupt
graph.invoke(Command(resume="Alice"), config)
# Get the subgraph's own checkpoint (between step_a and step_b)
parent_state = graph.get_state(config, subgraphs=True)
sub_config = parent_state.tasks[0].state.config
# Fork from the subgraph checkpoint
fork_config = graph.update_state(sub_config, {"value": ["forked"]})
result = graph.invoke(None, fork_config)
# step_b re-executes, step_a's result is preserved
更多关于配置子图检查点保存器的内容,请参阅子图持久化。