如何编辑图状态¶

人机交互 (HIL) 在 代理系统 中至关重要。手动更新图状态是一种常见的 HIL 交互模式，允许人类编辑操作（例如，调用哪个工具或如何调用）。

我们可以在 LangGraph 中使用 断点 来实现这一点：断点允许我们在特定步骤之前中断图执行。在这个断点处，我们可以手动更新图状态，然后从该位置恢复以继续执行。

设置¶

首先，我们需要安装所需的软件包。

在 [13]

%%capture --no-stderr
%pip install --quiet -U langgraph langchain_anthropic

%%capture --no-stderr %pip install --quiet -U langgraph langchain_anthropic

接下来，我们需要设置 Anthropic（我们将使用的 LLM）的 API 密钥。

在 [2]

import getpass
import os


def _set_env(var: str):
    if not os.environ.get(var):
        os.environ[var] = getpass.getpass(f"{var}: ")


_set_env("ANTHROPIC_API_KEY")

import getpass import os def _set_env(var: str): if not os.environ.get(var): os.environ[var] = getpass.getpass(f"{var}: ") _set_env("ANTHROPIC_API_KEY")

ANTHROPIC_API_KEY:  ········

为 LangGraph 开发设置 LangSmith

注册 LangSmith 以快速发现问题并提高 LangGraph 项目的性能。LangSmith 允许您使用跟踪数据来调试、测试和监控您使用 LangGraph 构建的 LLM 应用程序 - 阅读有关如何开始的更多信息 此处。

简单用法¶

让我们看看它的基本用法。

下面，我们做了三件事

1. 我们使用 interrupt_before 指定的步骤（节点）来指定 断点。

2. 我们设置了一个 检查点 来保存图在此节点之前的状态。

3. 我们使用 .update_state 来更新图的状态。

在 [1]

from typing_extensions import TypedDict
from langgraph.graph import StateGraph, START, END
from langgraph.checkpoint.memory import MemorySaver
from IPython.display import Image, display


class State(TypedDict):
    input: str


def step_1(state):
    print("---Step 1---")
    pass


def step_2(state):
    print("---Step 2---")
    pass


def step_3(state):
    print("---Step 3---")
    pass


builder = StateGraph(State)
builder.add_node("step_1", step_1)
builder.add_node("step_2", step_2)
builder.add_node("step_3", step_3)
builder.add_edge(START, "step_1")
builder.add_edge("step_1", "step_2")
builder.add_edge("step_2", "step_3")
builder.add_edge("step_3", END)

# Set up memory
memory = MemorySaver()

# Add
graph = builder.compile(checkpointer=memory, interrupt_before=["step_2"])

# View
display(Image(graph.get_graph().draw_mermaid_png()))

from typing_extensions import TypedDict from langgraph.graph import StateGraph, START, END from langgraph.checkpoint.memory import MemorySaver from IPython.display import Image, display class State(TypedDict): input: str def step_1(state): print("---Step 1---") pass def step_2(state): print("---Step 2---") pass def step_3(state): print("---Step 3---") pass builder = StateGraph(State) builder.add_node("step_1", step_1) builder.add_node("step_2", step_2) builder.add_node("step_3", step_3) builder.add_edge(START, "step_1") builder.add_edge("step_1", "step_2") builder.add_edge("step_2", "step_3") builder.add_edge("step_3", END) # 设置内存 memory = MemorySaver() # 添加 graph = builder.compile(checkpointer=memory, interrupt_before=["step_2"]) # 查看 display(Image(graph.get_graph().draw_mermaid_png()))