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

概述

在 LangGraph 中,Pregel 将 actorschannels 组合到单个应用中。Actors 从 channels 读取数据并向 channels 写入数据。Pregel 按照 Pregel 算法/批量同步并行 模型将应用的执行组织为多个步骤。 每个步骤包含三个阶段:
  • 计划:确定此步骤中要执行哪些 actors。例如,在第一步中,选择订阅特殊 input channels 的 actors;在后续步骤中,选择订阅上一步中更新的 channels 的 actors
  • 执行:并行执行所有选中的 actors,直到全部完成、某个失败或达到超时。在此阶段,channel 更新对 actors 不可见,直到下一步。
  • 更新:使用此步骤中 actors 写入的值更新 channels。
重复以上过程,直到没有 actors 被选择执行,或达到最大步骤数。

Actors

actor 是一个 PregelNode。它订阅 channels,从中读取数据,并向其写入数据。可以将其视为 Pregel 算法中的 actorPregelNodes 实现了 LangChain 的 Runnable 接口。

Channels

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

LastValue

LastValue 是默认的 channel 类型。它存储写入的最后一个值,覆盖任何先前的值。用于输入和输出值,或在步骤之间传递数据。

Topic

Topic 是一个可配置的发布/订阅 channel,适用于在 actors 之间发送多个值或跨步骤累积输出。可以配置为去重值或累积运行期间写入的所有值。

BinaryOperatorAggregate

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

示例

虽然大多数用户会通过 StateGraph API 或 entrypoint 装饰器与 Pregel 交互,但也可以直接与 Pregel 交互。 以下是几个不同的示例,让您了解 Pregel API。

高级 API

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