Pregel 实现了 LangGraph 的运行时,负责管理 LangGraph 应用程序的执行。 编译一个 StateGraph 或创建一个 entrypoint 会产生一个 Pregel 实例,该实例可以通过输入进行调用。 本指南将在较高层次上解释该运行时,并提供直接使用 Pregel 实现应用程序的说明。
注意: Pregel 运行时以 Google 的 Pregel 算法 命名,该算法描述了一种利用图进行大规模并行计算的高效方法。

概述

在 LangGraph 中,Pregel 将 参与者 和 通道 结合到一个单一的应用程序中。参与者 从通道读取数据并向通道写入数据。Pregel 将应用程序的执行组织成多个步骤,遵循 Pregel 算法 / 批量同步并行 模型。 每个步骤包含三个阶段:
  • 规划:确定在此步骤中要执行哪些 参与者。例如,在第一步中,选择订阅了特殊 输入 通道的 参与者;在后续步骤中,选择订阅了上一步中已更新通道的 参与者。
  • 执行:并行执行所有选定的 参与者,直到全部完成、其中一个失败或达到超时。在此阶段,通道更新对参与者是不可见的,直到下一步。
  • 更新:使用此步骤中 参与者 写入的值来更新通道。
重复此过程,直到没有 参与者 被选中执行,或者达到最大步骤数。

参与者

参与者 是一个 PregelNode。它订阅通道,从中读取数据,并向其写入数据。它可以被视为 Pregel 算法中的一个 参与者。PregelNode 实现了 LangChain 的 Runnable 接口。

通道

通道用于参与者(PregelNodes)之间的通信。每个通道都有一个值类型、一个更新类型和一个更新函数——该函数接收一系列更新并修改存储的值。通道可用于将数据从一个链发送到另一个链,或在未来的步骤中将数据从一个链发送到自身。

LastValue

LastValue 是默认的通道类型。它存储写入的最后一个值,覆盖任何之前的值。用于输入和输出值,或用于将数据从一个步骤传递到下一个步骤。

Topic

Topic 是一个可配置的 PubSub 通道,适用于在参与者之间发送多个值或跨步骤累积输出。它可以配置为对值进行去重,或累积运行期间写入的所有值。

BinaryOperatorAggregate

BinaryOperatorAggregate 存储一个持久值,该值通过将二元运算符应用于当前值和每次新的更新来更新。用于跨步骤计算运行聚合。

示例

虽然大多数用户将通过 StateGraph API 或 entrypoint 装饰器与 Pregel 交互,但也可以直接与 Pregel 交互。 下面是一些不同的示例,让你了解 Pregel API 的使用。

高级 API

LangGraph 提供了两个用于创建 Pregel 应用程序的高级 API:StateGraph (Graph API) 和 函数式 API。
StateGraph (Graph API) 是一个更高级别的抽象,简化了 Pregel 应用程序的创建。它允许你定义一个节点和边的图。当你编译该图时,StateGraph API 会自动为你创建 Pregel 应用程序。
编译后的 Pregel 实例将与一个节点和通道列表相关联。你可以通过打印来检查节点和通道。
你会看到类似这样的内容:
你应该会看到类似这样的内容