Skip to main content
LangGraph 提供了 时间旅行 功能,可以从先前的检查点恢复执行,既可以重放相同的状态,也可以修改它以探索替代方案。在所有情况下,恢复过去的执行都会在历史中产生一个新的分支。 要通过 LangSmith 部署 API(通过 LangGraph SDK)进行时间旅行:
  1. 运行图:使用 LangGraph SDKclient.runs.waitclient.runs.stream API 运行图并传入初始输入。
  2. 在现有线程中识别检查点:使用 client.threads.get_history 方法检索特定 thread_id 的执行历史,并定位所需的 checkpoint_id。 或者,在希望执行暂停的节点之前设置一个 断点。然后,您可以找到记录到该断点的最近检查点。
  3. (可选)修改图状态:使用 client.threads.update_state 方法修改图在检查点处的状态,并从替代状态恢复执行。
  4. 从检查点恢复执行:使用 client.runs.waitclient.runs.stream API,输入为 None,并指定相应的 thread_idcheckpoint_id

在工作流中使用时间旅行

from typing_extensions import TypedDict, NotRequired
from langgraph.graph import StateGraph, START, END
from langchain.chat_models import init_chat_model
from langgraph.checkpoint.memory import InMemorySaver

class State(TypedDict):
    topic: NotRequired[str]
    joke: NotRequired[str]

model = init_chat_model(
    "claude-sonnet-4-6",
    temperature=0,
)

def generate_topic(state: State):
    """LLM调用,为笑话生成一个主题"""
    msg = model.invoke("给我一个有趣的笑话主题")
    return {"topic": msg.content}

def write_joke(state: State):
    """LLM调用,根据主题编写笑话"""
    msg = model.invoke(f"写一个关于 {state['topic']} 的短笑话")
    return {"joke": msg.content}

# 构建工作流
builder = StateGraph(State)

# 添加节点
builder.add_node("generate_topic", generate_topic)
builder.add_node("write_joke", write_joke)

# 添加边以连接节点
builder.add_edge(START, "generate_topic")
builder.add_edge("generate_topic", "write_joke")

# 编译
graph = builder.compile()

1. 运行图

from langgraph_sdk import get_client
client = get_client(url=<DEPLOYMENT_URL>)

# 使用部署时命名为 "agent" 的图
assistant_id = "agent"

# 创建一个线程
thread = await client.threads.create()
thread_id = thread["thread_id"]

# 运行图
result = await client.runs.wait(
    thread_id,
    assistant_id,
    input={}
)

2. 识别检查点

# 状态按时间倒序返回。
states = await client.threads.get_history(thread_id)
selected_state = states[1]
print(selected_state)

3. 更新状态

update_state 将创建一个新的检查点。新的检查点将与同一个线程关联,但会有一个新的检查点 ID。
new_config = await client.threads.update_state(
    thread_id,
    {"topic": "chickens"},
    checkpoint_id=selected_state["checkpoint_id"]
)
print(new_config)

4. 从检查点恢复执行

await client.runs.wait(
    thread_id,
    assistant_id,
    input=None,
    checkpoint_id=new_config["checkpoint_id"]
)

了解更多