Pregel 实现了 LangGraph 的运行时,负责管理 LangGraph 应用程序的执行。
编译 StateGraph 或创建 @entrypoint 会生成一个 Pregel 实例,该实例可以使用输入进行调用。
本指南从高层次解释了运行时,并提供了使用 Pregel 直接实现应用程序的说明。
注意: Pregel 运行时以 Google 的 Pregel 算法 命名,该算法描述了一种使用图进行大规模并行计算的高效方法。
概述
在 LangGraph 中,Pregel 将 参与者 和 通道 组合成一个应用程序。参与者 从通道读取数据并向通道写入数据。Pregel 将应用程序的执行组织为多个步骤,遵循 Pregel 算法/批量同步并行 模型。 每个步骤包含三个阶段:- 计划:确定此步骤中要执行的参与者。例如,在第一步中,选择订阅特殊输入通道的参与者;在后续步骤中,选择订阅在上一步中更新的通道的参与者。
- 执行:并行执行所有选定的参与者,直到全部完成、一个失败或达到超时。在此阶段,通道更新对参与者不可见,直到下一步。
- 更新:使用此步骤中参与者写入的值更新通道。
参与者
参与者 是一个PregelNode。它订阅通道,从通道读取数据,并向通道写入数据。它可以被视为 Pregel 算法中的一个参与者。PregelNodes 实现了 LangChain 的 Runnable 接口。
通道
通道用于在参与者(PregelNodes)之间进行通信。每个通道具有值类型、更新类型和更新函数——更新函数接收一系列更新并修改存储的值。通道可用于将数据从一个链发送到另一个链,或在未来的步骤中将数据从链发送到自身。LangGraph 提供了许多内置通道:LastValue:默认通道,存储发送到通道的最后一个值,适用于输入和输出值,或用于将数据从一个步骤发送到下一步。Topic:可配置的 PubSub 主题,适用于在参与者之间发送多个值,或用于累积输出。可配置为去重值或在多个步骤中累积值。BinaryOperatorAggregate:存储持久值,通过将二元运算符应用于当前值和发送到通道的每个更新来更新,适用于在多个步骤中计算聚合;例如,total = BinaryOperatorAggregate(int, operator.add)
示例
虽然大多数用户将通过 StateGraph API 或@entrypoint 装饰器与 Pregel 交互,但也可以直接与 Pregel 交互。
以下是几个不同的示例,让您了解 Pregel API。
- 单节点
- 多个节点
- Topic
- BinaryOperatorAggregate
- 循环
高级 API
LangGraph 提供了两个用于创建 Pregel 应用程序的高级 API:StateGraph(图 API) 和 函数式 API。- StateGraph(图 API)
- 函数式 API
StateGraph(图 API) 是一个更高级的抽象,简化了 Pregel 应用程序的创建。它允许您定义节点和边的图。当您编译图时,StateGraph API 会自动为您创建 Pregel 应用程序。编译后的 Pregel 实例将与节点和通道列表相关联。您可以通过打印它们来检查节点和通道。您将看到类似以下内容:您应该看到类似以下内容

