@langchain/langgraph 0.4.9 → 1.0.0-alpha.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +26 -0
- package/README.md +1 -2
- package/dist/_virtual/rolldown_runtime.cjs +25 -0
- package/dist/channels/any_value.cjs +44 -63
- package/dist/channels/any_value.cjs.map +1 -0
- package/dist/channels/any_value.d.cts +24 -0
- package/dist/channels/any_value.d.cts.map +1 -0
- package/dist/channels/any_value.d.ts +16 -11
- package/dist/channels/any_value.d.ts.map +1 -0
- package/dist/channels/any_value.js +42 -57
- package/dist/channels/any_value.js.map +1 -1
- package/dist/channels/base.cjs +93 -133
- package/dist/channels/base.cjs.map +1 -0
- package/dist/channels/base.d.cts +79 -0
- package/dist/channels/base.d.cts.map +1 -0
- package/dist/channels/base.d.ts +77 -73
- package/dist/channels/base.d.ts.map +1 -0
- package/dist/channels/base.js +90 -127
- package/dist/channels/base.js.map +1 -1
- package/dist/channels/binop.cjs +47 -77
- package/dist/channels/binop.cjs.map +1 -0
- package/dist/channels/binop.d.cts +22 -0
- package/dist/channels/binop.d.cts.map +1 -0
- package/dist/channels/binop.d.ts +17 -12
- package/dist/channels/binop.d.ts.map +1 -0
- package/dist/channels/binop.js +45 -71
- package/dist/channels/binop.js.map +1 -1
- package/dist/channels/dynamic_barrier_value.cjs +72 -208
- package/dist/channels/dynamic_barrier_value.cjs.map +1 -0
- package/dist/channels/dynamic_barrier_value.d.cts +42 -0
- package/dist/channels/dynamic_barrier_value.d.cts.map +1 -0
- package/dist/channels/dynamic_barrier_value.d.ts +20 -35
- package/dist/channels/dynamic_barrier_value.d.ts.map +1 -0
- package/dist/channels/dynamic_barrier_value.js +69 -200
- package/dist/channels/dynamic_barrier_value.js.map +1 -1
- package/dist/channels/ephemeral_value.cjs +44 -70
- package/dist/channels/ephemeral_value.cjs.map +1 -0
- package/dist/channels/ephemeral_value.d.cts +22 -0
- package/dist/channels/ephemeral_value.d.cts.map +1 -0
- package/dist/channels/ephemeral_value.d.ts +18 -12
- package/dist/channels/ephemeral_value.d.ts.map +1 -0
- package/dist/channels/ephemeral_value.js +43 -65
- package/dist/channels/ephemeral_value.js.map +1 -1
- package/dist/channels/index.cjs +21 -10
- package/dist/channels/index.d.cts +9 -0
- package/dist/channels/index.d.ts +9 -7
- package/dist/channels/index.js +10 -3
- package/dist/channels/last_value.cjs +90 -140
- package/dist/channels/last_value.cjs.map +1 -0
- package/dist/channels/last_value.d.cts +42 -0
- package/dist/channels/last_value.d.cts.map +1 -0
- package/dist/channels/last_value.d.ts +27 -23
- package/dist/channels/last_value.d.ts.map +1 -0
- package/dist/channels/last_value.js +87 -133
- package/dist/channels/last_value.js.map +1 -1
- package/dist/channels/named_barrier_value.cjs +114 -170
- package/dist/channels/named_barrier_value.cjs.map +1 -0
- package/dist/channels/named_barrier_value.d.cts +46 -0
- package/dist/channels/named_barrier_value.d.cts.map +1 -0
- package/dist/channels/named_barrier_value.d.ts +32 -29
- package/dist/channels/named_barrier_value.d.ts.map +1 -0
- package/dist/channels/named_barrier_value.js +111 -163
- package/dist/channels/named_barrier_value.js.map +1 -1
- package/dist/channels/topic.cjs +63 -96
- package/dist/channels/topic.cjs.map +1 -0
- package/dist/channels/topic.d.cts +32 -0
- package/dist/channels/topic.d.cts.map +1 -0
- package/dist/channels/topic.d.ts +28 -19
- package/dist/channels/topic.d.ts.map +1 -0
- package/dist/channels/topic.js +61 -90
- package/dist/channels/topic.js.map +1 -1
- package/dist/constants.cjs +376 -452
- package/dist/constants.cjs.map +1 -0
- package/dist/constants.d.cts +285 -0
- package/dist/constants.d.cts.map +1 -0
- package/dist/constants.d.ts +113 -148
- package/dist/constants.d.ts.map +1 -0
- package/dist/constants.js +336 -444
- package/dist/constants.js.map +1 -1
- package/dist/errors.cjs +151 -179
- package/dist/errors.cjs.map +1 -0
- package/dist/errors.d.cts +81 -0
- package/dist/errors.d.cts.map +1 -0
- package/dist/errors.d.ts +56 -46
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +140 -165
- package/dist/errors.js.map +1 -1
- package/dist/func/index.cjs +292 -290
- package/dist/func/index.cjs.map +1 -0
- package/dist/func/index.d.cts +293 -0
- package/dist/func/index.d.cts.map +1 -0
- package/dist/func/index.d.ts +86 -72
- package/dist/func/index.d.ts.map +1 -0
- package/dist/func/index.js +288 -284
- package/dist/func/index.js.map +1 -1
- package/dist/func/types.d.cts +64 -0
- package/dist/func/types.d.cts.map +1 -0
- package/dist/func/types.d.ts +24 -19
- package/dist/func/types.d.ts.map +1 -0
- package/dist/graph/annotation.cjs +96 -120
- package/dist/graph/annotation.cjs.map +1 -0
- package/dist/graph/annotation.d.cts +116 -0
- package/dist/graph/annotation.d.cts.map +1 -0
- package/dist/graph/annotation.d.ts +33 -34
- package/dist/graph/annotation.d.ts.map +1 -0
- package/dist/graph/annotation.js +91 -112
- package/dist/graph/annotation.js.map +1 -1
- package/dist/graph/graph.cjs +419 -723
- package/dist/graph/graph.cjs.map +1 -0
- package/dist/graph/graph.d.cts +133 -0
- package/dist/graph/graph.d.cts.map +1 -0
- package/dist/graph/graph.d.ts +124 -91
- package/dist/graph/graph.d.ts.map +1 -0
- package/dist/graph/graph.js +412 -713
- package/dist/graph/graph.js.map +1 -1
- package/dist/graph/index.cjs +4 -17
- package/dist/graph/index.js +6 -5
- package/dist/graph/message.cjs +59 -104
- package/dist/graph/message.cjs.map +1 -0
- package/dist/graph/message.d.cts +19 -0
- package/dist/graph/message.d.cts.map +1 -0
- package/dist/graph/message.d.ts +11 -10
- package/dist/graph/message.d.ts.map +1 -0
- package/dist/graph/message.js +56 -98
- package/dist/graph/message.js.map +1 -1
- package/dist/graph/messages_annotation.cjs +106 -106
- package/dist/graph/messages_annotation.cjs.map +1 -0
- package/dist/graph/messages_annotation.d.cts +111 -0
- package/dist/graph/messages_annotation.d.cts.map +1 -0
- package/dist/graph/messages_annotation.d.ts +18 -9
- package/dist/graph/messages_annotation.d.ts.map +1 -0
- package/dist/graph/messages_annotation.js +100 -100
- package/dist/graph/messages_annotation.js.map +1 -1
- package/dist/graph/state.cjs +475 -779
- package/dist/graph/state.cjs.map +1 -0
- package/dist/graph/state.d.cts +231 -0
- package/dist/graph/state.d.cts.map +1 -0
- package/dist/graph/state.d.ts +148 -138
- package/dist/graph/state.d.ts.map +1 -0
- package/dist/graph/state.js +469 -769
- package/dist/graph/state.js.map +1 -1
- package/dist/graph/zod/index.cjs +10 -21
- package/dist/graph/zod/index.d.cts +3 -0
- package/dist/graph/zod/index.d.ts +3 -3
- package/dist/graph/zod/index.js +4 -4
- package/dist/graph/zod/meta.cjs +142 -177
- package/dist/graph/zod/meta.cjs.map +1 -0
- package/dist/graph/zod/meta.d.cts +116 -0
- package/dist/graph/zod/meta.d.cts.map +1 -0
- package/dist/graph/zod/meta.d.ts +99 -97
- package/dist/graph/zod/meta.d.ts.map +1 -0
- package/dist/graph/zod/meta.js +136 -170
- package/dist/graph/zod/meta.js.map +1 -1
- package/dist/graph/zod/plugin.cjs +36 -39
- package/dist/graph/zod/plugin.cjs.map +1 -0
- package/dist/graph/zod/plugin.js +34 -35
- package/dist/graph/zod/plugin.js.map +1 -1
- package/dist/graph/zod/schema.cjs +82 -110
- package/dist/graph/zod/schema.cjs.map +1 -0
- package/dist/graph/zod/schema.d.cts +38 -0
- package/dist/graph/zod/schema.d.cts.map +1 -0
- package/dist/graph/zod/schema.d.ts +12 -6
- package/dist/graph/zod/schema.d.ts.map +1 -0
- package/dist/graph/zod/schema.js +77 -103
- package/dist/graph/zod/schema.js.map +1 -1
- package/dist/graph/zod/zod-registry.cjs +41 -47
- package/dist/graph/zod/zod-registry.cjs.map +1 -0
- package/dist/graph/zod/zod-registry.d.cts +51 -0
- package/dist/graph/zod/zod-registry.d.cts.map +1 -0
- package/dist/graph/zod/zod-registry.d.ts +34 -26
- package/dist/graph/zod/zod-registry.d.ts.map +1 -0
- package/dist/graph/zod/zod-registry.js +37 -41
- package/dist/graph/zod/zod-registry.js.map +1 -1
- package/dist/hash.cjs +205 -267
- package/dist/hash.cjs.map +1 -0
- package/dist/hash.js +205 -265
- package/dist/hash.js.map +1 -1
- package/dist/index.cjs +111 -33
- package/dist/index.cjs.map +1 -0
- package/dist/index.d.cts +26 -0
- package/dist/index.d.ts +26 -5
- package/dist/index.js +19 -7
- package/dist/index.js.map +1 -1
- package/dist/interrupt.cjs +79 -85
- package/dist/interrupt.cjs.map +1 -0
- package/dist/interrupt.d.cts +49 -0
- package/dist/interrupt.d.cts.map +1 -0
- package/dist/interrupt.d.ts +6 -1
- package/dist/interrupt.d.ts.map +1 -0
- package/dist/interrupt.js +76 -80
- package/dist/interrupt.js.map +1 -1
- package/dist/prebuilt/agentName.cjs +139 -172
- package/dist/prebuilt/agentName.cjs.map +1 -0
- package/dist/prebuilt/agentName.d.cts +42 -0
- package/dist/prebuilt/agentName.d.cts.map +1 -0
- package/dist/prebuilt/agentName.d.ts +13 -21
- package/dist/prebuilt/agentName.d.ts.map +1 -0
- package/dist/prebuilt/agentName.js +139 -168
- package/dist/prebuilt/agentName.js.map +1 -1
- package/dist/prebuilt/agent_executor.cjs +42 -80
- package/dist/prebuilt/agent_executor.cjs.map +1 -0
- package/dist/prebuilt/agent_executor.d.cts +57 -0
- package/dist/prebuilt/agent_executor.d.cts.map +1 -0
- package/dist/prebuilt/agent_executor.d.ts +47 -38
- package/dist/prebuilt/agent_executor.d.ts.map +1 -0
- package/dist/prebuilt/agent_executor.js +40 -75
- package/dist/prebuilt/agent_executor.js.map +1 -1
- package/dist/prebuilt/chat_agent_executor.cjs +66 -128
- package/dist/prebuilt/chat_agent_executor.cjs.map +1 -0
- package/dist/prebuilt/chat_agent_executor.d.cts +23 -0
- package/dist/prebuilt/chat_agent_executor.d.cts.map +1 -0
- package/dist/prebuilt/chat_agent_executor.d.ts +18 -10
- package/dist/prebuilt/chat_agent_executor.d.ts.map +1 -0
- package/dist/prebuilt/chat_agent_executor.js +63 -123
- package/dist/prebuilt/chat_agent_executor.js.map +1 -1
- package/dist/prebuilt/index.cjs +15 -18
- package/dist/prebuilt/index.d.cts +8 -0
- package/dist/prebuilt/index.d.ts +8 -8
- package/dist/prebuilt/index.js +8 -7
- package/dist/prebuilt/interrupt.d.cts +73 -0
- package/dist/prebuilt/interrupt.d.cts.map +1 -0
- package/dist/prebuilt/interrupt.d.ts +32 -15
- package/dist/prebuilt/interrupt.d.ts.map +1 -0
- package/dist/prebuilt/react_agent_executor.cjs +317 -473
- package/dist/prebuilt/react_agent_executor.cjs.map +1 -0
- package/dist/prebuilt/react_agent_executor.d.cts +229 -0
- package/dist/prebuilt/react_agent_executor.d.cts.map +1 -0
- package/dist/prebuilt/react_agent_executor.d.ts +171 -143
- package/dist/prebuilt/react_agent_executor.d.ts.map +1 -0
- package/dist/prebuilt/react_agent_executor.js +315 -465
- package/dist/prebuilt/react_agent_executor.js.map +1 -1
- package/dist/prebuilt/tool_executor.cjs +45 -67
- package/dist/prebuilt/tool_executor.cjs.map +1 -0
- package/dist/prebuilt/tool_executor.d.cts +42 -0
- package/dist/prebuilt/tool_executor.d.cts.map +1 -0
- package/dist/prebuilt/tool_executor.d.ts +30 -24
- package/dist/prebuilt/tool_executor.d.ts.map +1 -0
- package/dist/prebuilt/tool_executor.js +44 -63
- package/dist/prebuilt/tool_executor.js.map +1 -1
- package/dist/prebuilt/tool_node.cjs +213 -275
- package/dist/prebuilt/tool_node.cjs.map +1 -0
- package/dist/prebuilt/tool_node.d.cts +151 -0
- package/dist/prebuilt/tool_node.d.cts.map +1 -0
- package/dist/prebuilt/tool_node.d.ts +30 -17
- package/dist/prebuilt/tool_node.d.ts.map +1 -0
- package/dist/prebuilt/tool_node.js +209 -268
- package/dist/prebuilt/tool_node.js.map +1 -1
- package/dist/pregel/algo.cjs +551 -692
- package/dist/pregel/algo.cjs.map +1 -0
- package/dist/pregel/algo.d.cts +13 -0
- package/dist/pregel/algo.d.cts.map +1 -0
- package/dist/pregel/algo.d.ts +8 -38
- package/dist/pregel/algo.d.ts.map +1 -0
- package/dist/pregel/algo.js +546 -683
- package/dist/pregel/algo.js.map +1 -1
- package/dist/pregel/call.cjs +46 -42
- package/dist/pregel/call.cjs.map +1 -0
- package/dist/pregel/call.js +43 -37
- package/dist/pregel/call.js.map +1 -1
- package/dist/pregel/debug.cjs +153 -223
- package/dist/pregel/debug.cjs.map +1 -0
- package/dist/pregel/debug.js +152 -215
- package/dist/pregel/debug.js.map +1 -1
- package/dist/pregel/index.cjs +1138 -1604
- package/dist/pregel/index.cjs.map +1 -0
- package/dist/pregel/index.d.cts +536 -0
- package/dist/pregel/index.d.cts.map +1 -0
- package/dist/pregel/index.d.ts +460 -422
- package/dist/pregel/index.d.ts.map +1 -0
- package/dist/pregel/index.js +1129 -1592
- package/dist/pregel/index.js.map +1 -1
- package/dist/pregel/io.cjs +127 -234
- package/dist/pregel/io.cjs.map +1 -0
- package/dist/pregel/io.js +122 -225
- package/dist/pregel/io.js.map +1 -1
- package/dist/pregel/loop.cjs +542 -954
- package/dist/pregel/loop.cjs.map +1 -0
- package/dist/pregel/loop.js +539 -948
- package/dist/pregel/loop.js.map +1 -1
- package/dist/pregel/messages.cjs +104 -196
- package/dist/pregel/messages.cjs.map +1 -0
- package/dist/pregel/messages.js +102 -191
- package/dist/pregel/messages.js.map +1 -1
- package/dist/pregel/read.cjs +150 -280
- package/dist/pregel/read.cjs.map +1 -0
- package/dist/pregel/read.d.cts +51 -0
- package/dist/pregel/read.d.cts.map +1 -0
- package/dist/pregel/read.d.ts +48 -46
- package/dist/pregel/read.d.ts.map +1 -0
- package/dist/pregel/read.js +147 -274
- package/dist/pregel/read.js.map +1 -1
- package/dist/pregel/remote.cjs +339 -458
- package/dist/pregel/remote.cjs.map +1 -0
- package/dist/pregel/remote.d.cts +121 -0
- package/dist/pregel/remote.d.cts.map +1 -0
- package/dist/pregel/remote.d.ts +79 -69
- package/dist/pregel/remote.d.ts.map +1 -0
- package/dist/pregel/remote.js +337 -453
- package/dist/pregel/remote.js.map +1 -1
- package/dist/pregel/retry.cjs +87 -138
- package/dist/pregel/retry.cjs.map +1 -0
- package/dist/pregel/retry.js +83 -130
- package/dist/pregel/retry.js.map +1 -1
- package/dist/pregel/runnable_types.d.cts +49 -0
- package/dist/pregel/runnable_types.d.cts.map +1 -0
- package/dist/pregel/runnable_types.d.ts +47 -17
- package/dist/pregel/runnable_types.d.ts.map +1 -0
- package/dist/pregel/runner.cjs +222 -315
- package/dist/pregel/runner.cjs.map +1 -0
- package/dist/pregel/runner.js +219 -308
- package/dist/pregel/runner.js.map +1 -1
- package/dist/pregel/stream.cjs +89 -130
- package/dist/pregel/stream.cjs.map +1 -0
- package/dist/pregel/stream.js +87 -125
- package/dist/pregel/stream.js.map +1 -1
- package/dist/pregel/types.cjs +25 -62
- package/dist/pregel/types.cjs.map +1 -0
- package/dist/pregel/types.d.cts +440 -0
- package/dist/pregel/types.d.cts.map +1 -0
- package/dist/pregel/types.d.ts +381 -432
- package/dist/pregel/types.d.ts.map +1 -0
- package/dist/pregel/types.js +23 -57
- package/dist/pregel/types.js.map +1 -1
- package/dist/pregel/utils/config.cjs +101 -147
- package/dist/pregel/utils/config.cjs.map +1 -0
- package/dist/pregel/utils/config.d.cts +36 -0
- package/dist/pregel/utils/config.d.cts.map +1 -0
- package/dist/pregel/utils/config.d.ts +12 -9
- package/dist/pregel/utils/config.d.ts.map +1 -0
- package/dist/pregel/utils/config.js +99 -143
- package/dist/pregel/utils/config.js.map +1 -1
- package/dist/pregel/utils/index.cjs +87 -125
- package/dist/pregel/utils/index.cjs.map +1 -0
- package/dist/pregel/utils/index.d.cts +51 -0
- package/dist/pregel/utils/index.d.cts.map +1 -0
- package/dist/pregel/utils/index.d.ts +45 -61
- package/dist/pregel/utils/index.d.ts.map +1 -0
- package/dist/pregel/utils/index.js +86 -120
- package/dist/pregel/utils/index.js.map +1 -1
- package/dist/pregel/utils/subgraph.cjs +15 -26
- package/dist/pregel/utils/subgraph.cjs.map +1 -0
- package/dist/pregel/utils/subgraph.js +12 -21
- package/dist/pregel/utils/subgraph.js.map +1 -1
- package/dist/pregel/validate.cjs +42 -92
- package/dist/pregel/validate.cjs.map +1 -0
- package/dist/pregel/validate.js +39 -84
- package/dist/pregel/validate.js.map +1 -1
- package/dist/pregel/write.cjs +87 -137
- package/dist/pregel/write.cjs.map +1 -0
- package/dist/pregel/write.d.cts +35 -0
- package/dist/pregel/write.d.cts.map +1 -0
- package/dist/pregel/write.d.ts +27 -23
- package/dist/pregel/write.d.ts.map +1 -0
- package/dist/pregel/write.js +83 -131
- package/dist/pregel/write.js.map +1 -1
- package/dist/remote.cjs +3 -6
- package/dist/remote.d.cts +2 -0
- package/dist/remote.d.ts +2 -1
- package/dist/remote.js +3 -2
- package/dist/setup/async_local_storage.cjs +10 -7
- package/dist/setup/async_local_storage.cjs.map +1 -0
- package/dist/setup/async_local_storage.js +7 -2
- package/dist/setup/async_local_storage.js.map +1 -1
- package/dist/ui/index.cjs +4 -0
- package/dist/ui/index.d.cts +5 -0
- package/dist/ui/index.d.ts +5 -0
- package/dist/ui/index.js +3 -0
- package/dist/ui/stream.cjs +145 -0
- package/dist/ui/stream.cjs.map +1 -0
- package/dist/ui/stream.d.cts +25 -0
- package/dist/ui/stream.d.cts.map +1 -0
- package/dist/ui/stream.d.ts +25 -0
- package/dist/ui/stream.d.ts.map +1 -0
- package/dist/ui/stream.js +143 -0
- package/dist/ui/stream.js.map +1 -0
- package/dist/ui/types.infer.d.cts +53 -0
- package/dist/ui/types.infer.d.cts.map +1 -0
- package/dist/ui/types.infer.d.ts +53 -0
- package/dist/ui/types.infer.d.ts.map +1 -0
- package/dist/ui/types.message.d.cts +95 -0
- package/dist/ui/types.message.d.cts.map +1 -0
- package/dist/ui/types.message.d.ts +95 -0
- package/dist/ui/types.message.d.ts.map +1 -0
- package/dist/ui/types.schema.d.cts +228 -0
- package/dist/ui/types.schema.d.cts.map +1 -0
- package/dist/ui/types.schema.d.ts +228 -0
- package/dist/ui/types.schema.d.ts.map +1 -0
- package/dist/utils.cjs +77 -147
- package/dist/utils.cjs.map +1 -0
- package/dist/utils.d.cts +32 -0
- package/dist/utils.d.cts.map +1 -0
- package/dist/utils.d.ts +29 -25
- package/dist/utils.d.ts.map +1 -0
- package/dist/utils.js +73 -140
- package/dist/utils.js.map +1 -1
- package/dist/web.cjs +96 -55
- package/dist/web.d.cts +23 -0
- package/dist/web.d.ts +23 -14
- package/dist/web.js +15 -8
- package/dist/writer.cjs +15 -0
- package/dist/writer.cjs.map +1 -0
- package/dist/writer.d.cts +5 -0
- package/dist/writer.d.cts.map +1 -0
- package/dist/writer.d.ts +5 -0
- package/dist/writer.d.ts.map +1 -0
- package/dist/writer.js +14 -0
- package/dist/writer.js.map +1 -0
- package/package.json +90 -83
- package/dist/channels/index.js.map +0 -1
- package/dist/func/types.cjs +0 -15
- package/dist/func/types.js +0 -12
- package/dist/func/types.js.map +0 -1
- package/dist/graph/index.d.ts +0 -4
- package/dist/graph/index.js.map +0 -1
- package/dist/graph/message.test.cjs +0 -196
- package/dist/graph/message.test.d.ts +0 -1
- package/dist/graph/message.test.js +0 -194
- package/dist/graph/message.test.js.map +0 -1
- package/dist/graph/zod/index.js.map +0 -1
- package/dist/graph/zod/plugin.d.ts +0 -28
- package/dist/hash.d.ts +0 -2
- package/dist/prebuilt/index.js.map +0 -1
- package/dist/prebuilt/interrupt.cjs +0 -3
- package/dist/prebuilt/interrupt.js +0 -2
- package/dist/prebuilt/interrupt.js.map +0 -1
- package/dist/pregel/call.d.ts +0 -16
- package/dist/pregel/debug.d.ts +0 -41
- package/dist/pregel/debug.test.cjs +0 -258
- package/dist/pregel/debug.test.d.ts +0 -1
- package/dist/pregel/debug.test.js +0 -256
- package/dist/pregel/debug.test.js.map +0 -1
- package/dist/pregel/io.d.ts +0 -30
- package/dist/pregel/io.mapCommand.test.cjs +0 -150
- package/dist/pregel/io.mapCommand.test.d.ts +0 -1
- package/dist/pregel/io.mapCommand.test.js +0 -148
- package/dist/pregel/io.mapCommand.test.js.map +0 -1
- package/dist/pregel/loop.d.ts +0 -147
- package/dist/pregel/messages.d.ts +0 -30
- package/dist/pregel/messages.test.cjs +0 -369
- package/dist/pregel/messages.test.d.ts +0 -1
- package/dist/pregel/messages.test.js +0 -367
- package/dist/pregel/messages.test.js.map +0 -1
- package/dist/pregel/read.test.cjs +0 -194
- package/dist/pregel/read.test.d.ts +0 -1
- package/dist/pregel/read.test.js +0 -192
- package/dist/pregel/read.test.js.map +0 -1
- package/dist/pregel/retry.d.ts +0 -17
- package/dist/pregel/runnable_types.cjs +0 -3
- package/dist/pregel/runnable_types.js +0 -2
- package/dist/pregel/runnable_types.js.map +0 -1
- package/dist/pregel/runner.d.ts +0 -79
- package/dist/pregel/runner.test.cjs +0 -66
- package/dist/pregel/runner.test.d.ts +0 -1
- package/dist/pregel/runner.test.js +0 -64
- package/dist/pregel/runner.test.js.map +0 -1
- package/dist/pregel/stream.d.ts +0 -40
- package/dist/pregel/utils/config.test.cjs +0 -214
- package/dist/pregel/utils/config.test.d.ts +0 -1
- package/dist/pregel/utils/config.test.js +0 -212
- package/dist/pregel/utils/config.test.js.map +0 -1
- package/dist/pregel/utils/subgraph.d.ts +0 -4
- package/dist/pregel/utils/subgraph.test.cjs +0 -83
- package/dist/pregel/utils/subgraph.test.d.ts +0 -1
- package/dist/pregel/utils/subgraph.test.js +0 -81
- package/dist/pregel/utils/subgraph.test.js.map +0 -1
- package/dist/pregel/validate.d.ts +0 -16
- package/dist/pregel/validate.test.cjs +0 -220
- package/dist/pregel/validate.test.d.ts +0 -1
- package/dist/pregel/validate.test.js +0 -218
- package/dist/pregel/validate.test.js.map +0 -1
- package/dist/pregel/write.test.cjs +0 -181
- package/dist/pregel/write.test.d.ts +0 -1
- package/dist/pregel/write.test.js +0 -179
- package/dist/pregel/write.test.js.map +0 -1
- package/dist/remote.js.map +0 -1
- package/dist/setup/async_local_storage.d.ts +0 -1
- package/dist/web.js.map +0 -1
- package/index.cjs +0 -1
- package/index.d.cts +0 -1
- package/index.d.ts +0 -1
- package/index.js +0 -1
- package/prebuilt.cjs +0 -1
- package/prebuilt.d.cts +0 -1
- package/prebuilt.d.ts +0 -1
- package/prebuilt.js +0 -1
- package/pregel.cjs +0 -1
- package/pregel.d.cts +0 -1
- package/pregel.d.ts +0 -1
- package/pregel.js +0 -1
- package/remote.cjs +0 -1
- package/remote.d.cts +0 -1
- package/remote.d.ts +0 -1
- package/remote.js +0 -1
- package/web.cjs +0 -1
- package/web.d.cts +0 -1
- package/web.d.ts +0 -1
- package/web.js +0 -1
- package/zod/schema.cjs +0 -1
- package/zod/schema.d.cts +0 -1
- package/zod/schema.d.ts +0 -1
- package/zod/schema.js +0 -1
- package/zod.cjs +0 -1
- package/zod.d.cts +0 -1
- package/zod.d.ts +0 -1
- package/zod.js +0 -1
package/dist/pregel/loop.cjs
CHANGED
|
@@ -1,960 +1,548 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
const
|
|
5
|
-
const
|
|
6
|
-
const
|
|
7
|
-
const
|
|
8
|
-
const
|
|
9
|
-
const
|
|
10
|
-
const
|
|
11
|
-
const
|
|
12
|
-
const
|
|
13
|
-
|
|
14
|
-
|
|
1
|
+
const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
|
|
2
|
+
const require_errors = require('../errors.cjs');
|
|
3
|
+
const require_base = require('../channels/base.cjs');
|
|
4
|
+
const require_constants = require('../constants.cjs');
|
|
5
|
+
const require_utils = require('../utils.cjs');
|
|
6
|
+
const require_hash = require('../hash.cjs');
|
|
7
|
+
const require_io = require('./io.cjs');
|
|
8
|
+
const require_index = require('./utils/index.cjs');
|
|
9
|
+
const require_algo = require('./algo.cjs');
|
|
10
|
+
const require_debug = require('./debug.cjs');
|
|
11
|
+
const require_stream = require('./stream.cjs');
|
|
12
|
+
const __langchain_langgraph_checkpoint = require_rolldown_runtime.__toESM(require("@langchain/langgraph-checkpoint"));
|
|
13
|
+
|
|
14
|
+
//#region src/pregel/loop.ts
|
|
15
15
|
const INPUT_DONE = Symbol.for("INPUT_DONE");
|
|
16
16
|
const INPUT_RESUMING = Symbol.for("INPUT_RESUMING");
|
|
17
17
|
const DEFAULT_LOOP_LIMIT = 25;
|
|
18
18
|
function createDuplexStream(...streams) {
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
}
|
|
26
|
-
},
|
|
27
|
-
modes: new Set(streams.flatMap((s) => Array.from(s.modes))),
|
|
28
|
-
});
|
|
29
|
-
}
|
|
30
|
-
class AsyncBatchedCache extends langgraph_checkpoint_1.BaseCache {
|
|
31
|
-
constructor(cache) {
|
|
32
|
-
super();
|
|
33
|
-
Object.defineProperty(this, "cache", {
|
|
34
|
-
enumerable: true,
|
|
35
|
-
configurable: true,
|
|
36
|
-
writable: true,
|
|
37
|
-
value: void 0
|
|
38
|
-
});
|
|
39
|
-
Object.defineProperty(this, "queue", {
|
|
40
|
-
enumerable: true,
|
|
41
|
-
configurable: true,
|
|
42
|
-
writable: true,
|
|
43
|
-
value: Promise.resolve()
|
|
44
|
-
});
|
|
45
|
-
this.cache = cache;
|
|
46
|
-
}
|
|
47
|
-
async get(keys) {
|
|
48
|
-
return this.enqueueOperation("get", keys);
|
|
49
|
-
}
|
|
50
|
-
async set(pairs) {
|
|
51
|
-
return this.enqueueOperation("set", pairs);
|
|
52
|
-
}
|
|
53
|
-
async clear(namespaces) {
|
|
54
|
-
return this.enqueueOperation("clear", namespaces);
|
|
55
|
-
}
|
|
56
|
-
async stop() {
|
|
57
|
-
await this.queue;
|
|
58
|
-
}
|
|
59
|
-
enqueueOperation(type, ...args) {
|
|
60
|
-
const newPromise = this.queue.then(() => {
|
|
61
|
-
// @ts-expect-error Tuple type warning
|
|
62
|
-
return this.cache[type](...args);
|
|
63
|
-
});
|
|
64
|
-
this.queue = newPromise.then(() => void 0, () => void 0);
|
|
65
|
-
return newPromise;
|
|
66
|
-
}
|
|
67
|
-
}
|
|
68
|
-
class PregelLoop {
|
|
69
|
-
get isResuming() {
|
|
70
|
-
let hasChannelVersions = false;
|
|
71
|
-
if (constants_js_1.START in this.checkpoint.channel_versions) {
|
|
72
|
-
// For common channels, we can short-circuit the check
|
|
73
|
-
hasChannelVersions = true;
|
|
74
|
-
}
|
|
75
|
-
else {
|
|
76
|
-
for (const chan in this.checkpoint.channel_versions) {
|
|
77
|
-
if (Object.prototype.hasOwnProperty.call(this.checkpoint.channel_versions, chan)) {
|
|
78
|
-
hasChannelVersions = true;
|
|
79
|
-
break;
|
|
80
|
-
}
|
|
81
|
-
}
|
|
82
|
-
}
|
|
83
|
-
const configHasResumingFlag = this.config.configurable?.[constants_js_1.CONFIG_KEY_RESUMING] !== undefined;
|
|
84
|
-
const configIsResuming = configHasResumingFlag && this.config.configurable?.[constants_js_1.CONFIG_KEY_RESUMING];
|
|
85
|
-
const inputIsNullOrUndefined = this.input === null || this.input === undefined;
|
|
86
|
-
const inputIsCommandResuming = (0, constants_js_1.isCommand)(this.input) && this.input.resume != null;
|
|
87
|
-
const inputIsResuming = this.input === INPUT_RESUMING;
|
|
88
|
-
const runIdMatchesPrevious = !this.isNested &&
|
|
89
|
-
this.config.metadata?.run_id !== undefined &&
|
|
90
|
-
this.checkpointMetadata?.run_id !== undefined &&
|
|
91
|
-
this.config.metadata.run_id ===
|
|
92
|
-
this.checkpointMetadata?.run_id;
|
|
93
|
-
return (hasChannelVersions &&
|
|
94
|
-
(configIsResuming ||
|
|
95
|
-
inputIsNullOrUndefined ||
|
|
96
|
-
inputIsCommandResuming ||
|
|
97
|
-
inputIsResuming ||
|
|
98
|
-
runIdMatchesPrevious));
|
|
99
|
-
}
|
|
100
|
-
constructor(params) {
|
|
101
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
102
|
-
Object.defineProperty(this, "input", {
|
|
103
|
-
enumerable: true,
|
|
104
|
-
configurable: true,
|
|
105
|
-
writable: true,
|
|
106
|
-
value: void 0
|
|
107
|
-
});
|
|
108
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
109
|
-
Object.defineProperty(this, "output", {
|
|
110
|
-
enumerable: true,
|
|
111
|
-
configurable: true,
|
|
112
|
-
writable: true,
|
|
113
|
-
value: void 0
|
|
114
|
-
});
|
|
115
|
-
Object.defineProperty(this, "config", {
|
|
116
|
-
enumerable: true,
|
|
117
|
-
configurable: true,
|
|
118
|
-
writable: true,
|
|
119
|
-
value: void 0
|
|
120
|
-
});
|
|
121
|
-
Object.defineProperty(this, "checkpointer", {
|
|
122
|
-
enumerable: true,
|
|
123
|
-
configurable: true,
|
|
124
|
-
writable: true,
|
|
125
|
-
value: void 0
|
|
126
|
-
});
|
|
127
|
-
Object.defineProperty(this, "checkpointerGetNextVersion", {
|
|
128
|
-
enumerable: true,
|
|
129
|
-
configurable: true,
|
|
130
|
-
writable: true,
|
|
131
|
-
value: void 0
|
|
132
|
-
});
|
|
133
|
-
Object.defineProperty(this, "channels", {
|
|
134
|
-
enumerable: true,
|
|
135
|
-
configurable: true,
|
|
136
|
-
writable: true,
|
|
137
|
-
value: void 0
|
|
138
|
-
});
|
|
139
|
-
Object.defineProperty(this, "checkpoint", {
|
|
140
|
-
enumerable: true,
|
|
141
|
-
configurable: true,
|
|
142
|
-
writable: true,
|
|
143
|
-
value: void 0
|
|
144
|
-
});
|
|
145
|
-
Object.defineProperty(this, "checkpointIdSaved", {
|
|
146
|
-
enumerable: true,
|
|
147
|
-
configurable: true,
|
|
148
|
-
writable: true,
|
|
149
|
-
value: void 0
|
|
150
|
-
});
|
|
151
|
-
Object.defineProperty(this, "checkpointConfig", {
|
|
152
|
-
enumerable: true,
|
|
153
|
-
configurable: true,
|
|
154
|
-
writable: true,
|
|
155
|
-
value: void 0
|
|
156
|
-
});
|
|
157
|
-
Object.defineProperty(this, "checkpointMetadata", {
|
|
158
|
-
enumerable: true,
|
|
159
|
-
configurable: true,
|
|
160
|
-
writable: true,
|
|
161
|
-
value: void 0
|
|
162
|
-
});
|
|
163
|
-
Object.defineProperty(this, "checkpointNamespace", {
|
|
164
|
-
enumerable: true,
|
|
165
|
-
configurable: true,
|
|
166
|
-
writable: true,
|
|
167
|
-
value: void 0
|
|
168
|
-
});
|
|
169
|
-
Object.defineProperty(this, "checkpointPendingWrites", {
|
|
170
|
-
enumerable: true,
|
|
171
|
-
configurable: true,
|
|
172
|
-
writable: true,
|
|
173
|
-
value: []
|
|
174
|
-
});
|
|
175
|
-
Object.defineProperty(this, "checkpointPreviousVersions", {
|
|
176
|
-
enumerable: true,
|
|
177
|
-
configurable: true,
|
|
178
|
-
writable: true,
|
|
179
|
-
value: void 0
|
|
180
|
-
});
|
|
181
|
-
Object.defineProperty(this, "step", {
|
|
182
|
-
enumerable: true,
|
|
183
|
-
configurable: true,
|
|
184
|
-
writable: true,
|
|
185
|
-
value: void 0
|
|
186
|
-
});
|
|
187
|
-
Object.defineProperty(this, "stop", {
|
|
188
|
-
enumerable: true,
|
|
189
|
-
configurable: true,
|
|
190
|
-
writable: true,
|
|
191
|
-
value: void 0
|
|
192
|
-
});
|
|
193
|
-
Object.defineProperty(this, "durability", {
|
|
194
|
-
enumerable: true,
|
|
195
|
-
configurable: true,
|
|
196
|
-
writable: true,
|
|
197
|
-
value: void 0
|
|
198
|
-
});
|
|
199
|
-
Object.defineProperty(this, "outputKeys", {
|
|
200
|
-
enumerable: true,
|
|
201
|
-
configurable: true,
|
|
202
|
-
writable: true,
|
|
203
|
-
value: void 0
|
|
204
|
-
});
|
|
205
|
-
Object.defineProperty(this, "streamKeys", {
|
|
206
|
-
enumerable: true,
|
|
207
|
-
configurable: true,
|
|
208
|
-
writable: true,
|
|
209
|
-
value: void 0
|
|
210
|
-
});
|
|
211
|
-
Object.defineProperty(this, "nodes", {
|
|
212
|
-
enumerable: true,
|
|
213
|
-
configurable: true,
|
|
214
|
-
writable: true,
|
|
215
|
-
value: void 0
|
|
216
|
-
});
|
|
217
|
-
Object.defineProperty(this, "skipDoneTasks", {
|
|
218
|
-
enumerable: true,
|
|
219
|
-
configurable: true,
|
|
220
|
-
writable: true,
|
|
221
|
-
value: void 0
|
|
222
|
-
});
|
|
223
|
-
Object.defineProperty(this, "prevCheckpointConfig", {
|
|
224
|
-
enumerable: true,
|
|
225
|
-
configurable: true,
|
|
226
|
-
writable: true,
|
|
227
|
-
value: void 0
|
|
228
|
-
});
|
|
229
|
-
Object.defineProperty(this, "status", {
|
|
230
|
-
enumerable: true,
|
|
231
|
-
configurable: true,
|
|
232
|
-
writable: true,
|
|
233
|
-
value: "pending"
|
|
234
|
-
});
|
|
235
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
236
|
-
Object.defineProperty(this, "tasks", {
|
|
237
|
-
enumerable: true,
|
|
238
|
-
configurable: true,
|
|
239
|
-
writable: true,
|
|
240
|
-
value: {}
|
|
241
|
-
});
|
|
242
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
243
|
-
Object.defineProperty(this, "stream", {
|
|
244
|
-
enumerable: true,
|
|
245
|
-
configurable: true,
|
|
246
|
-
writable: true,
|
|
247
|
-
value: void 0
|
|
248
|
-
});
|
|
249
|
-
Object.defineProperty(this, "checkpointerPromises", {
|
|
250
|
-
enumerable: true,
|
|
251
|
-
configurable: true,
|
|
252
|
-
writable: true,
|
|
253
|
-
value: []
|
|
254
|
-
});
|
|
255
|
-
Object.defineProperty(this, "isNested", {
|
|
256
|
-
enumerable: true,
|
|
257
|
-
configurable: true,
|
|
258
|
-
writable: true,
|
|
259
|
-
value: void 0
|
|
260
|
-
});
|
|
261
|
-
Object.defineProperty(this, "_checkpointerChainedPromise", {
|
|
262
|
-
enumerable: true,
|
|
263
|
-
configurable: true,
|
|
264
|
-
writable: true,
|
|
265
|
-
value: Promise.resolve()
|
|
266
|
-
});
|
|
267
|
-
Object.defineProperty(this, "store", {
|
|
268
|
-
enumerable: true,
|
|
269
|
-
configurable: true,
|
|
270
|
-
writable: true,
|
|
271
|
-
value: void 0
|
|
272
|
-
});
|
|
273
|
-
Object.defineProperty(this, "cache", {
|
|
274
|
-
enumerable: true,
|
|
275
|
-
configurable: true,
|
|
276
|
-
writable: true,
|
|
277
|
-
value: void 0
|
|
278
|
-
});
|
|
279
|
-
Object.defineProperty(this, "manager", {
|
|
280
|
-
enumerable: true,
|
|
281
|
-
configurable: true,
|
|
282
|
-
writable: true,
|
|
283
|
-
value: void 0
|
|
284
|
-
});
|
|
285
|
-
Object.defineProperty(this, "interruptAfter", {
|
|
286
|
-
enumerable: true,
|
|
287
|
-
configurable: true,
|
|
288
|
-
writable: true,
|
|
289
|
-
value: void 0
|
|
290
|
-
});
|
|
291
|
-
Object.defineProperty(this, "interruptBefore", {
|
|
292
|
-
enumerable: true,
|
|
293
|
-
configurable: true,
|
|
294
|
-
writable: true,
|
|
295
|
-
value: void 0
|
|
296
|
-
});
|
|
297
|
-
Object.defineProperty(this, "toInterrupt", {
|
|
298
|
-
enumerable: true,
|
|
299
|
-
configurable: true,
|
|
300
|
-
writable: true,
|
|
301
|
-
value: []
|
|
302
|
-
});
|
|
303
|
-
Object.defineProperty(this, "debug", {
|
|
304
|
-
enumerable: true,
|
|
305
|
-
configurable: true,
|
|
306
|
-
writable: true,
|
|
307
|
-
value: false
|
|
308
|
-
});
|
|
309
|
-
Object.defineProperty(this, "triggerToNodes", {
|
|
310
|
-
enumerable: true,
|
|
311
|
-
configurable: true,
|
|
312
|
-
writable: true,
|
|
313
|
-
value: void 0
|
|
314
|
-
});
|
|
315
|
-
this.input = params.input;
|
|
316
|
-
this.checkpointer = params.checkpointer;
|
|
317
|
-
// TODO: if managed values no longer needs graph we can replace with
|
|
318
|
-
// managed_specs, channel_specs
|
|
319
|
-
if (this.checkpointer !== undefined) {
|
|
320
|
-
this.checkpointerGetNextVersion = this.checkpointer.getNextVersion.bind(this.checkpointer);
|
|
321
|
-
}
|
|
322
|
-
else {
|
|
323
|
-
this.checkpointerGetNextVersion = algo_js_1.increment;
|
|
324
|
-
}
|
|
325
|
-
this.checkpoint = params.checkpoint;
|
|
326
|
-
this.checkpointMetadata = params.checkpointMetadata;
|
|
327
|
-
this.checkpointPreviousVersions = params.checkpointPreviousVersions;
|
|
328
|
-
this.channels = params.channels;
|
|
329
|
-
this.checkpointPendingWrites = params.checkpointPendingWrites;
|
|
330
|
-
this.step = params.step;
|
|
331
|
-
this.stop = params.stop;
|
|
332
|
-
this.config = params.config;
|
|
333
|
-
this.checkpointConfig = params.checkpointConfig;
|
|
334
|
-
this.isNested = params.isNested;
|
|
335
|
-
this.manager = params.manager;
|
|
336
|
-
this.outputKeys = params.outputKeys;
|
|
337
|
-
this.streamKeys = params.streamKeys;
|
|
338
|
-
this.nodes = params.nodes;
|
|
339
|
-
this.skipDoneTasks = params.skipDoneTasks;
|
|
340
|
-
this.store = params.store;
|
|
341
|
-
this.cache = params.cache ? new AsyncBatchedCache(params.cache) : undefined;
|
|
342
|
-
this.stream = params.stream;
|
|
343
|
-
this.checkpointNamespace = params.checkpointNamespace;
|
|
344
|
-
this.prevCheckpointConfig = params.prevCheckpointConfig;
|
|
345
|
-
this.interruptAfter = params.interruptAfter;
|
|
346
|
-
this.interruptBefore = params.interruptBefore;
|
|
347
|
-
this.durability = params.durability;
|
|
348
|
-
this.debug = params.debug;
|
|
349
|
-
this.triggerToNodes = params.triggerToNodes;
|
|
350
|
-
}
|
|
351
|
-
static async initialize(params) {
|
|
352
|
-
let { config, stream } = params;
|
|
353
|
-
if (stream !== undefined &&
|
|
354
|
-
config.configurable?.[constants_js_1.CONFIG_KEY_STREAM] !== undefined) {
|
|
355
|
-
stream = createDuplexStream(stream, config.configurable[constants_js_1.CONFIG_KEY_STREAM]);
|
|
356
|
-
}
|
|
357
|
-
const skipDoneTasks = config.configurable
|
|
358
|
-
? !("checkpoint_id" in config.configurable)
|
|
359
|
-
: true;
|
|
360
|
-
const scratchpad = config.configurable?.[constants_js_1.CONFIG_KEY_SCRATCHPAD];
|
|
361
|
-
if (config.configurable && scratchpad) {
|
|
362
|
-
if (scratchpad.subgraphCounter > 0) {
|
|
363
|
-
config = (0, index_js_1.patchConfigurable)(config, {
|
|
364
|
-
[constants_js_1.CONFIG_KEY_CHECKPOINT_NS]: [
|
|
365
|
-
config.configurable[constants_js_1.CONFIG_KEY_CHECKPOINT_NS],
|
|
366
|
-
scratchpad.subgraphCounter.toString(),
|
|
367
|
-
].join(constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR),
|
|
368
|
-
});
|
|
369
|
-
}
|
|
370
|
-
scratchpad.subgraphCounter += 1;
|
|
371
|
-
}
|
|
372
|
-
const isNested = constants_js_1.CONFIG_KEY_READ in (config.configurable ?? {});
|
|
373
|
-
if (!isNested &&
|
|
374
|
-
config.configurable?.checkpoint_ns !== undefined &&
|
|
375
|
-
config.configurable?.checkpoint_ns !== "") {
|
|
376
|
-
config = (0, index_js_1.patchConfigurable)(config, {
|
|
377
|
-
checkpoint_ns: "",
|
|
378
|
-
checkpoint_id: undefined,
|
|
379
|
-
});
|
|
380
|
-
}
|
|
381
|
-
let checkpointConfig = config;
|
|
382
|
-
if (config.configurable?.[constants_js_1.CONFIG_KEY_CHECKPOINT_MAP] !== undefined &&
|
|
383
|
-
config.configurable?.[constants_js_1.CONFIG_KEY_CHECKPOINT_MAP]?.[config.configurable?.checkpoint_ns]) {
|
|
384
|
-
checkpointConfig = (0, index_js_1.patchConfigurable)(config, {
|
|
385
|
-
checkpoint_id: config.configurable[constants_js_1.CONFIG_KEY_CHECKPOINT_MAP][config.configurable?.checkpoint_ns],
|
|
386
|
-
});
|
|
387
|
-
}
|
|
388
|
-
const checkpointNamespace = config.configurable?.checkpoint_ns?.split(constants_js_1.CHECKPOINT_NAMESPACE_SEPARATOR) ?? [];
|
|
389
|
-
const saved = (await params.checkpointer?.getTuple(checkpointConfig)) ?? {
|
|
390
|
-
config,
|
|
391
|
-
checkpoint: (0, langgraph_checkpoint_1.emptyCheckpoint)(),
|
|
392
|
-
metadata: { source: "input", step: -2, parents: {} },
|
|
393
|
-
pendingWrites: [],
|
|
394
|
-
};
|
|
395
|
-
checkpointConfig = {
|
|
396
|
-
...config,
|
|
397
|
-
...saved.config,
|
|
398
|
-
configurable: {
|
|
399
|
-
checkpoint_ns: "",
|
|
400
|
-
...config.configurable,
|
|
401
|
-
...saved.config.configurable,
|
|
402
|
-
},
|
|
403
|
-
};
|
|
404
|
-
const prevCheckpointConfig = saved.parentConfig;
|
|
405
|
-
const checkpoint = (0, langgraph_checkpoint_1.copyCheckpoint)(saved.checkpoint);
|
|
406
|
-
const checkpointMetadata = { ...saved.metadata };
|
|
407
|
-
const checkpointPendingWrites = saved.pendingWrites ?? [];
|
|
408
|
-
const channels = (0, base_js_1.emptyChannels)(params.channelSpecs, checkpoint);
|
|
409
|
-
const step = (checkpointMetadata.step ?? 0) + 1;
|
|
410
|
-
const stop = step + (config.recursionLimit ?? DEFAULT_LOOP_LIMIT) + 1;
|
|
411
|
-
const checkpointPreviousVersions = { ...checkpoint.channel_versions };
|
|
412
|
-
const store = params.store
|
|
413
|
-
? new langgraph_checkpoint_1.AsyncBatchedStore(params.store)
|
|
414
|
-
: undefined;
|
|
415
|
-
if (store) {
|
|
416
|
-
// Start the store. This is a batch store, so it will run continuously
|
|
417
|
-
await store.start();
|
|
418
|
-
}
|
|
419
|
-
return new PregelLoop({
|
|
420
|
-
input: params.input,
|
|
421
|
-
config,
|
|
422
|
-
checkpointer: params.checkpointer,
|
|
423
|
-
checkpoint,
|
|
424
|
-
checkpointMetadata,
|
|
425
|
-
checkpointConfig,
|
|
426
|
-
prevCheckpointConfig,
|
|
427
|
-
checkpointNamespace,
|
|
428
|
-
channels,
|
|
429
|
-
isNested,
|
|
430
|
-
manager: params.manager,
|
|
431
|
-
skipDoneTasks,
|
|
432
|
-
step,
|
|
433
|
-
stop,
|
|
434
|
-
checkpointPreviousVersions,
|
|
435
|
-
checkpointPendingWrites,
|
|
436
|
-
outputKeys: params.outputKeys ?? [],
|
|
437
|
-
streamKeys: params.streamKeys ?? [],
|
|
438
|
-
nodes: params.nodes,
|
|
439
|
-
stream,
|
|
440
|
-
store,
|
|
441
|
-
cache: params.cache,
|
|
442
|
-
interruptAfter: params.interruptAfter,
|
|
443
|
-
interruptBefore: params.interruptBefore,
|
|
444
|
-
durability: params.durability,
|
|
445
|
-
debug: params.debug,
|
|
446
|
-
triggerToNodes: params.triggerToNodes,
|
|
447
|
-
});
|
|
448
|
-
}
|
|
449
|
-
_checkpointerPutAfterPrevious(input) {
|
|
450
|
-
this._checkpointerChainedPromise = this._checkpointerChainedPromise.then(() => {
|
|
451
|
-
return this.checkpointer?.put(input.config, input.checkpoint, input.metadata, input.newVersions);
|
|
452
|
-
});
|
|
453
|
-
this.checkpointerPromises.push(this._checkpointerChainedPromise);
|
|
454
|
-
}
|
|
455
|
-
/**
|
|
456
|
-
* Put writes for a task, to be read by the next tick.
|
|
457
|
-
* @param taskId
|
|
458
|
-
* @param writes
|
|
459
|
-
*/
|
|
460
|
-
putWrites(taskId, writes) {
|
|
461
|
-
let writesCopy = writes;
|
|
462
|
-
if (writesCopy.length === 0)
|
|
463
|
-
return;
|
|
464
|
-
// deduplicate writes to special channels, last write wins
|
|
465
|
-
if (writesCopy.every(([key]) => key in langgraph_checkpoint_1.WRITES_IDX_MAP)) {
|
|
466
|
-
writesCopy = Array.from(new Map(writesCopy.map((w) => [w[0], w])).values());
|
|
467
|
-
}
|
|
468
|
-
// remove existing writes for this task
|
|
469
|
-
this.checkpointPendingWrites = this.checkpointPendingWrites.filter((w) => w[0] !== taskId);
|
|
470
|
-
// save writes
|
|
471
|
-
for (const [c, v] of writesCopy) {
|
|
472
|
-
this.checkpointPendingWrites.push([taskId, c, v]);
|
|
473
|
-
}
|
|
474
|
-
const config = (0, index_js_1.patchConfigurable)(this.checkpointConfig, {
|
|
475
|
-
[constants_js_1.CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
|
|
476
|
-
[constants_js_1.CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id,
|
|
477
|
-
});
|
|
478
|
-
if (this.durability !== "exit" && this.checkpointer != null) {
|
|
479
|
-
this.checkpointerPromises.push(this.checkpointer.putWrites(config, writesCopy, taskId));
|
|
480
|
-
}
|
|
481
|
-
if (this.tasks) {
|
|
482
|
-
this._outputWrites(taskId, writesCopy);
|
|
483
|
-
}
|
|
484
|
-
if (!writes.length || !this.cache || !this.tasks) {
|
|
485
|
-
return;
|
|
486
|
-
}
|
|
487
|
-
// only cache tasks with a cache key
|
|
488
|
-
const task = this.tasks[taskId];
|
|
489
|
-
if (task == null || task.cache_key == null) {
|
|
490
|
-
return;
|
|
491
|
-
}
|
|
492
|
-
// only cache successful tasks
|
|
493
|
-
if (writes[0][0] === constants_js_1.ERROR || writes[0][0] === constants_js_1.INTERRUPT) {
|
|
494
|
-
return;
|
|
495
|
-
}
|
|
496
|
-
void this.cache.set([
|
|
497
|
-
{
|
|
498
|
-
key: [task.cache_key.ns, task.cache_key.key],
|
|
499
|
-
value: task.writes,
|
|
500
|
-
ttl: task.cache_key.ttl,
|
|
501
|
-
},
|
|
502
|
-
]);
|
|
503
|
-
}
|
|
504
|
-
_outputWrites(taskId, writes, cached = false) {
|
|
505
|
-
const task = this.tasks[taskId];
|
|
506
|
-
if (task !== undefined) {
|
|
507
|
-
if (task.config !== undefined &&
|
|
508
|
-
(task.config.tags ?? []).includes(constants_js_1.TAG_HIDDEN)) {
|
|
509
|
-
return;
|
|
510
|
-
}
|
|
511
|
-
if (writes.length > 0) {
|
|
512
|
-
if (writes[0][0] === constants_js_1.INTERRUPT) {
|
|
513
|
-
// in `algo.ts` we append a bool to the task path to indicate
|
|
514
|
-
// whether or not a call was present. If so, we don't emit the
|
|
515
|
-
// the interrupt as it'll be emitted by the parent.
|
|
516
|
-
if (task.path?.[0] === constants_js_1.PUSH && task.path?.at(-1) === true)
|
|
517
|
-
return;
|
|
518
|
-
const interruptWrites = writes
|
|
519
|
-
.filter((w) => w[0] === constants_js_1.INTERRUPT)
|
|
520
|
-
.flatMap((w) => w[1]);
|
|
521
|
-
this._emit([
|
|
522
|
-
["updates", { [constants_js_1.INTERRUPT]: interruptWrites }],
|
|
523
|
-
["values", { [constants_js_1.INTERRUPT]: interruptWrites }],
|
|
524
|
-
]);
|
|
525
|
-
}
|
|
526
|
-
else if (writes[0][0] !== constants_js_1.ERROR) {
|
|
527
|
-
this._emit((0, utils_js_1.gatherIteratorSync)((0, utils_js_1.prefixGenerator)((0, io_js_1.mapOutputUpdates)(this.outputKeys, [[task, writes]], cached), "updates")));
|
|
528
|
-
}
|
|
529
|
-
}
|
|
530
|
-
if (!cached) {
|
|
531
|
-
this._emit((0, utils_js_1.gatherIteratorSync)((0, utils_js_1.prefixGenerator)((0, debug_js_1.mapDebugTaskResults)([[task, writes]], this.streamKeys), "tasks")));
|
|
532
|
-
}
|
|
533
|
-
}
|
|
534
|
-
}
|
|
535
|
-
async _matchCachedWrites() {
|
|
536
|
-
if (!this.cache)
|
|
537
|
-
return [];
|
|
538
|
-
const matched = [];
|
|
539
|
-
const serializeKey = ([ns, key]) => {
|
|
540
|
-
return `ns:${ns.join(",")}|key:${key}`;
|
|
541
|
-
};
|
|
542
|
-
const keys = [];
|
|
543
|
-
const keyMap = {};
|
|
544
|
-
for (const task of Object.values(this.tasks)) {
|
|
545
|
-
if (task.cache_key != null && !task.writes.length) {
|
|
546
|
-
keys.push([task.cache_key.ns, task.cache_key.key]);
|
|
547
|
-
keyMap[serializeKey([task.cache_key.ns, task.cache_key.key])] = task;
|
|
548
|
-
}
|
|
549
|
-
}
|
|
550
|
-
if (keys.length === 0)
|
|
551
|
-
return [];
|
|
552
|
-
const cache = await this.cache.get(keys);
|
|
553
|
-
for (const { key, value } of cache) {
|
|
554
|
-
const task = keyMap[serializeKey(key)];
|
|
555
|
-
if (task != null) {
|
|
556
|
-
// update the task with the cached writes
|
|
557
|
-
task.writes.push(...value);
|
|
558
|
-
matched.push({ task, result: value });
|
|
559
|
-
}
|
|
560
|
-
}
|
|
561
|
-
return matched;
|
|
562
|
-
}
|
|
563
|
-
/**
|
|
564
|
-
* Execute a single iteration of the Pregel loop.
|
|
565
|
-
* Returns true if more iterations are needed.
|
|
566
|
-
* @param params
|
|
567
|
-
*/
|
|
568
|
-
async tick(params) {
|
|
569
|
-
if (this.store && !this.store.isRunning) {
|
|
570
|
-
await this.store?.start();
|
|
571
|
-
}
|
|
572
|
-
const { inputKeys = [] } = params;
|
|
573
|
-
if (this.status !== "pending") {
|
|
574
|
-
throw new Error(`Cannot tick when status is no longer "pending". Current status: "${this.status}"`);
|
|
575
|
-
}
|
|
576
|
-
if (![INPUT_DONE, INPUT_RESUMING].includes(this.input)) {
|
|
577
|
-
await this._first(inputKeys);
|
|
578
|
-
}
|
|
579
|
-
else if (this.toInterrupt.length > 0) {
|
|
580
|
-
this.status = "interrupt_before";
|
|
581
|
-
throw new errors_js_1.GraphInterrupt();
|
|
582
|
-
}
|
|
583
|
-
else if (Object.values(this.tasks).every((task) => task.writes.length > 0)) {
|
|
584
|
-
// finish superstep
|
|
585
|
-
const writes = Object.values(this.tasks).flatMap((t) => t.writes);
|
|
586
|
-
// All tasks have finished
|
|
587
|
-
(0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
|
|
588
|
-
// produce values output
|
|
589
|
-
const valuesOutput = await (0, utils_js_1.gatherIterator)((0, utils_js_1.prefixGenerator)((0, io_js_1.mapOutputValues)(this.outputKeys, writes, this.channels), "values"));
|
|
590
|
-
this._emit(valuesOutput);
|
|
591
|
-
// clear pending writes
|
|
592
|
-
this.checkpointPendingWrites = [];
|
|
593
|
-
await this._putCheckpoint({ source: "loop" });
|
|
594
|
-
// after execution, check if we should interrupt
|
|
595
|
-
if ((0, algo_js_1.shouldInterrupt)(this.checkpoint, this.interruptAfter, Object.values(this.tasks))) {
|
|
596
|
-
this.status = "interrupt_after";
|
|
597
|
-
throw new errors_js_1.GraphInterrupt();
|
|
598
|
-
}
|
|
599
|
-
// unset resuming flag
|
|
600
|
-
if (this.config.configurable?.[constants_js_1.CONFIG_KEY_RESUMING] !== undefined) {
|
|
601
|
-
delete this.config.configurable?.[constants_js_1.CONFIG_KEY_RESUMING];
|
|
602
|
-
}
|
|
603
|
-
}
|
|
604
|
-
else {
|
|
605
|
-
return false;
|
|
606
|
-
}
|
|
607
|
-
if (this.step > this.stop) {
|
|
608
|
-
this.status = "out_of_steps";
|
|
609
|
-
return false;
|
|
610
|
-
}
|
|
611
|
-
const nextTasks = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, {
|
|
612
|
-
step: this.step,
|
|
613
|
-
checkpointer: this.checkpointer,
|
|
614
|
-
isResuming: this.isResuming,
|
|
615
|
-
manager: this.manager,
|
|
616
|
-
store: this.store,
|
|
617
|
-
stream: this.stream,
|
|
618
|
-
});
|
|
619
|
-
this.tasks = nextTasks;
|
|
620
|
-
// Produce debug output
|
|
621
|
-
if (this.checkpointer) {
|
|
622
|
-
this._emit(await (0, utils_js_1.gatherIterator)((0, utils_js_1.prefixGenerator)((0, debug_js_1.mapDebugCheckpoint)(this.checkpointConfig, this.channels, this.streamKeys, this.checkpointMetadata, Object.values(this.tasks), this.checkpointPendingWrites, this.prevCheckpointConfig, this.outputKeys), "checkpoints")));
|
|
623
|
-
}
|
|
624
|
-
if (Object.values(this.tasks).length === 0) {
|
|
625
|
-
this.status = "done";
|
|
626
|
-
return false;
|
|
627
|
-
}
|
|
628
|
-
// if there are pending writes from a previous loop, apply them
|
|
629
|
-
if (this.skipDoneTasks && this.checkpointPendingWrites.length > 0) {
|
|
630
|
-
for (const [tid, k, v] of this.checkpointPendingWrites) {
|
|
631
|
-
if (k === constants_js_1.ERROR || k === constants_js_1.INTERRUPT || k === constants_js_1.RESUME) {
|
|
632
|
-
continue;
|
|
633
|
-
}
|
|
634
|
-
const task = Object.values(this.tasks).find((t) => t.id === tid);
|
|
635
|
-
if (task) {
|
|
636
|
-
task.writes.push([k, v]);
|
|
637
|
-
}
|
|
638
|
-
}
|
|
639
|
-
for (const task of Object.values(this.tasks)) {
|
|
640
|
-
if (task.writes.length > 0) {
|
|
641
|
-
this._outputWrites(task.id, task.writes, true);
|
|
642
|
-
}
|
|
643
|
-
}
|
|
644
|
-
}
|
|
645
|
-
// if all tasks have finished, re-tick
|
|
646
|
-
if (Object.values(this.tasks).every((task) => task.writes.length > 0)) {
|
|
647
|
-
return this.tick({ inputKeys });
|
|
648
|
-
}
|
|
649
|
-
// Before execution, check if we should interrupt
|
|
650
|
-
if ((0, algo_js_1.shouldInterrupt)(this.checkpoint, this.interruptBefore, Object.values(this.tasks))) {
|
|
651
|
-
this.status = "interrupt_before";
|
|
652
|
-
throw new errors_js_1.GraphInterrupt();
|
|
653
|
-
}
|
|
654
|
-
// Produce debug output
|
|
655
|
-
const debugOutput = await (0, utils_js_1.gatherIterator)((0, utils_js_1.prefixGenerator)((0, debug_js_1.mapDebugTasks)(Object.values(this.tasks)), "tasks"));
|
|
656
|
-
this._emit(debugOutput);
|
|
657
|
-
return true;
|
|
658
|
-
}
|
|
659
|
-
async finishAndHandleError(error) {
|
|
660
|
-
// persist current checkpoint and writes
|
|
661
|
-
if (this.durability === "exit" &&
|
|
662
|
-
// if it's a top graph
|
|
663
|
-
(!this.isNested ||
|
|
664
|
-
// or a nested graph with error or interrupt
|
|
665
|
-
typeof error !== "undefined" ||
|
|
666
|
-
// or a nested graph with checkpointer: true
|
|
667
|
-
this.checkpointNamespace.every((part) => !part.includes(constants_js_1.CHECKPOINT_NAMESPACE_END)))) {
|
|
668
|
-
this._putCheckpoint(this.checkpointMetadata);
|
|
669
|
-
this._flushPendingWrites();
|
|
670
|
-
}
|
|
671
|
-
const suppress = this._suppressInterrupt(error);
|
|
672
|
-
if (suppress || error === undefined) {
|
|
673
|
-
this.output = (0, io_js_1.readChannels)(this.channels, this.outputKeys);
|
|
674
|
-
}
|
|
675
|
-
if (suppress) {
|
|
676
|
-
// emit one last "values" event, with pending writes applied
|
|
677
|
-
if (this.tasks !== undefined &&
|
|
678
|
-
this.checkpointPendingWrites.length > 0 &&
|
|
679
|
-
Object.values(this.tasks).some((task) => task.writes.length > 0)) {
|
|
680
|
-
(0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
|
|
681
|
-
this._emit((0, utils_js_1.gatherIteratorSync)((0, utils_js_1.prefixGenerator)((0, io_js_1.mapOutputValues)(this.outputKeys, Object.values(this.tasks).flatMap((t) => t.writes), this.channels), "values")));
|
|
682
|
-
}
|
|
683
|
-
// Emit INTERRUPT event
|
|
684
|
-
if ((0, errors_js_1.isGraphInterrupt)(error) && !error.interrupts.length) {
|
|
685
|
-
this._emit([
|
|
686
|
-
["updates", { [constants_js_1.INTERRUPT]: [] }],
|
|
687
|
-
["values", { [constants_js_1.INTERRUPT]: [] }],
|
|
688
|
-
]);
|
|
689
|
-
}
|
|
690
|
-
}
|
|
691
|
-
return suppress;
|
|
692
|
-
}
|
|
693
|
-
async acceptPush(task, writeIdx, call) {
|
|
694
|
-
if (this.interruptAfter?.length > 0 &&
|
|
695
|
-
(0, algo_js_1.shouldInterrupt)(this.checkpoint, this.interruptAfter, [task])) {
|
|
696
|
-
this.toInterrupt.push(task);
|
|
697
|
-
return;
|
|
698
|
-
}
|
|
699
|
-
const pushed = (0, algo_js_1._prepareSingleTask)([constants_js_1.PUSH, task.path ?? [], writeIdx, task.id, call], this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, task.config ?? {}, true, {
|
|
700
|
-
step: this.step,
|
|
701
|
-
checkpointer: this.checkpointer,
|
|
702
|
-
manager: this.manager,
|
|
703
|
-
store: this.store,
|
|
704
|
-
stream: this.stream,
|
|
705
|
-
});
|
|
706
|
-
if (!pushed)
|
|
707
|
-
return;
|
|
708
|
-
if (this.interruptBefore?.length > 0 &&
|
|
709
|
-
(0, algo_js_1.shouldInterrupt)(this.checkpoint, this.interruptBefore, [pushed])) {
|
|
710
|
-
this.toInterrupt.push(pushed);
|
|
711
|
-
return;
|
|
712
|
-
}
|
|
713
|
-
this._emit((0, utils_js_1.gatherIteratorSync)((0, utils_js_1.prefixGenerator)((0, debug_js_1.mapDebugTasks)([pushed]), "tasks")));
|
|
714
|
-
if (this.debug)
|
|
715
|
-
(0, debug_js_1.printStepTasks)(this.step, [pushed]);
|
|
716
|
-
this.tasks[pushed.id] = pushed;
|
|
717
|
-
if (this.skipDoneTasks)
|
|
718
|
-
this._matchWrites({ [pushed.id]: pushed });
|
|
719
|
-
const tasks = await this._matchCachedWrites();
|
|
720
|
-
for (const { task } of tasks) {
|
|
721
|
-
this._outputWrites(task.id, task.writes, true);
|
|
722
|
-
}
|
|
723
|
-
return pushed;
|
|
724
|
-
}
|
|
725
|
-
_suppressInterrupt(e) {
|
|
726
|
-
return (0, errors_js_1.isGraphInterrupt)(e) && !this.isNested;
|
|
727
|
-
}
|
|
728
|
-
async _first(inputKeys) {
|
|
729
|
-
/*
|
|
730
|
-
* Resuming from previous checkpoint requires
|
|
731
|
-
* - finding a previous checkpoint
|
|
732
|
-
* - receiving null input (outer graph) or RESUMING flag (subgraph)
|
|
733
|
-
*/
|
|
734
|
-
const { configurable } = this.config;
|
|
735
|
-
// take resume value from parent
|
|
736
|
-
const scratchpad = configurable?.[constants_js_1.CONFIG_KEY_SCRATCHPAD];
|
|
737
|
-
if (scratchpad && scratchpad.nullResume !== undefined) {
|
|
738
|
-
this.putWrites(constants_js_1.NULL_TASK_ID, [[constants_js_1.RESUME, scratchpad.nullResume]]);
|
|
739
|
-
}
|
|
740
|
-
// map command to writes
|
|
741
|
-
if ((0, constants_js_1.isCommand)(this.input)) {
|
|
742
|
-
const hasResume = this.input.resume != null;
|
|
743
|
-
if (this.input.resume != null &&
|
|
744
|
-
typeof this.input.resume === "object" &&
|
|
745
|
-
Object.keys(this.input.resume).every(hash_js_1.isXXH3)) {
|
|
746
|
-
this.config.configurable ??= {};
|
|
747
|
-
this.config.configurable[constants_js_1.CONFIG_KEY_RESUME_MAP] = this.input.resume;
|
|
748
|
-
}
|
|
749
|
-
if (hasResume && this.checkpointer == null) {
|
|
750
|
-
throw new Error("Cannot use Command(resume=...) without checkpointer");
|
|
751
|
-
}
|
|
752
|
-
const writes = {};
|
|
753
|
-
// group writes by task id
|
|
754
|
-
for (const [tid, key, value] of (0, io_js_1.mapCommand)(this.input, this.checkpointPendingWrites)) {
|
|
755
|
-
writes[tid] ??= [];
|
|
756
|
-
writes[tid].push([key, value]);
|
|
757
|
-
}
|
|
758
|
-
if (Object.keys(writes).length === 0) {
|
|
759
|
-
throw new errors_js_1.EmptyInputError("Received empty Command input");
|
|
760
|
-
}
|
|
761
|
-
// save writes
|
|
762
|
-
for (const [tid, ws] of Object.entries(writes)) {
|
|
763
|
-
this.putWrites(tid, ws);
|
|
764
|
-
}
|
|
765
|
-
}
|
|
766
|
-
// apply null writes
|
|
767
|
-
const nullWrites = (this.checkpointPendingWrites ?? [])
|
|
768
|
-
.filter((w) => w[0] === constants_js_1.NULL_TASK_ID)
|
|
769
|
-
.map((w) => w.slice(1));
|
|
770
|
-
if (nullWrites.length > 0) {
|
|
771
|
-
(0, algo_js_1._applyWrites)(this.checkpoint, this.channels, [
|
|
772
|
-
{
|
|
773
|
-
name: constants_js_1.INPUT,
|
|
774
|
-
writes: nullWrites,
|
|
775
|
-
triggers: [],
|
|
776
|
-
},
|
|
777
|
-
], this.checkpointerGetNextVersion, this.triggerToNodes);
|
|
778
|
-
}
|
|
779
|
-
const isCommandUpdateOrGoto = (0, constants_js_1.isCommand)(this.input) && nullWrites.length > 0;
|
|
780
|
-
if (this.isResuming || isCommandUpdateOrGoto) {
|
|
781
|
-
for (const channelName in this.channels) {
|
|
782
|
-
if (!Object.prototype.hasOwnProperty.call(this.channels, channelName))
|
|
783
|
-
continue;
|
|
784
|
-
if (this.checkpoint.channel_versions[channelName] !== undefined) {
|
|
785
|
-
const version = this.checkpoint.channel_versions[channelName];
|
|
786
|
-
this.checkpoint.versions_seen[constants_js_1.INTERRUPT] = {
|
|
787
|
-
...this.checkpoint.versions_seen[constants_js_1.INTERRUPT],
|
|
788
|
-
[channelName]: version,
|
|
789
|
-
};
|
|
790
|
-
}
|
|
791
|
-
}
|
|
792
|
-
// produce values output
|
|
793
|
-
const valuesOutput = await (0, utils_js_1.gatherIterator)((0, utils_js_1.prefixGenerator)((0, io_js_1.mapOutputValues)(this.outputKeys, true, this.channels), "values"));
|
|
794
|
-
this._emit(valuesOutput);
|
|
795
|
-
}
|
|
796
|
-
if (this.isResuming) {
|
|
797
|
-
this.input = INPUT_RESUMING;
|
|
798
|
-
}
|
|
799
|
-
else if (isCommandUpdateOrGoto) {
|
|
800
|
-
// we need to create a new checkpoint for Command(update=...) or Command(goto=...)
|
|
801
|
-
// in case the result of Command(goto=...) is an interrupt.
|
|
802
|
-
// If not done, the checkpoint containing the interrupt will be lost.
|
|
803
|
-
await this._putCheckpoint({ source: "input" });
|
|
804
|
-
this.input = INPUT_DONE;
|
|
805
|
-
}
|
|
806
|
-
else {
|
|
807
|
-
// map inputs to channel updates
|
|
808
|
-
const inputWrites = await (0, utils_js_1.gatherIterator)((0, io_js_1.mapInput)(inputKeys, this.input));
|
|
809
|
-
if (inputWrites.length > 0) {
|
|
810
|
-
const discardTasks = (0, algo_js_1._prepareNextTasks)(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, { step: this.step });
|
|
811
|
-
(0, algo_js_1._applyWrites)(this.checkpoint, this.channels, Object.values(discardTasks).concat([
|
|
812
|
-
{
|
|
813
|
-
name: constants_js_1.INPUT,
|
|
814
|
-
writes: inputWrites,
|
|
815
|
-
triggers: [],
|
|
816
|
-
},
|
|
817
|
-
]), this.checkpointerGetNextVersion, this.triggerToNodes);
|
|
818
|
-
// save input checkpoint
|
|
819
|
-
await this._putCheckpoint({ source: "input" });
|
|
820
|
-
this.input = INPUT_DONE;
|
|
821
|
-
}
|
|
822
|
-
else if (!(constants_js_1.CONFIG_KEY_RESUMING in (this.config.configurable ?? {}))) {
|
|
823
|
-
throw new errors_js_1.EmptyInputError(`Received no input writes for ${JSON.stringify(inputKeys, null, 2)}`);
|
|
824
|
-
}
|
|
825
|
-
else {
|
|
826
|
-
// done with input
|
|
827
|
-
this.input = INPUT_DONE;
|
|
828
|
-
}
|
|
829
|
-
}
|
|
830
|
-
if (!this.isNested) {
|
|
831
|
-
this.config = (0, index_js_1.patchConfigurable)(this.config, {
|
|
832
|
-
[constants_js_1.CONFIG_KEY_RESUMING]: this.isResuming,
|
|
833
|
-
});
|
|
834
|
-
}
|
|
835
|
-
}
|
|
836
|
-
_emit(values) {
|
|
837
|
-
for (const [mode, payload] of values) {
|
|
838
|
-
if (this.stream.modes.has(mode)) {
|
|
839
|
-
this.stream.push([this.checkpointNamespace, mode, payload]);
|
|
840
|
-
}
|
|
841
|
-
// debug mode is a "checkpoints" or "tasks" wrapped in an object
|
|
842
|
-
// TODO: consider deprecating this in 1.x
|
|
843
|
-
if ((mode === "checkpoints" || mode === "tasks") &&
|
|
844
|
-
this.stream.modes.has("debug")) {
|
|
845
|
-
const step = mode === "checkpoints" ? this.step - 1 : this.step;
|
|
846
|
-
const timestamp = new Date().toISOString();
|
|
847
|
-
const type = (() => {
|
|
848
|
-
if (mode === "checkpoints") {
|
|
849
|
-
return "checkpoint";
|
|
850
|
-
}
|
|
851
|
-
else if (typeof payload === "object" &&
|
|
852
|
-
payload != null &&
|
|
853
|
-
"result" in payload) {
|
|
854
|
-
return "task_result";
|
|
855
|
-
}
|
|
856
|
-
else {
|
|
857
|
-
return "task";
|
|
858
|
-
}
|
|
859
|
-
})();
|
|
860
|
-
this.stream.push([
|
|
861
|
-
this.checkpointNamespace,
|
|
862
|
-
"debug",
|
|
863
|
-
{ step, type, timestamp, payload },
|
|
864
|
-
]);
|
|
865
|
-
}
|
|
866
|
-
}
|
|
867
|
-
}
|
|
868
|
-
_putCheckpoint(inputMetadata) {
|
|
869
|
-
const exiting = this.checkpointMetadata === inputMetadata;
|
|
870
|
-
const doCheckpoint = this.checkpointer != null && (this.durability !== "exit" || exiting);
|
|
871
|
-
const storeCheckpoint = (checkpoint) => {
|
|
872
|
-
// store the previous checkpoint config for debug events
|
|
873
|
-
this.prevCheckpointConfig = this.checkpointConfig?.configurable
|
|
874
|
-
?.checkpoint_id
|
|
875
|
-
? this.checkpointConfig
|
|
876
|
-
: undefined;
|
|
877
|
-
// child graphs keep at most one checkpoint per parent checkpoint
|
|
878
|
-
// this is achieved by writing child checkpoints as progress is made
|
|
879
|
-
// (so that error recovery / resuming from interrupt don't lose work)
|
|
880
|
-
// but doing so always with an id equal to that of the parent checkpoint
|
|
881
|
-
this.checkpointConfig = (0, index_js_1.patchConfigurable)(this.checkpointConfig, {
|
|
882
|
-
[constants_js_1.CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
|
|
883
|
-
});
|
|
884
|
-
const channelVersions = { ...this.checkpoint.channel_versions };
|
|
885
|
-
const newVersions = (0, index_js_1.getNewChannelVersions)(this.checkpointPreviousVersions, channelVersions);
|
|
886
|
-
this.checkpointPreviousVersions = channelVersions;
|
|
887
|
-
// save it, without blocking
|
|
888
|
-
// if there's a previous checkpoint save in progress, wait for it
|
|
889
|
-
// ensuring checkpointers receive checkpoints in order
|
|
890
|
-
void this._checkpointerPutAfterPrevious({
|
|
891
|
-
config: { ...this.checkpointConfig },
|
|
892
|
-
checkpoint: (0, langgraph_checkpoint_1.copyCheckpoint)(checkpoint),
|
|
893
|
-
metadata: { ...this.checkpointMetadata },
|
|
894
|
-
newVersions,
|
|
895
|
-
});
|
|
896
|
-
this.checkpointConfig = {
|
|
897
|
-
...this.checkpointConfig,
|
|
898
|
-
configurable: {
|
|
899
|
-
...this.checkpointConfig.configurable,
|
|
900
|
-
checkpoint_id: this.checkpoint.id,
|
|
901
|
-
},
|
|
902
|
-
};
|
|
903
|
-
};
|
|
904
|
-
if (!exiting) {
|
|
905
|
-
this.checkpointMetadata = {
|
|
906
|
-
...inputMetadata,
|
|
907
|
-
step: this.step,
|
|
908
|
-
parents: this.config.configurable?.[constants_js_1.CONFIG_KEY_CHECKPOINT_MAP] ?? {},
|
|
909
|
-
};
|
|
910
|
-
}
|
|
911
|
-
// create new checkpoint
|
|
912
|
-
this.checkpoint = (0, base_js_1.createCheckpoint)(this.checkpoint, doCheckpoint ? this.channels : undefined, this.step, exiting ? { id: this.checkpoint.id } : undefined);
|
|
913
|
-
// Bail if no checkpointer
|
|
914
|
-
if (doCheckpoint)
|
|
915
|
-
storeCheckpoint(this.checkpoint);
|
|
916
|
-
if (!exiting) {
|
|
917
|
-
// increment step
|
|
918
|
-
this.step += 1;
|
|
919
|
-
}
|
|
920
|
-
}
|
|
921
|
-
_flushPendingWrites() {
|
|
922
|
-
if (this.checkpointer == null)
|
|
923
|
-
return;
|
|
924
|
-
if (this.checkpointPendingWrites.length === 0)
|
|
925
|
-
return;
|
|
926
|
-
// patch config
|
|
927
|
-
const config = (0, index_js_1.patchConfigurable)(this.checkpointConfig, {
|
|
928
|
-
[constants_js_1.CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
|
|
929
|
-
[constants_js_1.CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id,
|
|
930
|
-
});
|
|
931
|
-
// group writes by task id
|
|
932
|
-
const byTask = {};
|
|
933
|
-
for (const [tid, key, value] of this.checkpointPendingWrites) {
|
|
934
|
-
byTask[tid] ??= [];
|
|
935
|
-
byTask[tid].push([key, value]);
|
|
936
|
-
}
|
|
937
|
-
// submit writes to checkpointer
|
|
938
|
-
for (const [tid, ws] of Object.entries(byTask)) {
|
|
939
|
-
this.checkpointerPromises.push(this.checkpointer.putWrites(config, ws, tid));
|
|
940
|
-
}
|
|
941
|
-
}
|
|
942
|
-
_matchWrites(tasks) {
|
|
943
|
-
for (const [tid, k, v] of this.checkpointPendingWrites) {
|
|
944
|
-
if (k === constants_js_1.ERROR || k === constants_js_1.INTERRUPT || k === constants_js_1.RESUME) {
|
|
945
|
-
continue;
|
|
946
|
-
}
|
|
947
|
-
const task = Object.values(tasks).find((t) => t.id === tid);
|
|
948
|
-
if (task) {
|
|
949
|
-
task.writes.push([k, v]);
|
|
950
|
-
}
|
|
951
|
-
}
|
|
952
|
-
for (const task of Object.values(tasks)) {
|
|
953
|
-
if (task.writes.length > 0) {
|
|
954
|
-
this._outputWrites(task.id, task.writes, true);
|
|
955
|
-
}
|
|
956
|
-
}
|
|
957
|
-
}
|
|
19
|
+
return new require_stream.IterableReadableWritableStream({
|
|
20
|
+
passthroughFn: (value) => {
|
|
21
|
+
for (const stream of streams) if (stream.modes.has(value[1])) stream.push(value);
|
|
22
|
+
},
|
|
23
|
+
modes: new Set(streams.flatMap((s) => Array.from(s.modes)))
|
|
24
|
+
});
|
|
958
25
|
}
|
|
26
|
+
var AsyncBatchedCache = class extends __langchain_langgraph_checkpoint.BaseCache {
|
|
27
|
+
cache;
|
|
28
|
+
queue = Promise.resolve();
|
|
29
|
+
constructor(cache) {
|
|
30
|
+
super();
|
|
31
|
+
this.cache = cache;
|
|
32
|
+
}
|
|
33
|
+
async get(keys) {
|
|
34
|
+
return this.enqueueOperation("get", keys);
|
|
35
|
+
}
|
|
36
|
+
async set(pairs) {
|
|
37
|
+
return this.enqueueOperation("set", pairs);
|
|
38
|
+
}
|
|
39
|
+
async clear(namespaces) {
|
|
40
|
+
return this.enqueueOperation("clear", namespaces);
|
|
41
|
+
}
|
|
42
|
+
async stop() {
|
|
43
|
+
await this.queue;
|
|
44
|
+
}
|
|
45
|
+
enqueueOperation(type, ...args) {
|
|
46
|
+
const newPromise = this.queue.then(() => {
|
|
47
|
+
return this.cache[type](...args);
|
|
48
|
+
});
|
|
49
|
+
this.queue = newPromise.then(() => void 0, () => void 0);
|
|
50
|
+
return newPromise;
|
|
51
|
+
}
|
|
52
|
+
};
|
|
53
|
+
var PregelLoop = class PregelLoop {
|
|
54
|
+
input;
|
|
55
|
+
output;
|
|
56
|
+
config;
|
|
57
|
+
checkpointer;
|
|
58
|
+
checkpointerGetNextVersion;
|
|
59
|
+
channels;
|
|
60
|
+
checkpoint;
|
|
61
|
+
checkpointIdSaved;
|
|
62
|
+
checkpointConfig;
|
|
63
|
+
checkpointMetadata;
|
|
64
|
+
checkpointNamespace;
|
|
65
|
+
checkpointPendingWrites = [];
|
|
66
|
+
checkpointPreviousVersions;
|
|
67
|
+
step;
|
|
68
|
+
stop;
|
|
69
|
+
durability;
|
|
70
|
+
outputKeys;
|
|
71
|
+
streamKeys;
|
|
72
|
+
nodes;
|
|
73
|
+
skipDoneTasks;
|
|
74
|
+
prevCheckpointConfig;
|
|
75
|
+
updatedChannels;
|
|
76
|
+
status = "pending";
|
|
77
|
+
tasks = {};
|
|
78
|
+
stream;
|
|
79
|
+
checkpointerPromises = [];
|
|
80
|
+
isNested;
|
|
81
|
+
_checkpointerChainedPromise = Promise.resolve();
|
|
82
|
+
store;
|
|
83
|
+
cache;
|
|
84
|
+
manager;
|
|
85
|
+
interruptAfter;
|
|
86
|
+
interruptBefore;
|
|
87
|
+
toInterrupt = [];
|
|
88
|
+
debug = false;
|
|
89
|
+
triggerToNodes;
|
|
90
|
+
get isResuming() {
|
|
91
|
+
let hasChannelVersions = false;
|
|
92
|
+
if (require_constants.START in this.checkpoint.channel_versions) hasChannelVersions = true;
|
|
93
|
+
else for (const chan in this.checkpoint.channel_versions) if (Object.prototype.hasOwnProperty.call(this.checkpoint.channel_versions, chan)) {
|
|
94
|
+
hasChannelVersions = true;
|
|
95
|
+
break;
|
|
96
|
+
}
|
|
97
|
+
const configHasResumingFlag = this.config.configurable?.[require_constants.CONFIG_KEY_RESUMING] !== void 0;
|
|
98
|
+
const configIsResuming = configHasResumingFlag && this.config.configurable?.[require_constants.CONFIG_KEY_RESUMING];
|
|
99
|
+
const inputIsNullOrUndefined = this.input === null || this.input === void 0;
|
|
100
|
+
const inputIsCommandResuming = require_constants.isCommand(this.input) && this.input.resume != null;
|
|
101
|
+
const inputIsResuming = this.input === INPUT_RESUMING;
|
|
102
|
+
const runIdMatchesPrevious = !this.isNested && this.config.metadata?.run_id !== void 0 && this.checkpointMetadata?.run_id !== void 0 && this.config.metadata.run_id === this.checkpointMetadata?.run_id;
|
|
103
|
+
return hasChannelVersions && (configIsResuming || inputIsNullOrUndefined || inputIsCommandResuming || inputIsResuming || runIdMatchesPrevious);
|
|
104
|
+
}
|
|
105
|
+
constructor(params) {
|
|
106
|
+
this.input = params.input;
|
|
107
|
+
this.checkpointer = params.checkpointer;
|
|
108
|
+
if (this.checkpointer !== void 0) this.checkpointerGetNextVersion = this.checkpointer.getNextVersion.bind(this.checkpointer);
|
|
109
|
+
else this.checkpointerGetNextVersion = require_algo.increment;
|
|
110
|
+
this.checkpoint = params.checkpoint;
|
|
111
|
+
this.checkpointMetadata = params.checkpointMetadata;
|
|
112
|
+
this.checkpointPreviousVersions = params.checkpointPreviousVersions;
|
|
113
|
+
this.channels = params.channels;
|
|
114
|
+
this.checkpointPendingWrites = params.checkpointPendingWrites;
|
|
115
|
+
this.step = params.step;
|
|
116
|
+
this.stop = params.stop;
|
|
117
|
+
this.config = params.config;
|
|
118
|
+
this.checkpointConfig = params.checkpointConfig;
|
|
119
|
+
this.isNested = params.isNested;
|
|
120
|
+
this.manager = params.manager;
|
|
121
|
+
this.outputKeys = params.outputKeys;
|
|
122
|
+
this.streamKeys = params.streamKeys;
|
|
123
|
+
this.nodes = params.nodes;
|
|
124
|
+
this.skipDoneTasks = params.skipDoneTasks;
|
|
125
|
+
this.store = params.store;
|
|
126
|
+
this.cache = params.cache ? new AsyncBatchedCache(params.cache) : void 0;
|
|
127
|
+
this.stream = params.stream;
|
|
128
|
+
this.checkpointNamespace = params.checkpointNamespace;
|
|
129
|
+
this.prevCheckpointConfig = params.prevCheckpointConfig;
|
|
130
|
+
this.interruptAfter = params.interruptAfter;
|
|
131
|
+
this.interruptBefore = params.interruptBefore;
|
|
132
|
+
this.durability = params.durability;
|
|
133
|
+
this.debug = params.debug;
|
|
134
|
+
this.triggerToNodes = params.triggerToNodes;
|
|
135
|
+
}
|
|
136
|
+
static async initialize(params) {
|
|
137
|
+
let { config, stream } = params;
|
|
138
|
+
if (stream !== void 0 && config.configurable?.[require_constants.CONFIG_KEY_STREAM] !== void 0) stream = createDuplexStream(stream, config.configurable[require_constants.CONFIG_KEY_STREAM]);
|
|
139
|
+
const skipDoneTasks = config.configurable ? !("checkpoint_id" in config.configurable) : true;
|
|
140
|
+
const scratchpad = config.configurable?.[require_constants.CONFIG_KEY_SCRATCHPAD];
|
|
141
|
+
if (config.configurable && scratchpad) {
|
|
142
|
+
if (scratchpad.subgraphCounter > 0) config = require_index.patchConfigurable(config, { [require_constants.CONFIG_KEY_CHECKPOINT_NS]: [config.configurable[require_constants.CONFIG_KEY_CHECKPOINT_NS], scratchpad.subgraphCounter.toString()].join(require_constants.CHECKPOINT_NAMESPACE_SEPARATOR) });
|
|
143
|
+
scratchpad.subgraphCounter += 1;
|
|
144
|
+
}
|
|
145
|
+
const isNested = require_constants.CONFIG_KEY_READ in (config.configurable ?? {});
|
|
146
|
+
if (!isNested && config.configurable?.checkpoint_ns !== void 0 && config.configurable?.checkpoint_ns !== "") config = require_index.patchConfigurable(config, {
|
|
147
|
+
checkpoint_ns: "",
|
|
148
|
+
checkpoint_id: void 0
|
|
149
|
+
});
|
|
150
|
+
let checkpointConfig = config;
|
|
151
|
+
if (config.configurable?.[require_constants.CONFIG_KEY_CHECKPOINT_MAP] !== void 0 && config.configurable?.[require_constants.CONFIG_KEY_CHECKPOINT_MAP]?.[config.configurable?.checkpoint_ns]) checkpointConfig = require_index.patchConfigurable(config, { checkpoint_id: config.configurable[require_constants.CONFIG_KEY_CHECKPOINT_MAP][config.configurable?.checkpoint_ns] });
|
|
152
|
+
const checkpointNamespace = config.configurable?.checkpoint_ns?.split(require_constants.CHECKPOINT_NAMESPACE_SEPARATOR) ?? [];
|
|
153
|
+
const saved = await params.checkpointer?.getTuple(checkpointConfig) ?? {
|
|
154
|
+
config,
|
|
155
|
+
checkpoint: (0, __langchain_langgraph_checkpoint.emptyCheckpoint)(),
|
|
156
|
+
metadata: {
|
|
157
|
+
source: "input",
|
|
158
|
+
step: -2,
|
|
159
|
+
parents: {}
|
|
160
|
+
},
|
|
161
|
+
pendingWrites: []
|
|
162
|
+
};
|
|
163
|
+
checkpointConfig = {
|
|
164
|
+
...config,
|
|
165
|
+
...saved.config,
|
|
166
|
+
configurable: {
|
|
167
|
+
checkpoint_ns: "",
|
|
168
|
+
...config.configurable,
|
|
169
|
+
...saved.config.configurable
|
|
170
|
+
}
|
|
171
|
+
};
|
|
172
|
+
const prevCheckpointConfig = saved.parentConfig;
|
|
173
|
+
const checkpoint = (0, __langchain_langgraph_checkpoint.copyCheckpoint)(saved.checkpoint);
|
|
174
|
+
const checkpointMetadata = { ...saved.metadata };
|
|
175
|
+
const checkpointPendingWrites = saved.pendingWrites ?? [];
|
|
176
|
+
const channels = require_base.emptyChannels(params.channelSpecs, checkpoint);
|
|
177
|
+
const step = (checkpointMetadata.step ?? 0) + 1;
|
|
178
|
+
const stop = step + (config.recursionLimit ?? DEFAULT_LOOP_LIMIT) + 1;
|
|
179
|
+
const checkpointPreviousVersions = { ...checkpoint.channel_versions };
|
|
180
|
+
const store = params.store ? new __langchain_langgraph_checkpoint.AsyncBatchedStore(params.store) : void 0;
|
|
181
|
+
if (store) await store.start();
|
|
182
|
+
return new PregelLoop({
|
|
183
|
+
input: params.input,
|
|
184
|
+
config,
|
|
185
|
+
checkpointer: params.checkpointer,
|
|
186
|
+
checkpoint,
|
|
187
|
+
checkpointMetadata,
|
|
188
|
+
checkpointConfig,
|
|
189
|
+
prevCheckpointConfig,
|
|
190
|
+
checkpointNamespace,
|
|
191
|
+
channels,
|
|
192
|
+
isNested,
|
|
193
|
+
manager: params.manager,
|
|
194
|
+
skipDoneTasks,
|
|
195
|
+
step,
|
|
196
|
+
stop,
|
|
197
|
+
checkpointPreviousVersions,
|
|
198
|
+
checkpointPendingWrites,
|
|
199
|
+
outputKeys: params.outputKeys ?? [],
|
|
200
|
+
streamKeys: params.streamKeys ?? [],
|
|
201
|
+
nodes: params.nodes,
|
|
202
|
+
stream,
|
|
203
|
+
store,
|
|
204
|
+
cache: params.cache,
|
|
205
|
+
interruptAfter: params.interruptAfter,
|
|
206
|
+
interruptBefore: params.interruptBefore,
|
|
207
|
+
durability: params.durability,
|
|
208
|
+
debug: params.debug,
|
|
209
|
+
triggerToNodes: params.triggerToNodes
|
|
210
|
+
});
|
|
211
|
+
}
|
|
212
|
+
_checkpointerPutAfterPrevious(input) {
|
|
213
|
+
this._checkpointerChainedPromise = this._checkpointerChainedPromise.then(() => {
|
|
214
|
+
return this.checkpointer?.put(input.config, input.checkpoint, input.metadata, input.newVersions);
|
|
215
|
+
});
|
|
216
|
+
this.checkpointerPromises.push(this._checkpointerChainedPromise);
|
|
217
|
+
}
|
|
218
|
+
/**
|
|
219
|
+
* Put writes for a task, to be read by the next tick.
|
|
220
|
+
* @param taskId
|
|
221
|
+
* @param writes
|
|
222
|
+
*/
|
|
223
|
+
putWrites(taskId, writes) {
|
|
224
|
+
let writesCopy = writes;
|
|
225
|
+
if (writesCopy.length === 0) return;
|
|
226
|
+
if (writesCopy.every(([key]) => key in __langchain_langgraph_checkpoint.WRITES_IDX_MAP)) writesCopy = Array.from(new Map(writesCopy.map((w) => [w[0], w])).values());
|
|
227
|
+
this.checkpointPendingWrites = this.checkpointPendingWrites.filter((w) => w[0] !== taskId);
|
|
228
|
+
for (const [c, v] of writesCopy) this.checkpointPendingWrites.push([
|
|
229
|
+
taskId,
|
|
230
|
+
c,
|
|
231
|
+
v
|
|
232
|
+
]);
|
|
233
|
+
const config = require_index.patchConfigurable(this.checkpointConfig, {
|
|
234
|
+
[require_constants.CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
|
|
235
|
+
[require_constants.CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id
|
|
236
|
+
});
|
|
237
|
+
if (this.durability !== "exit" && this.checkpointer != null) this.checkpointerPromises.push(this.checkpointer.putWrites(config, writesCopy, taskId));
|
|
238
|
+
if (this.tasks) this._outputWrites(taskId, writesCopy);
|
|
239
|
+
if (!writes.length || !this.cache || !this.tasks) return;
|
|
240
|
+
const task = this.tasks[taskId];
|
|
241
|
+
if (task == null || task.cache_key == null) return;
|
|
242
|
+
if (writes[0][0] === require_constants.ERROR || writes[0][0] === require_constants.INTERRUPT) return;
|
|
243
|
+
this.cache.set([{
|
|
244
|
+
key: [task.cache_key.ns, task.cache_key.key],
|
|
245
|
+
value: task.writes,
|
|
246
|
+
ttl: task.cache_key.ttl
|
|
247
|
+
}]);
|
|
248
|
+
}
|
|
249
|
+
_outputWrites(taskId, writes, cached = false) {
|
|
250
|
+
const task = this.tasks[taskId];
|
|
251
|
+
if (task !== void 0) {
|
|
252
|
+
if (task.config !== void 0 && (task.config.tags ?? []).includes(require_constants.TAG_HIDDEN)) return;
|
|
253
|
+
if (writes.length > 0) {
|
|
254
|
+
if (writes[0][0] === require_constants.INTERRUPT) {
|
|
255
|
+
if (task.path?.[0] === require_constants.PUSH && task.path?.at(-1) === true) return;
|
|
256
|
+
const interruptWrites = writes.filter((w) => w[0] === require_constants.INTERRUPT).flatMap((w) => w[1]);
|
|
257
|
+
this._emit([["updates", { [require_constants.INTERRUPT]: interruptWrites }], ["values", { [require_constants.INTERRUPT]: interruptWrites }]]);
|
|
258
|
+
} else if (writes[0][0] !== require_constants.ERROR) this._emit(require_utils.gatherIteratorSync(require_utils.prefixGenerator(require_io.mapOutputUpdates(this.outputKeys, [[task, writes]], cached), "updates")));
|
|
259
|
+
}
|
|
260
|
+
if (!cached) this._emit(require_utils.gatherIteratorSync(require_utils.prefixGenerator(require_debug.mapDebugTaskResults([[task, writes]], this.streamKeys), "tasks")));
|
|
261
|
+
}
|
|
262
|
+
}
|
|
263
|
+
async _matchCachedWrites() {
|
|
264
|
+
if (!this.cache) return [];
|
|
265
|
+
const matched = [];
|
|
266
|
+
const serializeKey = ([ns, key]) => {
|
|
267
|
+
return `ns:${ns.join(",")}|key:${key}`;
|
|
268
|
+
};
|
|
269
|
+
const keys = [];
|
|
270
|
+
const keyMap = {};
|
|
271
|
+
for (const task of Object.values(this.tasks)) if (task.cache_key != null && !task.writes.length) {
|
|
272
|
+
keys.push([task.cache_key.ns, task.cache_key.key]);
|
|
273
|
+
keyMap[serializeKey([task.cache_key.ns, task.cache_key.key])] = task;
|
|
274
|
+
}
|
|
275
|
+
if (keys.length === 0) return [];
|
|
276
|
+
const cache = await this.cache.get(keys);
|
|
277
|
+
for (const { key, value } of cache) {
|
|
278
|
+
const task = keyMap[serializeKey(key)];
|
|
279
|
+
if (task != null) {
|
|
280
|
+
task.writes.push(...value);
|
|
281
|
+
matched.push({
|
|
282
|
+
task,
|
|
283
|
+
result: value
|
|
284
|
+
});
|
|
285
|
+
}
|
|
286
|
+
}
|
|
287
|
+
return matched;
|
|
288
|
+
}
|
|
289
|
+
/**
|
|
290
|
+
* Execute a single iteration of the Pregel loop.
|
|
291
|
+
* Returns true if more iterations are needed.
|
|
292
|
+
* @param params
|
|
293
|
+
*/
|
|
294
|
+
async tick(params) {
|
|
295
|
+
if (this.store && !this.store.isRunning) await this.store?.start();
|
|
296
|
+
const { inputKeys = [] } = params;
|
|
297
|
+
if (this.status !== "pending") throw new Error(`Cannot tick when status is no longer "pending". Current status: "${this.status}"`);
|
|
298
|
+
if (![INPUT_DONE, INPUT_RESUMING].includes(this.input)) await this._first(inputKeys);
|
|
299
|
+
else if (this.toInterrupt.length > 0) {
|
|
300
|
+
this.status = "interrupt_before";
|
|
301
|
+
throw new require_errors.GraphInterrupt();
|
|
302
|
+
} else if (Object.values(this.tasks).every((task) => task.writes.length > 0)) {
|
|
303
|
+
const writes = Object.values(this.tasks).flatMap((t) => t.writes);
|
|
304
|
+
this.updatedChannels = require_algo._applyWrites(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
|
|
305
|
+
const valuesOutput = await require_utils.gatherIterator(require_utils.prefixGenerator(require_io.mapOutputValues(this.outputKeys, writes, this.channels), "values"));
|
|
306
|
+
this._emit(valuesOutput);
|
|
307
|
+
this.checkpointPendingWrites = [];
|
|
308
|
+
await this._putCheckpoint({ source: "loop" });
|
|
309
|
+
if (require_algo.shouldInterrupt(this.checkpoint, this.interruptAfter, Object.values(this.tasks))) {
|
|
310
|
+
this.status = "interrupt_after";
|
|
311
|
+
throw new require_errors.GraphInterrupt();
|
|
312
|
+
}
|
|
313
|
+
if (this.config.configurable?.[require_constants.CONFIG_KEY_RESUMING] !== void 0) delete this.config.configurable?.[require_constants.CONFIG_KEY_RESUMING];
|
|
314
|
+
} else return false;
|
|
315
|
+
if (this.step > this.stop) {
|
|
316
|
+
this.status = "out_of_steps";
|
|
317
|
+
return false;
|
|
318
|
+
}
|
|
319
|
+
const nextTasks = require_algo._prepareNextTasks(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, {
|
|
320
|
+
step: this.step,
|
|
321
|
+
checkpointer: this.checkpointer,
|
|
322
|
+
isResuming: this.isResuming,
|
|
323
|
+
manager: this.manager,
|
|
324
|
+
store: this.store,
|
|
325
|
+
stream: this.stream,
|
|
326
|
+
triggerToNodes: this.triggerToNodes,
|
|
327
|
+
updatedChannels: this.updatedChannels
|
|
328
|
+
});
|
|
329
|
+
this.tasks = nextTasks;
|
|
330
|
+
if (this.checkpointer) this._emit(await require_utils.gatherIterator(require_utils.prefixGenerator(require_debug.mapDebugCheckpoint(this.checkpointConfig, this.channels, this.streamKeys, this.checkpointMetadata, Object.values(this.tasks), this.checkpointPendingWrites, this.prevCheckpointConfig, this.outputKeys), "checkpoints")));
|
|
331
|
+
if (Object.values(this.tasks).length === 0) {
|
|
332
|
+
this.status = "done";
|
|
333
|
+
return false;
|
|
334
|
+
}
|
|
335
|
+
if (this.skipDoneTasks && this.checkpointPendingWrites.length > 0) {
|
|
336
|
+
for (const [tid, k, v] of this.checkpointPendingWrites) {
|
|
337
|
+
if (k === require_constants.ERROR || k === require_constants.INTERRUPT || k === require_constants.RESUME) continue;
|
|
338
|
+
const task = Object.values(this.tasks).find((t) => t.id === tid);
|
|
339
|
+
if (task) task.writes.push([k, v]);
|
|
340
|
+
}
|
|
341
|
+
for (const task of Object.values(this.tasks)) if (task.writes.length > 0) this._outputWrites(task.id, task.writes, true);
|
|
342
|
+
}
|
|
343
|
+
if (Object.values(this.tasks).every((task) => task.writes.length > 0)) return this.tick({ inputKeys });
|
|
344
|
+
if (require_algo.shouldInterrupt(this.checkpoint, this.interruptBefore, Object.values(this.tasks))) {
|
|
345
|
+
this.status = "interrupt_before";
|
|
346
|
+
throw new require_errors.GraphInterrupt();
|
|
347
|
+
}
|
|
348
|
+
const debugOutput = await require_utils.gatherIterator(require_utils.prefixGenerator(require_debug.mapDebugTasks(Object.values(this.tasks)), "tasks"));
|
|
349
|
+
this._emit(debugOutput);
|
|
350
|
+
return true;
|
|
351
|
+
}
|
|
352
|
+
async finishAndHandleError(error) {
|
|
353
|
+
if (this.durability === "exit" && (!this.isNested || typeof error !== "undefined" || this.checkpointNamespace.every((part) => !part.includes(require_constants.CHECKPOINT_NAMESPACE_END)))) {
|
|
354
|
+
this._putCheckpoint(this.checkpointMetadata);
|
|
355
|
+
this._flushPendingWrites();
|
|
356
|
+
}
|
|
357
|
+
const suppress = this._suppressInterrupt(error);
|
|
358
|
+
if (suppress || error === void 0) this.output = require_io.readChannels(this.channels, this.outputKeys);
|
|
359
|
+
if (suppress) {
|
|
360
|
+
if (this.tasks !== void 0 && this.checkpointPendingWrites.length > 0 && Object.values(this.tasks).some((task) => task.writes.length > 0)) {
|
|
361
|
+
this.updatedChannels = require_algo._applyWrites(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
|
|
362
|
+
this._emit(require_utils.gatherIteratorSync(require_utils.prefixGenerator(require_io.mapOutputValues(this.outputKeys, Object.values(this.tasks).flatMap((t) => t.writes), this.channels), "values")));
|
|
363
|
+
}
|
|
364
|
+
if (require_errors.isGraphInterrupt(error) && !error.interrupts.length) this._emit([["updates", { [require_constants.INTERRUPT]: [] }], ["values", { [require_constants.INTERRUPT]: [] }]]);
|
|
365
|
+
}
|
|
366
|
+
return suppress;
|
|
367
|
+
}
|
|
368
|
+
async acceptPush(task, writeIdx, call) {
|
|
369
|
+
if (this.interruptAfter?.length > 0 && require_algo.shouldInterrupt(this.checkpoint, this.interruptAfter, [task])) {
|
|
370
|
+
this.toInterrupt.push(task);
|
|
371
|
+
return;
|
|
372
|
+
}
|
|
373
|
+
const pushed = require_algo._prepareSingleTask([
|
|
374
|
+
require_constants.PUSH,
|
|
375
|
+
task.path ?? [],
|
|
376
|
+
writeIdx,
|
|
377
|
+
task.id,
|
|
378
|
+
call
|
|
379
|
+
], this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, task.config ?? {}, true, {
|
|
380
|
+
step: this.step,
|
|
381
|
+
checkpointer: this.checkpointer,
|
|
382
|
+
manager: this.manager,
|
|
383
|
+
store: this.store,
|
|
384
|
+
stream: this.stream
|
|
385
|
+
});
|
|
386
|
+
if (!pushed) return;
|
|
387
|
+
if (this.interruptBefore?.length > 0 && require_algo.shouldInterrupt(this.checkpoint, this.interruptBefore, [pushed])) {
|
|
388
|
+
this.toInterrupt.push(pushed);
|
|
389
|
+
return;
|
|
390
|
+
}
|
|
391
|
+
this._emit(require_utils.gatherIteratorSync(require_utils.prefixGenerator(require_debug.mapDebugTasks([pushed]), "tasks")));
|
|
392
|
+
if (this.debug) require_debug.printStepTasks(this.step, [pushed]);
|
|
393
|
+
this.tasks[pushed.id] = pushed;
|
|
394
|
+
if (this.skipDoneTasks) this._matchWrites({ [pushed.id]: pushed });
|
|
395
|
+
const tasks = await this._matchCachedWrites();
|
|
396
|
+
for (const { task: task$1 } of tasks) this._outputWrites(task$1.id, task$1.writes, true);
|
|
397
|
+
return pushed;
|
|
398
|
+
}
|
|
399
|
+
_suppressInterrupt(e) {
|
|
400
|
+
return require_errors.isGraphInterrupt(e) && !this.isNested;
|
|
401
|
+
}
|
|
402
|
+
async _first(inputKeys) {
|
|
403
|
+
const { configurable } = this.config;
|
|
404
|
+
const scratchpad = configurable?.[require_constants.CONFIG_KEY_SCRATCHPAD];
|
|
405
|
+
if (scratchpad && scratchpad.nullResume !== void 0) this.putWrites(require_constants.NULL_TASK_ID, [[require_constants.RESUME, scratchpad.nullResume]]);
|
|
406
|
+
if (require_constants.isCommand(this.input)) {
|
|
407
|
+
const hasResume = this.input.resume != null;
|
|
408
|
+
if (this.input.resume != null && typeof this.input.resume === "object" && Object.keys(this.input.resume).every(require_hash.isXXH3)) {
|
|
409
|
+
this.config.configurable ??= {};
|
|
410
|
+
this.config.configurable[require_constants.CONFIG_KEY_RESUME_MAP] = this.input.resume;
|
|
411
|
+
}
|
|
412
|
+
if (hasResume && this.checkpointer == null) throw new Error("Cannot use Command(resume=...) without checkpointer");
|
|
413
|
+
const writes = {};
|
|
414
|
+
for (const [tid, key, value] of require_io.mapCommand(this.input, this.checkpointPendingWrites)) {
|
|
415
|
+
writes[tid] ??= [];
|
|
416
|
+
writes[tid].push([key, value]);
|
|
417
|
+
}
|
|
418
|
+
if (Object.keys(writes).length === 0) throw new require_errors.EmptyInputError("Received empty Command input");
|
|
419
|
+
for (const [tid, ws] of Object.entries(writes)) this.putWrites(tid, ws);
|
|
420
|
+
}
|
|
421
|
+
const nullWrites = (this.checkpointPendingWrites ?? []).filter((w) => w[0] === require_constants.NULL_TASK_ID).map((w) => w.slice(1));
|
|
422
|
+
if (nullWrites.length > 0) require_algo._applyWrites(this.checkpoint, this.channels, [{
|
|
423
|
+
name: require_constants.INPUT,
|
|
424
|
+
writes: nullWrites,
|
|
425
|
+
triggers: []
|
|
426
|
+
}], this.checkpointerGetNextVersion, this.triggerToNodes);
|
|
427
|
+
const isCommandUpdateOrGoto = require_constants.isCommand(this.input) && nullWrites.length > 0;
|
|
428
|
+
if (this.isResuming || isCommandUpdateOrGoto) {
|
|
429
|
+
for (const channelName in this.channels) {
|
|
430
|
+
if (!Object.prototype.hasOwnProperty.call(this.channels, channelName)) continue;
|
|
431
|
+
if (this.checkpoint.channel_versions[channelName] !== void 0) {
|
|
432
|
+
const version = this.checkpoint.channel_versions[channelName];
|
|
433
|
+
this.checkpoint.versions_seen[require_constants.INTERRUPT] = {
|
|
434
|
+
...this.checkpoint.versions_seen[require_constants.INTERRUPT],
|
|
435
|
+
[channelName]: version
|
|
436
|
+
};
|
|
437
|
+
}
|
|
438
|
+
}
|
|
439
|
+
const valuesOutput = await require_utils.gatherIterator(require_utils.prefixGenerator(require_io.mapOutputValues(this.outputKeys, true, this.channels), "values"));
|
|
440
|
+
this._emit(valuesOutput);
|
|
441
|
+
}
|
|
442
|
+
if (this.isResuming) this.input = INPUT_RESUMING;
|
|
443
|
+
else if (isCommandUpdateOrGoto) {
|
|
444
|
+
await this._putCheckpoint({ source: "input" });
|
|
445
|
+
this.input = INPUT_DONE;
|
|
446
|
+
} else {
|
|
447
|
+
const inputWrites = await require_utils.gatherIterator(require_io.mapInput(inputKeys, this.input));
|
|
448
|
+
if (inputWrites.length > 0) {
|
|
449
|
+
const discardTasks = require_algo._prepareNextTasks(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, { step: this.step });
|
|
450
|
+
this.updatedChannels = require_algo._applyWrites(this.checkpoint, this.channels, Object.values(discardTasks).concat([{
|
|
451
|
+
name: require_constants.INPUT,
|
|
452
|
+
writes: inputWrites,
|
|
453
|
+
triggers: []
|
|
454
|
+
}]), this.checkpointerGetNextVersion, this.triggerToNodes);
|
|
455
|
+
await this._putCheckpoint({ source: "input" });
|
|
456
|
+
this.input = INPUT_DONE;
|
|
457
|
+
} else if (!(require_constants.CONFIG_KEY_RESUMING in (this.config.configurable ?? {}))) throw new require_errors.EmptyInputError(`Received no input writes for ${JSON.stringify(inputKeys, null, 2)}`);
|
|
458
|
+
else this.input = INPUT_DONE;
|
|
459
|
+
}
|
|
460
|
+
if (!this.isNested) this.config = require_index.patchConfigurable(this.config, { [require_constants.CONFIG_KEY_RESUMING]: this.isResuming });
|
|
461
|
+
}
|
|
462
|
+
_emit(values) {
|
|
463
|
+
for (const [mode, payload] of values) {
|
|
464
|
+
if (this.stream.modes.has(mode)) this.stream.push([
|
|
465
|
+
this.checkpointNamespace,
|
|
466
|
+
mode,
|
|
467
|
+
payload
|
|
468
|
+
]);
|
|
469
|
+
if ((mode === "checkpoints" || mode === "tasks") && this.stream.modes.has("debug")) {
|
|
470
|
+
const step = mode === "checkpoints" ? this.step - 1 : this.step;
|
|
471
|
+
const timestamp = (/* @__PURE__ */ new Date()).toISOString();
|
|
472
|
+
const type = (() => {
|
|
473
|
+
if (mode === "checkpoints") return "checkpoint";
|
|
474
|
+
else if (typeof payload === "object" && payload != null && "result" in payload) return "task_result";
|
|
475
|
+
else return "task";
|
|
476
|
+
})();
|
|
477
|
+
this.stream.push([
|
|
478
|
+
this.checkpointNamespace,
|
|
479
|
+
"debug",
|
|
480
|
+
{
|
|
481
|
+
step,
|
|
482
|
+
type,
|
|
483
|
+
timestamp,
|
|
484
|
+
payload
|
|
485
|
+
}
|
|
486
|
+
]);
|
|
487
|
+
}
|
|
488
|
+
}
|
|
489
|
+
}
|
|
490
|
+
_putCheckpoint(inputMetadata) {
|
|
491
|
+
const exiting = this.checkpointMetadata === inputMetadata;
|
|
492
|
+
const doCheckpoint = this.checkpointer != null && (this.durability !== "exit" || exiting);
|
|
493
|
+
const storeCheckpoint = (checkpoint) => {
|
|
494
|
+
this.prevCheckpointConfig = this.checkpointConfig?.configurable?.checkpoint_id ? this.checkpointConfig : void 0;
|
|
495
|
+
this.checkpointConfig = require_index.patchConfigurable(this.checkpointConfig, { [require_constants.CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "" });
|
|
496
|
+
const channelVersions = { ...this.checkpoint.channel_versions };
|
|
497
|
+
const newVersions = require_index.getNewChannelVersions(this.checkpointPreviousVersions, channelVersions);
|
|
498
|
+
this.checkpointPreviousVersions = channelVersions;
|
|
499
|
+
this._checkpointerPutAfterPrevious({
|
|
500
|
+
config: { ...this.checkpointConfig },
|
|
501
|
+
checkpoint: (0, __langchain_langgraph_checkpoint.copyCheckpoint)(checkpoint),
|
|
502
|
+
metadata: { ...this.checkpointMetadata },
|
|
503
|
+
newVersions
|
|
504
|
+
});
|
|
505
|
+
this.checkpointConfig = {
|
|
506
|
+
...this.checkpointConfig,
|
|
507
|
+
configurable: {
|
|
508
|
+
...this.checkpointConfig.configurable,
|
|
509
|
+
checkpoint_id: this.checkpoint.id
|
|
510
|
+
}
|
|
511
|
+
};
|
|
512
|
+
};
|
|
513
|
+
if (!exiting) this.checkpointMetadata = {
|
|
514
|
+
...inputMetadata,
|
|
515
|
+
step: this.step,
|
|
516
|
+
parents: this.config.configurable?.[require_constants.CONFIG_KEY_CHECKPOINT_MAP] ?? {}
|
|
517
|
+
};
|
|
518
|
+
this.checkpoint = require_base.createCheckpoint(this.checkpoint, doCheckpoint ? this.channels : void 0, this.step, exiting ? { id: this.checkpoint.id } : void 0);
|
|
519
|
+
if (doCheckpoint) storeCheckpoint(this.checkpoint);
|
|
520
|
+
if (!exiting) this.step += 1;
|
|
521
|
+
}
|
|
522
|
+
_flushPendingWrites() {
|
|
523
|
+
if (this.checkpointer == null) return;
|
|
524
|
+
if (this.checkpointPendingWrites.length === 0) return;
|
|
525
|
+
const config = require_index.patchConfigurable(this.checkpointConfig, {
|
|
526
|
+
[require_constants.CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
|
|
527
|
+
[require_constants.CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id
|
|
528
|
+
});
|
|
529
|
+
const byTask = {};
|
|
530
|
+
for (const [tid, key, value] of this.checkpointPendingWrites) {
|
|
531
|
+
byTask[tid] ??= [];
|
|
532
|
+
byTask[tid].push([key, value]);
|
|
533
|
+
}
|
|
534
|
+
for (const [tid, ws] of Object.entries(byTask)) this.checkpointerPromises.push(this.checkpointer.putWrites(config, ws, tid));
|
|
535
|
+
}
|
|
536
|
+
_matchWrites(tasks) {
|
|
537
|
+
for (const [tid, k, v] of this.checkpointPendingWrites) {
|
|
538
|
+
if (k === require_constants.ERROR || k === require_constants.INTERRUPT || k === require_constants.RESUME) continue;
|
|
539
|
+
const task = Object.values(tasks).find((t) => t.id === tid);
|
|
540
|
+
if (task) task.writes.push([k, v]);
|
|
541
|
+
}
|
|
542
|
+
for (const task of Object.values(tasks)) if (task.writes.length > 0) this._outputWrites(task.id, task.writes, true);
|
|
543
|
+
}
|
|
544
|
+
};
|
|
545
|
+
|
|
546
|
+
//#endregion
|
|
959
547
|
exports.PregelLoop = PregelLoop;
|
|
960
|
-
//# sourceMappingURL=loop.
|
|
548
|
+
//# sourceMappingURL=loop.cjs.map
|