如何创建步骤序列¶

前提条件

本指南假设您熟悉以下内容

如何定义和更新图状态

本指南演示了如何构建一个简单的步骤序列。我们将演示

如何构建顺序图
用于构建类似图的内置简写。

总结¶

要添加节点序列，我们使用 .add_node 和 .add_edge 方法来操作我们的图

from langgraph.graph import START, StateGraph

graph_builder = StateGraph(State)

# Add nodes
graph_builder.add_node(step_1)
graph_builder.add_node(step_2)
graph_builder.add_node(step_3)

# Add edges
graph_builder.add_edge(START, "step_1")
graph_builder.add_edge("step_1", "step_2")
graph_builder.add_edge("step_2", "step_3")

API 参考: START | StateGraph

我们也可以使用内置的简写 .add_sequence

graph_builder = StateGraph(State).add_sequence([step_1, step_2, step_3])
graph_builder.add_edge(START, "step_1")

为什么使用 LangGraph 将应用程序步骤拆分为序列？

LangGraph 使您可以轻松地为应用程序添加底层持久化层。这允许在节点执行之间检查状态，因此您的 LangGraph 节点可以控制

状态更新如何被检查点化
中断如何在人工参与工作流程中恢复
我们如何使用 LangGraph 的时间旅行功能“倒带”和分支执行

它们还决定了执行步骤如何被流式传输，以及如何使用 LangGraph Studio 可视化和调试您的应用程序。

设置¶

首先，让我们安装 langgraph

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

设置 LangSmith 以获得更好的调试体验

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

构建图¶

让我们演示一个简单的用法示例。我们将创建一个包含三个步骤的序列

在状态的键中填充一个值
更新相同的值
填充不同的值

定义状态¶

让我们首先定义我们的状态。这控制着图的模式，并且还可以指定如何应用更新。有关更多详细信息，请参阅本指南。

在我们的例子中，我们将只跟踪两个值

from typing_extensions import TypedDict


class State(TypedDict):
    value_1: str
    value_2: int

定义节点¶

我们的节点只是读取我们图的状态并对其进行更新的 Python 函数。此函数的第一个参数始终是状态

def step_1(state: State):
    return {"value_1": "a"}


def step_2(state: State):
    current_value_1 = state["value_1"]
    return {"value_1": f"{current_value_1} b"}


def step_3(state: State):
    return {"value_2": 10}

注意

请注意，在向状态发出更新时，每个节点只需指定其希望更新的键的值。

默认情况下，这将覆盖相应键的值。您还可以使用 reducers 来控制更新的处理方式 —— 例如，您可以将连续的更新附加到键上。有关更多详细信息，请参阅本指南。

定义图¶

我们使用 StateGraph 来定义一个在此状态下运行的图。

然后我们将使用 add_node 和 add_edge 来填充我们的图并定义其控制流。

from langgraph.graph import START, StateGraph

graph_builder = StateGraph(State)

# Add nodes
graph_builder.add_node(step_1)
graph_builder.add_node(step_2)
graph_builder.add_node(step_3)

# Add edges
graph_builder.add_edge(START, "step_1")
graph_builder.add_edge("step_1", "step_2")
graph_builder.add_edge("step_2", "step_3")

API 参考: START | StateGraph

指定自定义名称

您可以使用 .add_node 为节点指定自定义名称

graph_builder.add_node("my_node", step_1)

请注意

.add_edge 接受节点的名称，对于函数，默认名称为 node.__name__。
我们必须指定图的入口点。为此，我们添加一个带有 START 节点的边。
当没有更多节点要执行时，图会停止。

接下来，我们编译我们的图。这提供了一些对图结构的基本检查（例如，识别孤立节点）。如果我们通过检查点向我们的应用程序添加持久性，它也会在这里传入。

graph = graph_builder.compile()

LangGraph 提供了用于可视化您的图的内置实用程序。让我们检查一下我们的序列。有关可视化的详细信息，请参阅本指南。

from IPython.display import Image, display

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

用法¶

让我们继续进行一个简单的调用

graph.invoke({"value_1": "c"})

{'value_1': 'a b', 'value_2': 10}

请注意

我们通过为单个状态键提供一个值来启动调用。我们必须始终为至少一个键提供一个值。
我们传入的值被第一个节点覆盖了。
第二个节点更新了该值。
第三个节点填充了不同的值。

内置简写¶

前提条件

.add_sequence 需要 langgraph>=0.2.46

LangGraph 包含一个内置的简写 .add_sequence 以方便使用

graph_builder = StateGraph(State).add_sequence([step_1, step_2, step_3])
graph_builder.add_edge(START, "step_1")

graph = graph_builder.compile()

graph.invoke({"value_1": "c"})

{'value_1': 'a b', 'value_2': 10}