@langchain/langgraph 1.0.7 → 1.0.12

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.
Files changed (383) hide show
  1. package/package.json +7 -7
  2. package/CHANGELOG.md +0 -286
  3. package/dist/_virtual/rolldown_runtime.cjs +0 -25
  4. package/dist/channels/any_value.cjs +0 -46
  5. package/dist/channels/any_value.cjs.map +0 -1
  6. package/dist/channels/any_value.d.cts +0 -24
  7. package/dist/channels/any_value.d.cts.map +0 -1
  8. package/dist/channels/any_value.d.ts +0 -24
  9. package/dist/channels/any_value.d.ts.map +0 -1
  10. package/dist/channels/any_value.js +0 -46
  11. package/dist/channels/any_value.js.map +0 -1
  12. package/dist/channels/base.cjs +0 -102
  13. package/dist/channels/base.cjs.map +0 -1
  14. package/dist/channels/base.d.cts +0 -79
  15. package/dist/channels/base.d.cts.map +0 -1
  16. package/dist/channels/base.d.ts +0 -79
  17. package/dist/channels/base.d.ts.map +0 -1
  18. package/dist/channels/base.js +0 -97
  19. package/dist/channels/base.js.map +0 -1
  20. package/dist/channels/binop.cjs +0 -49
  21. package/dist/channels/binop.cjs.map +0 -1
  22. package/dist/channels/binop.d.cts +0 -22
  23. package/dist/channels/binop.d.cts.map +0 -1
  24. package/dist/channels/binop.d.ts +0 -22
  25. package/dist/channels/binop.d.ts.map +0 -1
  26. package/dist/channels/binop.js +0 -49
  27. package/dist/channels/binop.js.map +0 -1
  28. package/dist/channels/dynamic_barrier_value.cjs +0 -76
  29. package/dist/channels/dynamic_barrier_value.cjs.map +0 -1
  30. package/dist/channels/dynamic_barrier_value.d.cts +0 -42
  31. package/dist/channels/dynamic_barrier_value.d.cts.map +0 -1
  32. package/dist/channels/dynamic_barrier_value.d.ts +0 -42
  33. package/dist/channels/dynamic_barrier_value.d.ts.map +0 -1
  34. package/dist/channels/dynamic_barrier_value.js +0 -76
  35. package/dist/channels/dynamic_barrier_value.js.map +0 -1
  36. package/dist/channels/ephemeral_value.cjs +0 -46
  37. package/dist/channels/ephemeral_value.cjs.map +0 -1
  38. package/dist/channels/ephemeral_value.d.cts +0 -22
  39. package/dist/channels/ephemeral_value.d.cts.map +0 -1
  40. package/dist/channels/ephemeral_value.d.ts +0 -22
  41. package/dist/channels/ephemeral_value.d.ts.map +0 -1
  42. package/dist/channels/ephemeral_value.js +0 -46
  43. package/dist/channels/ephemeral_value.js.map +0 -1
  44. package/dist/channels/index.cjs +0 -21
  45. package/dist/channels/index.d.cts +0 -9
  46. package/dist/channels/index.d.ts +0 -9
  47. package/dist/channels/index.js +0 -10
  48. package/dist/channels/last_value.cjs +0 -98
  49. package/dist/channels/last_value.cjs.map +0 -1
  50. package/dist/channels/last_value.d.cts +0 -44
  51. package/dist/channels/last_value.d.cts.map +0 -1
  52. package/dist/channels/last_value.d.ts +0 -44
  53. package/dist/channels/last_value.d.ts.map +0 -1
  54. package/dist/channels/last_value.js +0 -97
  55. package/dist/channels/last_value.js.map +0 -1
  56. package/dist/channels/named_barrier_value.cjs +0 -118
  57. package/dist/channels/named_barrier_value.cjs.map +0 -1
  58. package/dist/channels/named_barrier_value.d.cts +0 -46
  59. package/dist/channels/named_barrier_value.d.cts.map +0 -1
  60. package/dist/channels/named_barrier_value.d.ts +0 -46
  61. package/dist/channels/named_barrier_value.d.ts.map +0 -1
  62. package/dist/channels/named_barrier_value.js +0 -116
  63. package/dist/channels/named_barrier_value.js.map +0 -1
  64. package/dist/channels/topic.cjs +0 -65
  65. package/dist/channels/topic.cjs.map +0 -1
  66. package/dist/channels/topic.d.cts +0 -32
  67. package/dist/channels/topic.d.cts.map +0 -1
  68. package/dist/channels/topic.d.ts +0 -32
  69. package/dist/channels/topic.d.ts.map +0 -1
  70. package/dist/channels/topic.js +0 -65
  71. package/dist/channels/topic.js.map +0 -1
  72. package/dist/constants.cjs +0 -406
  73. package/dist/constants.cjs.map +0 -1
  74. package/dist/constants.d.cts +0 -285
  75. package/dist/constants.d.cts.map +0 -1
  76. package/dist/constants.d.ts +0 -285
  77. package/dist/constants.d.ts.map +0 -1
  78. package/dist/constants.js +0 -362
  79. package/dist/constants.js.map +0 -1
  80. package/dist/errors.cjs +0 -163
  81. package/dist/errors.cjs.map +0 -1
  82. package/dist/errors.d.cts +0 -81
  83. package/dist/errors.d.cts.map +0 -1
  84. package/dist/errors.d.ts +0 -81
  85. package/dist/errors.d.ts.map +0 -1
  86. package/dist/errors.js +0 -146
  87. package/dist/errors.js.map +0 -1
  88. package/dist/func/index.cjs +0 -301
  89. package/dist/func/index.cjs.map +0 -1
  90. package/dist/func/index.d.cts +0 -293
  91. package/dist/func/index.d.cts.map +0 -1
  92. package/dist/func/index.d.ts +0 -293
  93. package/dist/func/index.d.ts.map +0 -1
  94. package/dist/func/index.js +0 -298
  95. package/dist/func/index.js.map +0 -1
  96. package/dist/func/types.d.cts +0 -64
  97. package/dist/func/types.d.cts.map +0 -1
  98. package/dist/func/types.d.ts +0 -64
  99. package/dist/func/types.d.ts.map +0 -1
  100. package/dist/graph/annotation.cjs +0 -101
  101. package/dist/graph/annotation.cjs.map +0 -1
  102. package/dist/graph/annotation.d.cts +0 -116
  103. package/dist/graph/annotation.d.cts.map +0 -1
  104. package/dist/graph/annotation.d.ts +0 -116
  105. package/dist/graph/annotation.d.ts.map +0 -1
  106. package/dist/graph/annotation.js +0 -99
  107. package/dist/graph/annotation.js.map +0 -1
  108. package/dist/graph/graph.cjs +0 -423
  109. package/dist/graph/graph.cjs.map +0 -1
  110. package/dist/graph/graph.d.cts +0 -132
  111. package/dist/graph/graph.d.cts.map +0 -1
  112. package/dist/graph/graph.d.ts +0 -132
  113. package/dist/graph/graph.d.ts.map +0 -1
  114. package/dist/graph/graph.js +0 -420
  115. package/dist/graph/graph.js.map +0 -1
  116. package/dist/graph/index.cjs +0 -5
  117. package/dist/graph/index.js +0 -7
  118. package/dist/graph/message.cjs +0 -99
  119. package/dist/graph/message.cjs.map +0 -1
  120. package/dist/graph/message.d.cts +0 -40
  121. package/dist/graph/message.d.cts.map +0 -1
  122. package/dist/graph/message.d.ts +0 -40
  123. package/dist/graph/message.d.ts.map +0 -1
  124. package/dist/graph/message.js +0 -95
  125. package/dist/graph/message.js.map +0 -1
  126. package/dist/graph/messages_annotation.cjs +0 -110
  127. package/dist/graph/messages_annotation.cjs.map +0 -1
  128. package/dist/graph/messages_annotation.d.cts +0 -112
  129. package/dist/graph/messages_annotation.d.cts.map +0 -1
  130. package/dist/graph/messages_annotation.d.ts +0 -112
  131. package/dist/graph/messages_annotation.d.ts.map +0 -1
  132. package/dist/graph/messages_annotation.js +0 -107
  133. package/dist/graph/messages_annotation.js.map +0 -1
  134. package/dist/graph/state.cjs +0 -501
  135. package/dist/graph/state.cjs.map +0 -1
  136. package/dist/graph/state.d.cts +0 -236
  137. package/dist/graph/state.d.cts.map +0 -1
  138. package/dist/graph/state.d.ts +0 -236
  139. package/dist/graph/state.d.ts.map +0 -1
  140. package/dist/graph/state.js +0 -499
  141. package/dist/graph/state.js.map +0 -1
  142. package/dist/graph/zod/index.cjs +0 -10
  143. package/dist/graph/zod/index.d.cts +0 -3
  144. package/dist/graph/zod/index.d.ts +0 -3
  145. package/dist/graph/zod/index.js +0 -5
  146. package/dist/graph/zod/meta.cjs +0 -145
  147. package/dist/graph/zod/meta.cjs.map +0 -1
  148. package/dist/graph/zod/meta.d.cts +0 -116
  149. package/dist/graph/zod/meta.d.cts.map +0 -1
  150. package/dist/graph/zod/meta.d.ts +0 -116
  151. package/dist/graph/zod/meta.d.ts.map +0 -1
  152. package/dist/graph/zod/meta.js +0 -141
  153. package/dist/graph/zod/meta.js.map +0 -1
  154. package/dist/graph/zod/plugin.cjs +0 -41
  155. package/dist/graph/zod/plugin.cjs.map +0 -1
  156. package/dist/graph/zod/plugin.js +0 -41
  157. package/dist/graph/zod/plugin.js.map +0 -1
  158. package/dist/graph/zod/schema.cjs +0 -99
  159. package/dist/graph/zod/schema.cjs.map +0 -1
  160. package/dist/graph/zod/schema.d.cts +0 -38
  161. package/dist/graph/zod/schema.d.cts.map +0 -1
  162. package/dist/graph/zod/schema.d.ts +0 -38
  163. package/dist/graph/zod/schema.d.ts.map +0 -1
  164. package/dist/graph/zod/schema.js +0 -94
  165. package/dist/graph/zod/schema.js.map +0 -1
  166. package/dist/graph/zod/zod-registry.cjs +0 -43
  167. package/dist/graph/zod/zod-registry.cjs.map +0 -1
  168. package/dist/graph/zod/zod-registry.d.cts +0 -51
  169. package/dist/graph/zod/zod-registry.d.cts.map +0 -1
  170. package/dist/graph/zod/zod-registry.d.ts +0 -51
  171. package/dist/graph/zod/zod-registry.d.ts.map +0 -1
  172. package/dist/graph/zod/zod-registry.js +0 -41
  173. package/dist/graph/zod/zod-registry.js.map +0 -1
  174. package/dist/hash.cjs +0 -271
  175. package/dist/hash.cjs.map +0 -1
  176. package/dist/hash.js +0 -269
  177. package/dist/hash.js.map +0 -1
  178. package/dist/index.cjs +0 -112
  179. package/dist/index.cjs.map +0 -1
  180. package/dist/index.d.cts +0 -26
  181. package/dist/index.d.ts +0 -26
  182. package/dist/index.js +0 -22
  183. package/dist/index.js.map +0 -1
  184. package/dist/interrupt.cjs +0 -82
  185. package/dist/interrupt.cjs.map +0 -1
  186. package/dist/interrupt.d.cts +0 -58
  187. package/dist/interrupt.d.cts.map +0 -1
  188. package/dist/interrupt.d.ts +0 -58
  189. package/dist/interrupt.d.ts.map +0 -1
  190. package/dist/interrupt.js +0 -81
  191. package/dist/interrupt.js.map +0 -1
  192. package/dist/prebuilt/agentName.cjs +0 -153
  193. package/dist/prebuilt/agentName.cjs.map +0 -1
  194. package/dist/prebuilt/agentName.d.cts +0 -42
  195. package/dist/prebuilt/agentName.d.cts.map +0 -1
  196. package/dist/prebuilt/agentName.d.ts +0 -42
  197. package/dist/prebuilt/agentName.d.ts.map +0 -1
  198. package/dist/prebuilt/agentName.js +0 -152
  199. package/dist/prebuilt/agentName.js.map +0 -1
  200. package/dist/prebuilt/agent_executor.cjs +0 -44
  201. package/dist/prebuilt/agent_executor.cjs.map +0 -1
  202. package/dist/prebuilt/agent_executor.d.cts +0 -58
  203. package/dist/prebuilt/agent_executor.d.cts.map +0 -1
  204. package/dist/prebuilt/agent_executor.d.ts +0 -58
  205. package/dist/prebuilt/agent_executor.d.ts.map +0 -1
  206. package/dist/prebuilt/agent_executor.js +0 -44
  207. package/dist/prebuilt/agent_executor.js.map +0 -1
  208. package/dist/prebuilt/chat_agent_executor.cjs +0 -68
  209. package/dist/prebuilt/chat_agent_executor.cjs.map +0 -1
  210. package/dist/prebuilt/chat_agent_executor.d.cts +0 -23
  211. package/dist/prebuilt/chat_agent_executor.d.cts.map +0 -1
  212. package/dist/prebuilt/chat_agent_executor.d.ts +0 -23
  213. package/dist/prebuilt/chat_agent_executor.d.ts.map +0 -1
  214. package/dist/prebuilt/chat_agent_executor.js +0 -67
  215. package/dist/prebuilt/chat_agent_executor.js.map +0 -1
  216. package/dist/prebuilt/index.cjs +0 -15
  217. package/dist/prebuilt/index.d.cts +0 -8
  218. package/dist/prebuilt/index.d.ts +0 -8
  219. package/dist/prebuilt/index.js +0 -8
  220. package/dist/prebuilt/interrupt.d.cts +0 -61
  221. package/dist/prebuilt/interrupt.d.cts.map +0 -1
  222. package/dist/prebuilt/interrupt.d.ts +0 -61
  223. package/dist/prebuilt/interrupt.d.ts.map +0 -1
  224. package/dist/prebuilt/react_agent_executor.cjs +0 -344
  225. package/dist/prebuilt/react_agent_executor.cjs.map +0 -1
  226. package/dist/prebuilt/react_agent_executor.d.cts +0 -230
  227. package/dist/prebuilt/react_agent_executor.d.cts.map +0 -1
  228. package/dist/prebuilt/react_agent_executor.d.ts +0 -230
  229. package/dist/prebuilt/react_agent_executor.d.ts.map +0 -1
  230. package/dist/prebuilt/react_agent_executor.js +0 -342
  231. package/dist/prebuilt/react_agent_executor.js.map +0 -1
  232. package/dist/prebuilt/tool_executor.cjs +0 -48
  233. package/dist/prebuilt/tool_executor.cjs.map +0 -1
  234. package/dist/prebuilt/tool_executor.d.cts +0 -42
  235. package/dist/prebuilt/tool_executor.d.cts.map +0 -1
  236. package/dist/prebuilt/tool_executor.d.ts +0 -42
  237. package/dist/prebuilt/tool_executor.d.ts.map +0 -1
  238. package/dist/prebuilt/tool_executor.js +0 -47
  239. package/dist/prebuilt/tool_executor.js.map +0 -1
  240. package/dist/prebuilt/tool_node.cjs +0 -248
  241. package/dist/prebuilt/tool_node.cjs.map +0 -1
  242. package/dist/prebuilt/tool_node.d.cts +0 -182
  243. package/dist/prebuilt/tool_node.d.cts.map +0 -1
  244. package/dist/prebuilt/tool_node.d.ts +0 -182
  245. package/dist/prebuilt/tool_node.d.ts.map +0 -1
  246. package/dist/prebuilt/tool_node.js +0 -246
  247. package/dist/prebuilt/tool_node.js.map +0 -1
  248. package/dist/pregel/algo.cjs +0 -573
  249. package/dist/pregel/algo.cjs.map +0 -1
  250. package/dist/pregel/algo.d.cts +0 -13
  251. package/dist/pregel/algo.d.cts.map +0 -1
  252. package/dist/pregel/algo.d.ts +0 -13
  253. package/dist/pregel/algo.d.ts.map +0 -1
  254. package/dist/pregel/algo.js +0 -567
  255. package/dist/pregel/algo.js.map +0 -1
  256. package/dist/pregel/call.cjs +0 -53
  257. package/dist/pregel/call.cjs.map +0 -1
  258. package/dist/pregel/call.js +0 -50
  259. package/dist/pregel/call.js.map +0 -1
  260. package/dist/pregel/debug.cjs +0 -175
  261. package/dist/pregel/debug.cjs.map +0 -1
  262. package/dist/pregel/debug.js +0 -169
  263. package/dist/pregel/debug.js.map +0 -1
  264. package/dist/pregel/index.cjs +0 -1170
  265. package/dist/pregel/index.cjs.map +0 -1
  266. package/dist/pregel/index.d.cts +0 -546
  267. package/dist/pregel/index.d.cts.map +0 -1
  268. package/dist/pregel/index.d.ts +0 -546
  269. package/dist/pregel/index.d.ts.map +0 -1
  270. package/dist/pregel/index.js +0 -1168
  271. package/dist/pregel/index.js.map +0 -1
  272. package/dist/pregel/io.cjs +0 -140
  273. package/dist/pregel/io.cjs.map +0 -1
  274. package/dist/pregel/io.js +0 -135
  275. package/dist/pregel/io.js.map +0 -1
  276. package/dist/pregel/loop.cjs +0 -548
  277. package/dist/pregel/loop.cjs.map +0 -1
  278. package/dist/pregel/loop.js +0 -547
  279. package/dist/pregel/loop.js.map +0 -1
  280. package/dist/pregel/messages.cjs +0 -108
  281. package/dist/pregel/messages.cjs.map +0 -1
  282. package/dist/pregel/messages.js +0 -107
  283. package/dist/pregel/messages.js.map +0 -1
  284. package/dist/pregel/read.cjs +0 -152
  285. package/dist/pregel/read.cjs.map +0 -1
  286. package/dist/pregel/read.d.cts +0 -51
  287. package/dist/pregel/read.d.cts.map +0 -1
  288. package/dist/pregel/read.d.ts +0 -51
  289. package/dist/pregel/read.d.ts.map +0 -1
  290. package/dist/pregel/read.js +0 -150
  291. package/dist/pregel/read.js.map +0 -1
  292. package/dist/pregel/remote.cjs +0 -349
  293. package/dist/pregel/remote.cjs.map +0 -1
  294. package/dist/pregel/remote.d.cts +0 -123
  295. package/dist/pregel/remote.d.cts.map +0 -1
  296. package/dist/pregel/remote.d.ts +0 -123
  297. package/dist/pregel/remote.d.ts.map +0 -1
  298. package/dist/pregel/remote.js +0 -348
  299. package/dist/pregel/remote.js.map +0 -1
  300. package/dist/pregel/retry.cjs +0 -92
  301. package/dist/pregel/retry.cjs.map +0 -1
  302. package/dist/pregel/retry.js +0 -92
  303. package/dist/pregel/retry.js.map +0 -1
  304. package/dist/pregel/runnable_types.d.cts +0 -49
  305. package/dist/pregel/runnable_types.d.cts.map +0 -1
  306. package/dist/pregel/runnable_types.d.ts +0 -49
  307. package/dist/pregel/runnable_types.d.ts.map +0 -1
  308. package/dist/pregel/runner.cjs +0 -228
  309. package/dist/pregel/runner.cjs.map +0 -1
  310. package/dist/pregel/runner.js +0 -228
  311. package/dist/pregel/runner.js.map +0 -1
  312. package/dist/pregel/stream.cjs +0 -199
  313. package/dist/pregel/stream.cjs.map +0 -1
  314. package/dist/pregel/stream.js +0 -196
  315. package/dist/pregel/stream.js.map +0 -1
  316. package/dist/pregel/types.cjs +0 -27
  317. package/dist/pregel/types.cjs.map +0 -1
  318. package/dist/pregel/types.d.cts +0 -453
  319. package/dist/pregel/types.d.cts.map +0 -1
  320. package/dist/pregel/types.d.ts +0 -453
  321. package/dist/pregel/types.d.ts.map +0 -1
  322. package/dist/pregel/types.js +0 -25
  323. package/dist/pregel/types.js.map +0 -1
  324. package/dist/pregel/utils/config.cjs +0 -122
  325. package/dist/pregel/utils/config.cjs.map +0 -1
  326. package/dist/pregel/utils/config.d.cts +0 -36
  327. package/dist/pregel/utils/config.d.cts.map +0 -1
  328. package/dist/pregel/utils/config.d.ts +0 -36
  329. package/dist/pregel/utils/config.d.ts.map +0 -1
  330. package/dist/pregel/utils/config.js +0 -115
  331. package/dist/pregel/utils/config.js.map +0 -1
  332. package/dist/pregel/utils/index.cjs +0 -103
  333. package/dist/pregel/utils/index.cjs.map +0 -1
  334. package/dist/pregel/utils/index.d.cts +0 -51
  335. package/dist/pregel/utils/index.d.cts.map +0 -1
  336. package/dist/pregel/utils/index.d.ts +0 -51
  337. package/dist/pregel/utils/index.d.ts.map +0 -1
  338. package/dist/pregel/utils/index.js +0 -97
  339. package/dist/pregel/utils/index.js.map +0 -1
  340. package/dist/pregel/utils/subgraph.cjs +0 -19
  341. package/dist/pregel/utils/subgraph.cjs.map +0 -1
  342. package/dist/pregel/utils/subgraph.js +0 -17
  343. package/dist/pregel/utils/subgraph.js.map +0 -1
  344. package/dist/pregel/validate.cjs +0 -45
  345. package/dist/pregel/validate.cjs.map +0 -1
  346. package/dist/pregel/validate.js +0 -44
  347. package/dist/pregel/validate.js.map +0 -1
  348. package/dist/pregel/write.cjs +0 -97
  349. package/dist/pregel/write.cjs.map +0 -1
  350. package/dist/pregel/write.d.cts +0 -35
  351. package/dist/pregel/write.d.cts.map +0 -1
  352. package/dist/pregel/write.d.ts +0 -35
  353. package/dist/pregel/write.d.ts.map +0 -1
  354. package/dist/pregel/write.js +0 -95
  355. package/dist/pregel/write.js.map +0 -1
  356. package/dist/remote.cjs +0 -3
  357. package/dist/remote.d.cts +0 -2
  358. package/dist/remote.d.ts +0 -2
  359. package/dist/remote.js +0 -3
  360. package/dist/setup/async_local_storage.cjs +0 -12
  361. package/dist/setup/async_local_storage.cjs.map +0 -1
  362. package/dist/setup/async_local_storage.js +0 -11
  363. package/dist/setup/async_local_storage.js.map +0 -1
  364. package/dist/utils.cjs +0 -88
  365. package/dist/utils.cjs.map +0 -1
  366. package/dist/utils.d.cts +0 -32
  367. package/dist/utils.d.cts.map +0 -1
  368. package/dist/utils.d.ts +0 -32
  369. package/dist/utils.d.ts.map +0 -1
  370. package/dist/utils.js +0 -81
  371. package/dist/utils.js.map +0 -1
  372. package/dist/web.cjs +0 -96
  373. package/dist/web.d.cts +0 -23
  374. package/dist/web.d.ts +0 -23
  375. package/dist/web.js +0 -15
  376. package/dist/writer.cjs +0 -15
  377. package/dist/writer.cjs.map +0 -1
  378. package/dist/writer.d.cts +0 -7
  379. package/dist/writer.d.cts.map +0 -1
  380. package/dist/writer.d.ts +0 -7
  381. package/dist/writer.d.ts.map +0 -1
  382. package/dist/writer.js +0 -14
  383. package/dist/writer.js.map +0 -1
@@ -1 +0,0 @@
1
- {"version":3,"file":"loop.cjs","names":["IterableReadableWritableStream","BaseCache","START","CONFIG_KEY_RESUMING","isCommand","increment","CONFIG_KEY_STREAM","CONFIG_KEY_SCRATCHPAD","patchConfigurable","CONFIG_KEY_CHECKPOINT_NS","CHECKPOINT_NAMESPACE_SEPARATOR","CONFIG_KEY_READ","CONFIG_KEY_CHECKPOINT_MAP","saved: CheckpointTuple","emptyChannels","AsyncBatchedStore","WRITES_IDX_MAP","CONFIG_KEY_CHECKPOINT_ID","ERROR","INTERRUPT","TAG_HIDDEN","PUSH","gatherIteratorSync","prefixGenerator","mapOutputUpdates","mapDebugTaskResults","matched: {\n task: PregelExecutableTask<string, string>;\n result: unknown;\n }[]","keys: CacheFullKey[]","keyMap: Record<string, PregelExecutableTask<string, string>>","GraphInterrupt","_applyWrites","gatherIterator","mapOutputValues","shouldInterrupt","_prepareNextTasks","mapDebugCheckpoint","RESUME","mapDebugTasks","CHECKPOINT_NAMESPACE_END","readChannels","isGraphInterrupt","_prepareSingleTask","task","NULL_TASK_ID","isXXH3","CONFIG_KEY_RESUME_MAP","writes: { [key: string]: PendingWrite[] }","mapCommand","EmptyInputError","INPUT","mapInput","getNewChannelVersions","createCheckpoint","byTask: Record<string, PendingWrite<string>[]>"],"sources":["../../src/pregel/loop.ts"],"sourcesContent":["import type { RunnableConfig } from \"@langchain/core/runnables\";\nimport type { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport {\n BaseCheckpointSaver,\n Checkpoint,\n CheckpointTuple,\n copyCheckpoint,\n emptyCheckpoint,\n PendingWrite,\n CheckpointPendingWrite,\n CheckpointMetadata,\n All,\n BaseStore,\n AsyncBatchedStore,\n WRITES_IDX_MAP,\n BaseCache,\n CacheFullKey,\n CacheNamespace,\n} from \"@langchain/langgraph-checkpoint\";\n\nimport {\n BaseChannel,\n createCheckpoint,\n emptyChannels,\n} from \"../channels/base.js\";\nimport type {\n Call,\n CallTaskPath,\n Durability,\n PregelExecutableTask,\n PregelScratchpad,\n StreamMode,\n} from \"./types.js\";\nimport {\n isCommand,\n CHECKPOINT_NAMESPACE_SEPARATOR,\n Command,\n CONFIG_KEY_CHECKPOINT_MAP,\n CONFIG_KEY_READ,\n CONFIG_KEY_RESUMING,\n CONFIG_KEY_STREAM,\n ERROR,\n INPUT,\n INTERRUPT,\n NULL_TASK_ID,\n RESUME,\n TAG_HIDDEN,\n PUSH,\n CONFIG_KEY_SCRATCHPAD,\n CONFIG_KEY_CHECKPOINT_NS,\n CHECKPOINT_NAMESPACE_END,\n CONFIG_KEY_CHECKPOINT_ID,\n CONFIG_KEY_RESUME_MAP,\n START,\n} from \"../constants.js\";\nimport {\n _applyWrites,\n _prepareNextTasks,\n _prepareSingleTask,\n increment,\n shouldInterrupt,\n WritesProtocol,\n} from \"./algo.js\";\nimport {\n gatherIterator,\n gatherIteratorSync,\n prefixGenerator,\n} from \"../utils.js\";\nimport {\n mapCommand,\n mapInput,\n mapOutputUpdates,\n mapOutputValues,\n readChannels,\n} from \"./io.js\";\nimport {\n EmptyInputError,\n GraphInterrupt,\n isGraphInterrupt,\n} from \"../errors.js\";\nimport { getNewChannelVersions, patchConfigurable } from \"./utils/index.js\";\nimport {\n mapDebugTasks,\n mapDebugCheckpoint,\n mapDebugTaskResults,\n printStepTasks,\n} from \"./debug.js\";\nimport { PregelNode } from \"./read.js\";\nimport { LangGraphRunnableConfig } from \"./runnable_types.js\";\nimport { IterableReadableWritableStream, StreamChunk } from \"./stream.js\";\nimport { isXXH3 } from \"../hash.js\";\n\nconst INPUT_DONE = Symbol.for(\"INPUT_DONE\");\nconst INPUT_RESUMING = Symbol.for(\"INPUT_RESUMING\");\nconst DEFAULT_LOOP_LIMIT = 25;\n\nexport type PregelLoopInitializeParams = {\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n input?: any | Command;\n config: RunnableConfig;\n checkpointer?: BaseCheckpointSaver;\n outputKeys: string | string[];\n streamKeys: string | string[];\n nodes: Record<string, PregelNode>;\n channelSpecs: Record<string, BaseChannel>;\n stream: IterableReadableWritableStream;\n store?: BaseStore;\n cache?: BaseCache<PendingWrite<string>[]>;\n interruptAfter: string[] | All;\n interruptBefore: string[] | All;\n durability: Durability;\n manager?: CallbackManagerForChainRun;\n debug: boolean;\n triggerToNodes: Record<string, string[]>;\n};\n\ntype PregelLoopParams = {\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n input?: any | Command;\n config: RunnableConfig;\n checkpointer?: BaseCheckpointSaver;\n checkpoint: Checkpoint;\n checkpointMetadata: CheckpointMetadata;\n checkpointPreviousVersions: Record<string, string | number>;\n checkpointPendingWrites: CheckpointPendingWrite[];\n checkpointConfig: RunnableConfig;\n channels: Record<string, BaseChannel>;\n step: number;\n stop: number;\n outputKeys: string | string[];\n streamKeys: string | string[];\n nodes: Record<string, PregelNode>;\n checkpointNamespace: string[];\n skipDoneTasks: boolean;\n isNested: boolean;\n manager?: CallbackManagerForChainRun;\n stream: IterableReadableWritableStream;\n store?: AsyncBatchedStore;\n cache?: BaseCache<PendingWrite<string>[]>;\n prevCheckpointConfig: RunnableConfig | undefined;\n interruptAfter: string[] | All;\n interruptBefore: string[] | All;\n durability: Durability;\n debug: boolean;\n triggerToNodes: Record<string, string[]>;\n};\n\nfunction createDuplexStream(...streams: IterableReadableWritableStream[]) {\n return new IterableReadableWritableStream({\n passthroughFn: (value: StreamChunk) => {\n for (const stream of streams) {\n if (stream.modes.has(value[1])) {\n stream.push(value);\n }\n }\n },\n modes: new Set(streams.flatMap((s) => Array.from(s.modes))),\n });\n}\n\nclass AsyncBatchedCache extends BaseCache<PendingWrite<string>[]> {\n protected cache: BaseCache<PendingWrite<string>[]>;\n\n private queue: Promise<unknown> = Promise.resolve();\n\n constructor(cache: BaseCache<unknown>) {\n super();\n this.cache = cache as BaseCache<PendingWrite<string>[]>;\n }\n\n async get(keys: CacheFullKey[]) {\n return this.enqueueOperation(\"get\", keys);\n }\n\n async set(\n pairs: {\n key: CacheFullKey;\n value: PendingWrite<string>[];\n ttl?: number;\n }[]\n ) {\n return this.enqueueOperation(\"set\", pairs);\n }\n\n async clear(namespaces: CacheNamespace[]) {\n return this.enqueueOperation(\"clear\", namespaces);\n }\n\n async stop() {\n await this.queue;\n }\n\n private enqueueOperation<Type extends \"get\" | \"set\" | \"clear\">(\n type: Type,\n ...args: Parameters<(typeof this.cache)[Type]>\n ) {\n const newPromise = this.queue.then(() => {\n // @ts-expect-error Tuple type warning\n return this.cache[type](...args) as Promise<\n ReturnType<(typeof this.cache)[Type]>\n >;\n });\n\n this.queue = newPromise.then(\n () => void 0,\n () => void 0\n );\n\n return newPromise;\n }\n}\n\nexport class PregelLoop {\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n protected input?: any | Command;\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n output: any;\n\n config: LangGraphRunnableConfig;\n\n protected checkpointer?: BaseCheckpointSaver;\n\n protected checkpointerGetNextVersion: (current: number | undefined) => number;\n\n channels: Record<string, BaseChannel>;\n\n protected checkpoint: Checkpoint;\n\n protected checkpointIdSaved: string | undefined;\n\n protected checkpointConfig: RunnableConfig;\n\n checkpointMetadata: CheckpointMetadata;\n\n protected checkpointNamespace: string[];\n\n protected checkpointPendingWrites: CheckpointPendingWrite[] = [];\n\n protected checkpointPreviousVersions: Record<string, string | number>;\n\n step: number;\n\n protected stop: number;\n\n protected durability: Durability;\n\n protected outputKeys: string | string[];\n\n protected streamKeys: string | string[];\n\n protected nodes: Record<string, PregelNode>;\n\n protected skipDoneTasks: boolean;\n\n protected prevCheckpointConfig: RunnableConfig | undefined;\n\n protected updatedChannels: Set<string> | undefined;\n\n status:\n | \"pending\"\n | \"done\"\n | \"interrupt_before\"\n | \"interrupt_after\"\n | \"out_of_steps\" = \"pending\";\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n tasks: Record<string, PregelExecutableTask<any, any>> = {};\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n stream: IterableReadableWritableStream;\n\n checkpointerPromises: Promise<unknown>[] = [];\n\n isNested: boolean;\n\n protected _checkpointerChainedPromise: Promise<unknown> = Promise.resolve();\n\n store?: AsyncBatchedStore;\n\n cache?: AsyncBatchedCache;\n\n manager?: CallbackManagerForChainRun;\n\n interruptAfter: string[] | All;\n\n interruptBefore: string[] | All;\n\n toInterrupt: PregelExecutableTask<string, string>[] = [];\n\n debug: boolean = false;\n\n triggerToNodes: Record<string, string[]>;\n\n get isResuming() {\n let hasChannelVersions = false;\n if (START in this.checkpoint.channel_versions) {\n // For common channels, we can short-circuit the check\n hasChannelVersions = true;\n } else {\n for (const chan in this.checkpoint.channel_versions) {\n if (\n Object.prototype.hasOwnProperty.call(\n this.checkpoint.channel_versions,\n chan\n )\n ) {\n hasChannelVersions = true;\n break;\n }\n }\n }\n\n const configHasResumingFlag =\n this.config.configurable?.[CONFIG_KEY_RESUMING] !== undefined;\n const configIsResuming =\n configHasResumingFlag && this.config.configurable?.[CONFIG_KEY_RESUMING];\n\n const inputIsNullOrUndefined =\n this.input === null || this.input === undefined;\n const inputIsCommandResuming =\n isCommand(this.input) && this.input.resume != null;\n const inputIsResuming = this.input === INPUT_RESUMING;\n\n const runIdMatchesPrevious =\n !this.isNested &&\n this.config.metadata?.run_id !== undefined &&\n (this.checkpointMetadata as { run_id?: unknown })?.run_id !== undefined &&\n this.config.metadata.run_id ===\n (this.checkpointMetadata as { run_id?: unknown })?.run_id;\n\n return (\n hasChannelVersions &&\n (configIsResuming ||\n inputIsNullOrUndefined ||\n inputIsCommandResuming ||\n inputIsResuming ||\n runIdMatchesPrevious)\n );\n }\n\n constructor(params: PregelLoopParams) {\n this.input = params.input;\n this.checkpointer = params.checkpointer;\n // TODO: if managed values no longer needs graph we can replace with\n // managed_specs, channel_specs\n if (this.checkpointer !== undefined) {\n this.checkpointerGetNextVersion = this.checkpointer.getNextVersion.bind(\n this.checkpointer\n );\n } else {\n this.checkpointerGetNextVersion = increment;\n }\n this.checkpoint = params.checkpoint;\n this.checkpointMetadata = params.checkpointMetadata;\n this.checkpointPreviousVersions = params.checkpointPreviousVersions;\n this.channels = params.channels;\n this.checkpointPendingWrites = params.checkpointPendingWrites;\n this.step = params.step;\n this.stop = params.stop;\n this.config = params.config;\n this.checkpointConfig = params.checkpointConfig;\n this.isNested = params.isNested;\n this.manager = params.manager;\n this.outputKeys = params.outputKeys;\n this.streamKeys = params.streamKeys;\n this.nodes = params.nodes;\n this.skipDoneTasks = params.skipDoneTasks;\n this.store = params.store;\n this.cache = params.cache ? new AsyncBatchedCache(params.cache) : undefined;\n this.stream = params.stream;\n this.checkpointNamespace = params.checkpointNamespace;\n this.prevCheckpointConfig = params.prevCheckpointConfig;\n this.interruptAfter = params.interruptAfter;\n this.interruptBefore = params.interruptBefore;\n this.durability = params.durability;\n this.debug = params.debug;\n this.triggerToNodes = params.triggerToNodes;\n }\n\n static async initialize(params: PregelLoopInitializeParams) {\n let { config, stream } = params;\n if (\n stream !== undefined &&\n config.configurable?.[CONFIG_KEY_STREAM] !== undefined\n ) {\n stream = createDuplexStream(\n stream,\n config.configurable[CONFIG_KEY_STREAM]\n );\n }\n const skipDoneTasks = config.configurable\n ? !(\"checkpoint_id\" in config.configurable)\n : true;\n\n const scratchpad = config.configurable?.[CONFIG_KEY_SCRATCHPAD] as\n | PregelScratchpad\n | undefined;\n\n if (config.configurable && scratchpad) {\n if (scratchpad.subgraphCounter > 0) {\n config = patchConfigurable(config, {\n [CONFIG_KEY_CHECKPOINT_NS]: [\n config.configurable[CONFIG_KEY_CHECKPOINT_NS],\n scratchpad.subgraphCounter.toString(),\n ].join(CHECKPOINT_NAMESPACE_SEPARATOR),\n });\n }\n\n scratchpad.subgraphCounter += 1;\n }\n\n const isNested = CONFIG_KEY_READ in (config.configurable ?? {});\n if (\n !isNested &&\n config.configurable?.checkpoint_ns !== undefined &&\n config.configurable?.checkpoint_ns !== \"\"\n ) {\n config = patchConfigurable(config, {\n checkpoint_ns: \"\",\n checkpoint_id: undefined,\n });\n }\n let checkpointConfig = config;\n if (\n config.configurable?.[CONFIG_KEY_CHECKPOINT_MAP] !== undefined &&\n config.configurable?.[CONFIG_KEY_CHECKPOINT_MAP]?.[\n config.configurable?.checkpoint_ns\n ]\n ) {\n checkpointConfig = patchConfigurable(config, {\n checkpoint_id:\n config.configurable[CONFIG_KEY_CHECKPOINT_MAP][\n config.configurable?.checkpoint_ns\n ],\n });\n }\n const checkpointNamespace =\n config.configurable?.checkpoint_ns?.split(\n CHECKPOINT_NAMESPACE_SEPARATOR\n ) ?? [];\n\n const saved: CheckpointTuple = (await params.checkpointer?.getTuple(\n checkpointConfig\n )) ?? {\n config,\n checkpoint: emptyCheckpoint(),\n metadata: { source: \"input\", step: -2, parents: {} },\n pendingWrites: [],\n };\n checkpointConfig = {\n ...config,\n ...saved.config,\n configurable: {\n checkpoint_ns: \"\",\n ...config.configurable,\n ...saved.config.configurable,\n },\n };\n const prevCheckpointConfig = saved.parentConfig;\n const checkpoint = copyCheckpoint(saved.checkpoint);\n const checkpointMetadata = { ...saved.metadata } as CheckpointMetadata;\n const checkpointPendingWrites = saved.pendingWrites ?? [];\n\n const channels = emptyChannels(params.channelSpecs, checkpoint);\n\n const step = (checkpointMetadata.step ?? 0) + 1;\n const stop = step + (config.recursionLimit ?? DEFAULT_LOOP_LIMIT) + 1;\n const checkpointPreviousVersions = { ...checkpoint.channel_versions };\n\n const store = params.store\n ? new AsyncBatchedStore(params.store)\n : undefined;\n\n if (store) {\n // Start the store. This is a batch store, so it will run continuously\n await store.start();\n }\n return new PregelLoop({\n input: params.input,\n config,\n checkpointer: params.checkpointer,\n checkpoint,\n checkpointMetadata,\n checkpointConfig,\n prevCheckpointConfig,\n checkpointNamespace,\n channels,\n isNested,\n manager: params.manager,\n skipDoneTasks,\n step,\n stop,\n checkpointPreviousVersions,\n checkpointPendingWrites,\n outputKeys: params.outputKeys ?? [],\n streamKeys: params.streamKeys ?? [],\n nodes: params.nodes,\n stream,\n store,\n cache: params.cache,\n interruptAfter: params.interruptAfter,\n interruptBefore: params.interruptBefore,\n durability: params.durability,\n debug: params.debug,\n triggerToNodes: params.triggerToNodes,\n });\n }\n\n protected _checkpointerPutAfterPrevious(input: {\n config: RunnableConfig;\n checkpoint: Checkpoint;\n metadata: CheckpointMetadata;\n newVersions: Record<string, string | number>;\n }) {\n this._checkpointerChainedPromise = this._checkpointerChainedPromise.then(\n () => {\n return this.checkpointer?.put(\n input.config,\n input.checkpoint,\n input.metadata,\n input.newVersions\n );\n }\n );\n this.checkpointerPromises.push(this._checkpointerChainedPromise);\n }\n\n /**\n * Put writes for a task, to be read by the next tick.\n * @param taskId\n * @param writes\n */\n putWrites(taskId: string, writes: PendingWrite<string>[]) {\n let writesCopy = writes;\n if (writesCopy.length === 0) return;\n\n // deduplicate writes to special channels, last write wins\n if (writesCopy.every(([key]) => key in WRITES_IDX_MAP)) {\n writesCopy = Array.from(\n new Map(writesCopy.map((w) => [w[0], w])).values()\n );\n }\n\n // remove existing writes for this task\n this.checkpointPendingWrites = this.checkpointPendingWrites.filter(\n (w) => w[0] !== taskId\n );\n\n // save writes\n for (const [c, v] of writesCopy) {\n this.checkpointPendingWrites.push([taskId, c, v]);\n }\n\n const config = patchConfigurable(this.checkpointConfig, {\n [CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? \"\",\n [CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id,\n });\n\n if (this.durability !== \"exit\" && this.checkpointer != null) {\n this.checkpointerPromises.push(\n this.checkpointer.putWrites(config, writesCopy, taskId)\n );\n }\n\n if (this.tasks) {\n this._outputWrites(taskId, writesCopy);\n }\n\n if (!writes.length || !this.cache || !this.tasks) {\n return;\n }\n\n // only cache tasks with a cache key\n const task = this.tasks[taskId];\n if (task == null || task.cache_key == null) {\n return;\n }\n\n // only cache successful tasks\n if (writes[0][0] === ERROR || writes[0][0] === INTERRUPT) {\n return;\n }\n\n void this.cache.set([\n {\n key: [task.cache_key.ns, task.cache_key.key],\n value: task.writes,\n ttl: task.cache_key.ttl,\n },\n ]);\n }\n\n _outputWrites(taskId: string, writes: [string, unknown][], cached = false) {\n const task = this.tasks[taskId];\n if (task !== undefined) {\n if (\n task.config !== undefined &&\n (task.config.tags ?? []).includes(TAG_HIDDEN)\n ) {\n return;\n }\n\n if (writes.length > 0) {\n if (writes[0][0] === INTERRUPT) {\n // in `algo.ts` we append a bool to the task path to indicate\n // whether or not a call was present. If so, we don't emit the\n // the interrupt as it'll be emitted by the parent.\n if (task.path?.[0] === PUSH && task.path?.at(-1) === true) return;\n\n const interruptWrites = writes\n .filter((w) => w[0] === INTERRUPT)\n .flatMap((w) => w[1] as string[]);\n\n this._emit([\n [\"updates\", { [INTERRUPT]: interruptWrites }],\n [\"values\", { [INTERRUPT]: interruptWrites }],\n ]);\n } else if (writes[0][0] !== ERROR) {\n this._emit(\n gatherIteratorSync(\n prefixGenerator(\n mapOutputUpdates(this.outputKeys, [[task, writes]], cached),\n \"updates\"\n )\n )\n );\n }\n }\n if (!cached) {\n this._emit(\n gatherIteratorSync(\n prefixGenerator(\n mapDebugTaskResults([[task, writes]], this.streamKeys),\n \"tasks\"\n )\n )\n );\n }\n }\n }\n\n async _matchCachedWrites() {\n if (!this.cache) return [];\n\n const matched: {\n task: PregelExecutableTask<string, string>;\n result: unknown;\n }[] = [];\n\n const serializeKey = ([ns, key]: CacheFullKey) => {\n return `ns:${ns.join(\",\")}|key:${key}`;\n };\n\n const keys: CacheFullKey[] = [];\n const keyMap: Record<string, PregelExecutableTask<string, string>> = {};\n\n for (const task of Object.values(this.tasks)) {\n if (task.cache_key != null && !task.writes.length) {\n keys.push([task.cache_key.ns, task.cache_key.key]);\n keyMap[serializeKey([task.cache_key.ns, task.cache_key.key])] = task;\n }\n }\n\n if (keys.length === 0) return [];\n const cache = await this.cache.get(keys);\n\n for (const { key, value } of cache) {\n const task = keyMap[serializeKey(key)];\n if (task != null) {\n // update the task with the cached writes\n task.writes.push(...value);\n matched.push({ task, result: value });\n }\n }\n\n return matched;\n }\n\n /**\n * Execute a single iteration of the Pregel loop.\n * Returns true if more iterations are needed.\n * @param params\n */\n async tick(params: { inputKeys?: string | string[] }): Promise<boolean> {\n if (this.store && !this.store.isRunning) {\n await this.store?.start();\n }\n const { inputKeys = [] } = params;\n if (this.status !== \"pending\") {\n throw new Error(\n `Cannot tick when status is no longer \"pending\". Current status: \"${this.status}\"`\n );\n }\n if (![INPUT_DONE, INPUT_RESUMING].includes(this.input)) {\n await this._first(inputKeys);\n } else if (this.toInterrupt.length > 0) {\n this.status = \"interrupt_before\";\n throw new GraphInterrupt();\n } else if (\n Object.values(this.tasks).every((task) => task.writes.length > 0)\n ) {\n // finish superstep\n const writes = Object.values(this.tasks).flatMap((t) => t.writes);\n // All tasks have finished\n this.updatedChannels = _applyWrites(\n this.checkpoint,\n this.channels,\n Object.values(this.tasks),\n this.checkpointerGetNextVersion,\n this.triggerToNodes\n );\n // produce values output\n const valuesOutput = await gatherIterator(\n prefixGenerator(\n mapOutputValues(this.outputKeys, writes, this.channels),\n \"values\"\n )\n );\n this._emit(valuesOutput);\n // clear pending writes\n this.checkpointPendingWrites = [];\n await this._putCheckpoint({ source: \"loop\" });\n // after execution, check if we should interrupt\n if (\n shouldInterrupt(\n this.checkpoint,\n this.interruptAfter,\n Object.values(this.tasks)\n )\n ) {\n this.status = \"interrupt_after\";\n throw new GraphInterrupt();\n }\n\n // unset resuming flag\n if (this.config.configurable?.[CONFIG_KEY_RESUMING] !== undefined) {\n delete this.config.configurable?.[CONFIG_KEY_RESUMING];\n }\n } else {\n return false;\n }\n if (this.step > this.stop) {\n this.status = \"out_of_steps\";\n return false;\n }\n\n const nextTasks = _prepareNextTasks(\n this.checkpoint,\n this.checkpointPendingWrites,\n this.nodes,\n this.channels,\n this.config,\n true,\n {\n step: this.step,\n checkpointer: this.checkpointer,\n isResuming: this.isResuming,\n manager: this.manager,\n store: this.store,\n stream: this.stream,\n triggerToNodes: this.triggerToNodes,\n updatedChannels: this.updatedChannels,\n }\n );\n this.tasks = nextTasks;\n\n // Produce debug output\n if (this.checkpointer) {\n this._emit(\n await gatherIterator(\n prefixGenerator(\n mapDebugCheckpoint(\n this.checkpointConfig,\n this.channels,\n this.streamKeys,\n this.checkpointMetadata,\n Object.values(this.tasks),\n this.checkpointPendingWrites,\n this.prevCheckpointConfig,\n this.outputKeys\n ),\n \"checkpoints\"\n )\n )\n );\n }\n\n if (Object.values(this.tasks).length === 0) {\n this.status = \"done\";\n return false;\n }\n // if there are pending writes from a previous loop, apply them\n if (this.skipDoneTasks && this.checkpointPendingWrites.length > 0) {\n for (const [tid, k, v] of this.checkpointPendingWrites) {\n if (k === ERROR || k === INTERRUPT || k === RESUME) {\n continue;\n }\n const task = Object.values(this.tasks).find((t) => t.id === tid);\n if (task) {\n task.writes.push([k, v]);\n }\n }\n for (const task of Object.values(this.tasks)) {\n if (task.writes.length > 0) {\n this._outputWrites(task.id, task.writes, true);\n }\n }\n }\n // if all tasks have finished, re-tick\n if (Object.values(this.tasks).every((task) => task.writes.length > 0)) {\n return this.tick({ inputKeys });\n }\n\n // Before execution, check if we should interrupt\n if (\n shouldInterrupt(\n this.checkpoint,\n this.interruptBefore,\n Object.values(this.tasks)\n )\n ) {\n this.status = \"interrupt_before\";\n throw new GraphInterrupt();\n }\n\n // Produce debug output\n const debugOutput = await gatherIterator(\n prefixGenerator(mapDebugTasks(Object.values(this.tasks)), \"tasks\")\n );\n this._emit(debugOutput);\n\n return true;\n }\n\n async finishAndHandleError(error?: Error) {\n // persist current checkpoint and writes\n if (\n this.durability === \"exit\" &&\n // if it's a top graph\n (!this.isNested ||\n // or a nested graph with error or interrupt\n typeof error !== \"undefined\" ||\n // or a nested graph with checkpointer: true\n this.checkpointNamespace.every(\n (part) => !part.includes(CHECKPOINT_NAMESPACE_END)\n ))\n ) {\n this._putCheckpoint(this.checkpointMetadata);\n this._flushPendingWrites();\n }\n\n const suppress = this._suppressInterrupt(error);\n if (suppress || error === undefined) {\n this.output = readChannels(this.channels, this.outputKeys);\n }\n if (suppress) {\n // emit one last \"values\" event, with pending writes applied\n if (\n this.tasks !== undefined &&\n this.checkpointPendingWrites.length > 0 &&\n Object.values(this.tasks).some((task) => task.writes.length > 0)\n ) {\n this.updatedChannels = _applyWrites(\n this.checkpoint,\n this.channels,\n Object.values(this.tasks),\n this.checkpointerGetNextVersion,\n this.triggerToNodes\n );\n\n this._emit(\n gatherIteratorSync(\n prefixGenerator(\n mapOutputValues(\n this.outputKeys,\n Object.values(this.tasks).flatMap((t) => t.writes),\n this.channels\n ),\n \"values\"\n )\n )\n );\n }\n\n // Emit INTERRUPT event\n if (isGraphInterrupt(error) && !error.interrupts.length) {\n this._emit([\n [\"updates\", { [INTERRUPT]: [] }],\n [\"values\", { [INTERRUPT]: [] }],\n ]);\n }\n }\n return suppress;\n }\n\n async acceptPush(\n task: PregelExecutableTask<string, string>,\n writeIdx: number,\n call?: Call\n ): Promise<PregelExecutableTask<string, string> | void> {\n if (\n this.interruptAfter?.length > 0 &&\n shouldInterrupt(this.checkpoint, this.interruptAfter, [task])\n ) {\n this.toInterrupt.push(task);\n return;\n }\n\n const pushed = _prepareSingleTask(\n [PUSH, task.path ?? [], writeIdx, task.id, call] as CallTaskPath,\n this.checkpoint,\n this.checkpointPendingWrites,\n this.nodes,\n this.channels,\n task.config ?? {},\n true,\n {\n step: this.step,\n checkpointer: this.checkpointer,\n manager: this.manager,\n store: this.store,\n stream: this.stream,\n }\n );\n\n if (!pushed) return;\n if (\n this.interruptBefore?.length > 0 &&\n shouldInterrupt(this.checkpoint, this.interruptBefore, [pushed])\n ) {\n this.toInterrupt.push(pushed);\n return;\n }\n\n this._emit(\n gatherIteratorSync(prefixGenerator(mapDebugTasks([pushed]), \"tasks\"))\n );\n\n if (this.debug) printStepTasks(this.step, [pushed]);\n this.tasks[pushed.id] = pushed;\n if (this.skipDoneTasks) this._matchWrites({ [pushed.id]: pushed });\n\n const tasks = await this._matchCachedWrites();\n for (const { task } of tasks) {\n this._outputWrites(task.id, task.writes, true);\n }\n\n return pushed;\n }\n\n protected _suppressInterrupt(e?: Error): boolean {\n return isGraphInterrupt(e) && !this.isNested;\n }\n\n protected async _first(inputKeys: string | string[]) {\n /*\n * Resuming from previous checkpoint requires\n * - finding a previous checkpoint\n * - receiving null input (outer graph) or RESUMING flag (subgraph)\n */\n\n const { configurable } = this.config;\n\n // take resume value from parent\n const scratchpad = configurable?.[\n CONFIG_KEY_SCRATCHPAD\n ] as PregelScratchpad;\n\n if (scratchpad && scratchpad.nullResume !== undefined) {\n this.putWrites(NULL_TASK_ID, [[RESUME, scratchpad.nullResume]]);\n }\n\n // map command to writes\n if (isCommand(this.input)) {\n const hasResume = this.input.resume != null;\n\n if (\n this.input.resume != null &&\n typeof this.input.resume === \"object\" &&\n Object.keys(this.input.resume).every(isXXH3)\n ) {\n this.config.configurable ??= {};\n this.config.configurable[CONFIG_KEY_RESUME_MAP] = this.input.resume;\n }\n\n if (hasResume && this.checkpointer == null) {\n throw new Error(\"Cannot use Command(resume=...) without checkpointer\");\n }\n\n const writes: { [key: string]: PendingWrite[] } = {};\n\n // group writes by task id\n for (const [tid, key, value] of mapCommand(\n this.input,\n this.checkpointPendingWrites\n )) {\n writes[tid] ??= [];\n writes[tid].push([key, value]);\n }\n if (Object.keys(writes).length === 0) {\n throw new EmptyInputError(\"Received empty Command input\");\n }\n\n // save writes\n for (const [tid, ws] of Object.entries(writes)) {\n this.putWrites(tid, ws);\n }\n }\n\n // apply null writes\n const nullWrites = (this.checkpointPendingWrites ?? [])\n .filter((w) => w[0] === NULL_TASK_ID)\n .map((w) => w.slice(1)) as PendingWrite<string>[];\n if (nullWrites.length > 0) {\n _applyWrites(\n this.checkpoint,\n this.channels,\n [\n {\n name: INPUT,\n writes: nullWrites,\n triggers: [],\n },\n ],\n this.checkpointerGetNextVersion,\n this.triggerToNodes\n );\n }\n const isCommandUpdateOrGoto =\n isCommand(this.input) && nullWrites.length > 0;\n if (this.isResuming || isCommandUpdateOrGoto) {\n for (const channelName in this.channels) {\n if (!Object.prototype.hasOwnProperty.call(this.channels, channelName))\n continue;\n if (this.checkpoint.channel_versions[channelName] !== undefined) {\n const version = this.checkpoint.channel_versions[channelName];\n this.checkpoint.versions_seen[INTERRUPT] = {\n ...this.checkpoint.versions_seen[INTERRUPT],\n [channelName]: version,\n };\n }\n }\n // produce values output\n const valuesOutput = await gatherIterator(\n prefixGenerator(\n mapOutputValues(this.outputKeys, true, this.channels),\n \"values\"\n )\n );\n this._emit(valuesOutput);\n }\n if (this.isResuming) {\n this.input = INPUT_RESUMING;\n } else if (isCommandUpdateOrGoto) {\n // we need to create a new checkpoint for Command(update=...) or Command(goto=...)\n // in case the result of Command(goto=...) is an interrupt.\n // If not done, the checkpoint containing the interrupt will be lost.\n await this._putCheckpoint({ source: \"input\" });\n this.input = INPUT_DONE;\n } else {\n // map inputs to channel updates\n const inputWrites = await gatherIterator(mapInput(inputKeys, this.input));\n if (inputWrites.length > 0) {\n const discardTasks = _prepareNextTasks(\n this.checkpoint,\n this.checkpointPendingWrites,\n this.nodes,\n this.channels,\n this.config,\n true,\n { step: this.step }\n );\n this.updatedChannels = _applyWrites(\n this.checkpoint,\n this.channels,\n (Object.values(discardTasks) as WritesProtocol[]).concat([\n {\n name: INPUT,\n writes: inputWrites as PendingWrite[],\n triggers: [],\n },\n ]),\n this.checkpointerGetNextVersion,\n this.triggerToNodes\n );\n // save input checkpoint\n await this._putCheckpoint({ source: \"input\" });\n\n this.input = INPUT_DONE;\n } else if (!(CONFIG_KEY_RESUMING in (this.config.configurable ?? {}))) {\n throw new EmptyInputError(\n `Received no input writes for ${JSON.stringify(inputKeys, null, 2)}`\n );\n } else {\n // done with input\n this.input = INPUT_DONE;\n }\n }\n if (!this.isNested) {\n this.config = patchConfigurable(this.config, {\n [CONFIG_KEY_RESUMING]: this.isResuming,\n });\n }\n }\n\n protected _emit(values: [StreamMode, unknown][]) {\n for (const [mode, payload] of values) {\n if (this.stream.modes.has(mode)) {\n this.stream.push([this.checkpointNamespace, mode, payload]);\n }\n\n // debug mode is a \"checkpoints\" or \"tasks\" wrapped in an object\n // TODO: consider deprecating this in 1.x\n if (\n (mode === \"checkpoints\" || mode === \"tasks\") &&\n this.stream.modes.has(\"debug\")\n ) {\n const step = mode === \"checkpoints\" ? this.step - 1 : this.step;\n const timestamp = new Date().toISOString();\n const type = (() => {\n if (mode === \"checkpoints\") {\n return \"checkpoint\";\n } else if (\n typeof payload === \"object\" &&\n payload != null &&\n \"result\" in payload\n ) {\n return \"task_result\";\n } else {\n return \"task\";\n }\n })();\n\n this.stream.push([\n this.checkpointNamespace,\n \"debug\",\n { step, type, timestamp, payload },\n ]);\n }\n }\n }\n\n protected _putCheckpoint(\n inputMetadata: Omit<CheckpointMetadata, \"step\" | \"parents\">\n ) {\n const exiting = this.checkpointMetadata === inputMetadata;\n\n const doCheckpoint =\n this.checkpointer != null && (this.durability !== \"exit\" || exiting);\n\n const storeCheckpoint = (checkpoint: Checkpoint) => {\n // store the previous checkpoint config for debug events\n this.prevCheckpointConfig = this.checkpointConfig?.configurable\n ?.checkpoint_id\n ? this.checkpointConfig\n : undefined;\n\n // child graphs keep at most one checkpoint per parent checkpoint\n // this is achieved by writing child checkpoints as progress is made\n // (so that error recovery / resuming from interrupt don't lose work)\n // but doing so always with an id equal to that of the parent checkpoint\n this.checkpointConfig = patchConfigurable(this.checkpointConfig, {\n [CONFIG_KEY_CHECKPOINT_NS]:\n this.config.configurable?.checkpoint_ns ?? \"\",\n });\n\n const channelVersions = { ...this.checkpoint.channel_versions };\n const newVersions = getNewChannelVersions(\n this.checkpointPreviousVersions,\n channelVersions\n );\n this.checkpointPreviousVersions = channelVersions;\n // save it, without blocking\n // if there's a previous checkpoint save in progress, wait for it\n // ensuring checkpointers receive checkpoints in order\n void this._checkpointerPutAfterPrevious({\n config: { ...this.checkpointConfig },\n checkpoint: copyCheckpoint(checkpoint),\n metadata: { ...this.checkpointMetadata },\n newVersions,\n });\n this.checkpointConfig = {\n ...this.checkpointConfig,\n configurable: {\n ...this.checkpointConfig.configurable,\n checkpoint_id: this.checkpoint.id,\n },\n };\n };\n\n if (!exiting) {\n this.checkpointMetadata = {\n ...inputMetadata,\n step: this.step,\n parents: this.config.configurable?.[CONFIG_KEY_CHECKPOINT_MAP] ?? {},\n };\n }\n\n // create new checkpoint\n this.checkpoint = createCheckpoint(\n this.checkpoint,\n doCheckpoint ? this.channels : undefined,\n this.step,\n exiting ? { id: this.checkpoint.id } : undefined\n );\n\n // Bail if no checkpointer\n if (doCheckpoint) storeCheckpoint(this.checkpoint);\n\n if (!exiting) {\n // increment step\n this.step += 1;\n }\n }\n\n protected _flushPendingWrites() {\n if (this.checkpointer == null) return;\n if (this.checkpointPendingWrites.length === 0) return;\n\n // patch config\n const config = patchConfigurable(this.checkpointConfig, {\n [CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? \"\",\n [CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id,\n });\n\n // group writes by task id\n const byTask: Record<string, PendingWrite<string>[]> = {};\n for (const [tid, key, value] of this.checkpointPendingWrites) {\n byTask[tid] ??= [];\n byTask[tid].push([key, value]);\n }\n\n // submit writes to checkpointer\n for (const [tid, ws] of Object.entries(byTask)) {\n this.checkpointerPromises.push(\n this.checkpointer.putWrites(config, ws, tid)\n );\n }\n }\n\n protected _matchWrites(\n tasks: Record<string, PregelExecutableTask<string, string>>\n ) {\n for (const [tid, k, v] of this.checkpointPendingWrites) {\n if (k === ERROR || k === INTERRUPT || k === RESUME) {\n continue;\n }\n const task = Object.values(tasks).find((t) => t.id === tid);\n if (task) {\n task.writes.push([k, v]);\n }\n }\n for (const task of Object.values(tasks)) {\n if (task.writes.length > 0) {\n this._outputWrites(task.id, task.writes, true);\n }\n }\n }\n}\n"],"mappings":";;;;;;;;;;;;;;AA4FA,MAAM,aAAa,OAAO,IAAI;AAC9B,MAAM,iBAAiB,OAAO,IAAI;AAClC,MAAM,qBAAqB;AAqD3B,SAAS,mBAAmB,GAAG,SAA2C;AACxE,QAAO,IAAIA,8CAA+B;EACxC,gBAAgB,UAAuB;AACrC,QAAK,MAAM,UAAU,QACnB,KAAI,OAAO,MAAM,IAAI,MAAM,IACzB,QAAO,KAAK;;EAIlB,OAAO,IAAI,IAAI,QAAQ,SAAS,MAAM,MAAM,KAAK,EAAE;;;AAIvD,IAAM,oBAAN,cAAgCC,2CAAkC;CAChE,AAAU;CAEV,AAAQ,QAA0B,QAAQ;CAE1C,YAAY,OAA2B;AACrC;AACA,OAAK,QAAQ;;CAGf,MAAM,IAAI,MAAsB;AAC9B,SAAO,KAAK,iBAAiB,OAAO;;CAGtC,MAAM,IACJ,OAKA;AACA,SAAO,KAAK,iBAAiB,OAAO;;CAGtC,MAAM,MAAM,YAA8B;AACxC,SAAO,KAAK,iBAAiB,SAAS;;CAGxC,MAAM,OAAO;AACX,QAAM,KAAK;;CAGb,AAAQ,iBACN,MACA,GAAG,MACH;EACA,MAAM,aAAa,KAAK,MAAM,WAAW;AAEvC,UAAO,KAAK,MAAM,MAAM,GAAG;;AAK7B,OAAK,QAAQ,WAAW,WAChB,KAAK,SACL,KAAK;AAGb,SAAO;;;AAIX,IAAa,aAAb,MAAa,WAAW;CAEtB,AAAU;CAGV;CAEA;CAEA,AAAU;CAEV,AAAU;CAEV;CAEA,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV;CAEA,AAAU;CAEV,AAAU,0BAAoD;CAE9D,AAAU;CAEV;CAEA,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,AAAU;CAEV,SAKqB;CAGrB,QAAwD;CAGxD;CAEA,uBAA2C;CAE3C;CAEA,AAAU,8BAAgD,QAAQ;CAElE;CAEA;CAEA;CAEA;CAEA;CAEA,cAAsD;CAEtD,QAAiB;CAEjB;CAEA,IAAI,aAAa;EACf,IAAI,qBAAqB;AACzB,MAAIC,2BAAS,KAAK,WAAW,iBAE3B,sBAAqB;MAErB,MAAK,MAAM,QAAQ,KAAK,WAAW,iBACjC,KACE,OAAO,UAAU,eAAe,KAC9B,KAAK,WAAW,kBAChB,OAEF;AACA,wBAAqB;AACrB;;EAKN,MAAM,wBACJ,KAAK,OAAO,eAAeC,2CAAyB;EACtD,MAAM,mBACJ,yBAAyB,KAAK,OAAO,eAAeA;EAEtD,MAAM,yBACJ,KAAK,UAAU,QAAQ,KAAK,UAAU;EACxC,MAAM,yBACJC,4BAAU,KAAK,UAAU,KAAK,MAAM,UAAU;EAChD,MAAM,kBAAkB,KAAK,UAAU;EAEvC,MAAM,uBACJ,CAAC,KAAK,YACN,KAAK,OAAO,UAAU,WAAW,UAChC,KAAK,oBAA6C,WAAW,UAC9D,KAAK,OAAO,SAAS,WAClB,KAAK,oBAA6C;AAEvD,SACE,uBACC,oBACC,0BACA,0BACA,mBACA;;CAIN,YAAY,QAA0B;AACpC,OAAK,QAAQ,OAAO;AACpB,OAAK,eAAe,OAAO;AAG3B,MAAI,KAAK,iBAAiB,OACxB,MAAK,6BAA6B,KAAK,aAAa,eAAe,KACjE,KAAK;MAGP,MAAK,6BAA6BC;AAEpC,OAAK,aAAa,OAAO;AACzB,OAAK,qBAAqB,OAAO;AACjC,OAAK,6BAA6B,OAAO;AACzC,OAAK,WAAW,OAAO;AACvB,OAAK,0BAA0B,OAAO;AACtC,OAAK,OAAO,OAAO;AACnB,OAAK,OAAO,OAAO;AACnB,OAAK,SAAS,OAAO;AACrB,OAAK,mBAAmB,OAAO;AAC/B,OAAK,WAAW,OAAO;AACvB,OAAK,UAAU,OAAO;AACtB,OAAK,aAAa,OAAO;AACzB,OAAK,aAAa,OAAO;AACzB,OAAK,QAAQ,OAAO;AACpB,OAAK,gBAAgB,OAAO;AAC5B,OAAK,QAAQ,OAAO;AACpB,OAAK,QAAQ,OAAO,QAAQ,IAAI,kBAAkB,OAAO,SAAS;AAClE,OAAK,SAAS,OAAO;AACrB,OAAK,sBAAsB,OAAO;AAClC,OAAK,uBAAuB,OAAO;AACnC,OAAK,iBAAiB,OAAO;AAC7B,OAAK,kBAAkB,OAAO;AAC9B,OAAK,aAAa,OAAO;AACzB,OAAK,QAAQ,OAAO;AACpB,OAAK,iBAAiB,OAAO;;CAG/B,aAAa,WAAW,QAAoC;EAC1D,IAAI,EAAE,QAAQ,WAAW;AACzB,MACE,WAAW,UACX,OAAO,eAAeC,yCAAuB,OAE7C,UAAS,mBACP,QACA,OAAO,aAAaA;EAGxB,MAAM,gBAAgB,OAAO,eACzB,EAAE,mBAAmB,OAAO,gBAC5B;EAEJ,MAAM,aAAa,OAAO,eAAeC;AAIzC,MAAI,OAAO,gBAAgB,YAAY;AACrC,OAAI,WAAW,kBAAkB,EAC/B,UAASC,gCAAkB,QAAQ,GAChCC,6CAA2B,CAC1B,OAAO,aAAaA,6CACpB,WAAW,gBAAgB,YAC3B,KAAKC;AAIX,cAAW,mBAAmB;;EAGhC,MAAM,WAAWC,sCAAoB,OAAO,gBAAgB;AAC5D,MACE,CAAC,YACD,OAAO,cAAc,kBAAkB,UACvC,OAAO,cAAc,kBAAkB,GAEvC,UAASH,gCAAkB,QAAQ;GACjC,eAAe;GACf,eAAe;;EAGnB,IAAI,mBAAmB;AACvB,MACE,OAAO,eAAeI,iDAA+B,UACrD,OAAO,eAAeA,+CACpB,OAAO,cAAc,eAGvB,oBAAmBJ,gCAAkB,QAAQ,EAC3C,eACE,OAAO,aAAaI,6CAClB,OAAO,cAAc;EAI7B,MAAM,sBACJ,OAAO,cAAc,eAAe,MAClCF,qDACG;EAEP,MAAMG,QAA0B,MAAM,OAAO,cAAc,SACzD,qBACI;GACJ;GACA;GACA,UAAU;IAAE,QAAQ;IAAS,MAAM;IAAI,SAAS;;GAChD,eAAe;;AAEjB,qBAAmB;GACjB,GAAG;GACH,GAAG,MAAM;GACT,cAAc;IACZ,eAAe;IACf,GAAG,OAAO;IACV,GAAG,MAAM,OAAO;;;EAGpB,MAAM,uBAAuB,MAAM;EACnC,MAAM,kEAA4B,MAAM;EACxC,MAAM,qBAAqB,EAAE,GAAG,MAAM;EACtC,MAAM,0BAA0B,MAAM,iBAAiB;EAEvD,MAAM,WAAWC,2BAAc,OAAO,cAAc;EAEpD,MAAM,QAAQ,mBAAmB,QAAQ,KAAK;EAC9C,MAAM,OAAO,QAAQ,OAAO,kBAAkB,sBAAsB;EACpE,MAAM,6BAA6B,EAAE,GAAG,WAAW;EAEnD,MAAM,QAAQ,OAAO,QACjB,IAAIC,mDAAkB,OAAO,SAC7B;AAEJ,MAAI,MAEF,OAAM,MAAM;AAEd,SAAO,IAAI,WAAW;GACpB,OAAO,OAAO;GACd;GACA,cAAc,OAAO;GACrB;GACA;GACA;GACA;GACA;GACA;GACA;GACA,SAAS,OAAO;GAChB;GACA;GACA;GACA;GACA;GACA,YAAY,OAAO,cAAc;GACjC,YAAY,OAAO,cAAc;GACjC,OAAO,OAAO;GACd;GACA;GACA,OAAO,OAAO;GACd,gBAAgB,OAAO;GACvB,iBAAiB,OAAO;GACxB,YAAY,OAAO;GACnB,OAAO,OAAO;GACd,gBAAgB,OAAO;;;CAI3B,AAAU,8BAA8B,OAKrC;AACD,OAAK,8BAA8B,KAAK,4BAA4B,WAC5D;AACJ,UAAO,KAAK,cAAc,IACxB,MAAM,QACN,MAAM,YACN,MAAM,UACN,MAAM;;AAIZ,OAAK,qBAAqB,KAAK,KAAK;;;;;;;CAQtC,UAAU,QAAgB,QAAgC;EACxD,IAAI,aAAa;AACjB,MAAI,WAAW,WAAW,EAAG;AAG7B,MAAI,WAAW,OAAO,CAAC,SAAS,OAAOC,iDACrC,cAAa,MAAM,KACjB,IAAI,IAAI,WAAW,KAAK,MAAM,CAAC,EAAE,IAAI,KAAK;AAK9C,OAAK,0BAA0B,KAAK,wBAAwB,QACzD,MAAM,EAAE,OAAO;AAIlB,OAAK,MAAM,CAAC,GAAG,MAAM,WACnB,MAAK,wBAAwB,KAAK;GAAC;GAAQ;GAAG;;EAGhD,MAAM,SAASR,gCAAkB,KAAK,kBAAkB;IACrDC,6CAA2B,KAAK,OAAO,cAAc,iBAAiB;IACtEQ,6CAA2B,KAAK,WAAW;;AAG9C,MAAI,KAAK,eAAe,UAAU,KAAK,gBAAgB,KACrD,MAAK,qBAAqB,KACxB,KAAK,aAAa,UAAU,QAAQ,YAAY;AAIpD,MAAI,KAAK,MACP,MAAK,cAAc,QAAQ;AAG7B,MAAI,CAAC,OAAO,UAAU,CAAC,KAAK,SAAS,CAAC,KAAK,MACzC;EAIF,MAAM,OAAO,KAAK,MAAM;AACxB,MAAI,QAAQ,QAAQ,KAAK,aAAa,KACpC;AAIF,MAAI,OAAO,GAAG,OAAOC,2BAAS,OAAO,GAAG,OAAOC,4BAC7C;AAGF,EAAK,KAAK,MAAM,IAAI,CAClB;GACE,KAAK,CAAC,KAAK,UAAU,IAAI,KAAK,UAAU;GACxC,OAAO,KAAK;GACZ,KAAK,KAAK,UAAU;;;CAK1B,cAAc,QAAgB,QAA6B,SAAS,OAAO;EACzE,MAAM,OAAO,KAAK,MAAM;AACxB,MAAI,SAAS,QAAW;AACtB,OACE,KAAK,WAAW,WACf,KAAK,OAAO,QAAQ,IAAI,SAASC,8BAElC;AAGF,OAAI,OAAO,SAAS,GAClB;QAAI,OAAO,GAAG,OAAOD,6BAAW;AAI9B,SAAI,KAAK,OAAO,OAAOE,0BAAQ,KAAK,MAAM,GAAG,QAAQ,KAAM;KAE3D,MAAM,kBAAkB,OACrB,QAAQ,MAAM,EAAE,OAAOF,6BACvB,SAAS,MAAM,EAAE;AAEpB,UAAK,MAAM,CACT,CAAC,WAAW,GAAGA,8BAAY,oBAC3B,CAAC,UAAU,GAAGA,8BAAY;eAEnB,OAAO,GAAG,OAAOD,wBAC1B,MAAK,MACHI,iCACEC,8BACEC,4BAAiB,KAAK,YAAY,CAAC,CAAC,MAAM,UAAU,SACpD;;AAMV,OAAI,CAAC,OACH,MAAK,MACHF,iCACEC,8BACEE,kCAAoB,CAAC,CAAC,MAAM,UAAU,KAAK,aAC3C;;;CAQZ,MAAM,qBAAqB;AACzB,MAAI,CAAC,KAAK,MAAO,QAAO;EAExB,MAAMC,UAGA;EAEN,MAAM,gBAAgB,CAAC,IAAI,SAAuB;AAChD,UAAO,MAAM,GAAG,KAAK,KAAK,OAAO;;EAGnC,MAAMC,OAAuB;EAC7B,MAAMC,SAA+D;AAErE,OAAK,MAAM,QAAQ,OAAO,OAAO,KAAK,OACpC,KAAI,KAAK,aAAa,QAAQ,CAAC,KAAK,OAAO,QAAQ;AACjD,QAAK,KAAK,CAAC,KAAK,UAAU,IAAI,KAAK,UAAU;AAC7C,UAAO,aAAa,CAAC,KAAK,UAAU,IAAI,KAAK,UAAU,SAAS;;AAIpE,MAAI,KAAK,WAAW,EAAG,QAAO;EAC9B,MAAM,QAAQ,MAAM,KAAK,MAAM,IAAI;AAEnC,OAAK,MAAM,EAAE,KAAK,WAAW,OAAO;GAClC,MAAM,OAAO,OAAO,aAAa;AACjC,OAAI,QAAQ,MAAM;AAEhB,SAAK,OAAO,KAAK,GAAG;AACpB,YAAQ,KAAK;KAAE;KAAM,QAAQ;;;;AAIjC,SAAO;;;;;;;CAQT,MAAM,KAAK,QAA6D;AACtE,MAAI,KAAK,SAAS,CAAC,KAAK,MAAM,UAC5B,OAAM,KAAK,OAAO;EAEpB,MAAM,EAAE,YAAY,OAAO;AAC3B,MAAI,KAAK,WAAW,UAClB,OAAM,IAAI,MACR,oEAAoE,KAAK,OAAO;AAGpF,MAAI,CAAC,CAAC,YAAY,gBAAgB,SAAS,KAAK,OAC9C,OAAM,KAAK,OAAO;WACT,KAAK,YAAY,SAAS,GAAG;AACtC,QAAK,SAAS;AACd,SAAM,IAAIC;aAEV,OAAO,OAAO,KAAK,OAAO,OAAO,SAAS,KAAK,OAAO,SAAS,IAC/D;GAEA,MAAM,SAAS,OAAO,OAAO,KAAK,OAAO,SAAS,MAAM,EAAE;AAE1D,QAAK,kBAAkBC,0BACrB,KAAK,YACL,KAAK,UACL,OAAO,OAAO,KAAK,QACnB,KAAK,4BACL,KAAK;GAGP,MAAM,eAAe,MAAMC,6BACzBR,8BACES,2BAAgB,KAAK,YAAY,QAAQ,KAAK,WAC9C;AAGJ,QAAK,MAAM;AAEX,QAAK,0BAA0B;AAC/B,SAAM,KAAK,eAAe,EAAE,QAAQ;AAEpC,OACEC,6BACE,KAAK,YACL,KAAK,gBACL,OAAO,OAAO,KAAK,SAErB;AACA,SAAK,SAAS;AACd,UAAM,IAAIJ;;AAIZ,OAAI,KAAK,OAAO,eAAe1B,2CAAyB,OACtD,QAAO,KAAK,OAAO,eAAeA;QAGpC,QAAO;AAET,MAAI,KAAK,OAAO,KAAK,MAAM;AACzB,QAAK,SAAS;AACd,UAAO;;EAGT,MAAM,YAAY+B,+BAChB,KAAK,YACL,KAAK,yBACL,KAAK,OACL,KAAK,UACL,KAAK,QACL,MACA;GACE,MAAM,KAAK;GACX,cAAc,KAAK;GACnB,YAAY,KAAK;GACjB,SAAS,KAAK;GACd,OAAO,KAAK;GACZ,QAAQ,KAAK;GACb,gBAAgB,KAAK;GACrB,iBAAiB,KAAK;;AAG1B,OAAK,QAAQ;AAGb,MAAI,KAAK,aACP,MAAK,MACH,MAAMH,6BACJR,8BACEY,iCACE,KAAK,kBACL,KAAK,UACL,KAAK,YACL,KAAK,oBACL,OAAO,OAAO,KAAK,QACnB,KAAK,yBACL,KAAK,sBACL,KAAK,aAEP;AAMR,MAAI,OAAO,OAAO,KAAK,OAAO,WAAW,GAAG;AAC1C,QAAK,SAAS;AACd,UAAO;;AAGT,MAAI,KAAK,iBAAiB,KAAK,wBAAwB,SAAS,GAAG;AACjE,QAAK,MAAM,CAAC,KAAK,GAAG,MAAM,KAAK,yBAAyB;AACtD,QAAI,MAAMjB,2BAAS,MAAMC,+BAAa,MAAMiB,yBAC1C;IAEF,MAAM,OAAO,OAAO,OAAO,KAAK,OAAO,MAAM,MAAM,EAAE,OAAO;AAC5D,QAAI,KACF,MAAK,OAAO,KAAK,CAAC,GAAG;;AAGzB,QAAK,MAAM,QAAQ,OAAO,OAAO,KAAK,OACpC,KAAI,KAAK,OAAO,SAAS,EACvB,MAAK,cAAc,KAAK,IAAI,KAAK,QAAQ;;AAK/C,MAAI,OAAO,OAAO,KAAK,OAAO,OAAO,SAAS,KAAK,OAAO,SAAS,GACjE,QAAO,KAAK,KAAK,EAAE;AAIrB,MACEH,6BACE,KAAK,YACL,KAAK,iBACL,OAAO,OAAO,KAAK,SAErB;AACA,QAAK,SAAS;AACd,SAAM,IAAIJ;;EAIZ,MAAM,cAAc,MAAME,6BACxBR,8BAAgBc,4BAAc,OAAO,OAAO,KAAK,SAAS;AAE5D,OAAK,MAAM;AAEX,SAAO;;CAGT,MAAM,qBAAqB,OAAe;AAExC,MACE,KAAK,eAAe,WAEnB,CAAC,KAAK,YAEL,OAAO,UAAU,eAEjB,KAAK,oBAAoB,OACtB,SAAS,CAAC,KAAK,SAASC,+CAE7B;AACA,QAAK,eAAe,KAAK;AACzB,QAAK;;EAGP,MAAM,WAAW,KAAK,mBAAmB;AACzC,MAAI,YAAY,UAAU,OACxB,MAAK,SAASC,wBAAa,KAAK,UAAU,KAAK;AAEjD,MAAI,UAAU;AAEZ,OACE,KAAK,UAAU,UACf,KAAK,wBAAwB,SAAS,KACtC,OAAO,OAAO,KAAK,OAAO,MAAM,SAAS,KAAK,OAAO,SAAS,IAC9D;AACA,SAAK,kBAAkBT,0BACrB,KAAK,YACL,KAAK,UACL,OAAO,OAAO,KAAK,QACnB,KAAK,4BACL,KAAK;AAGP,SAAK,MACHR,iCACEC,8BACES,2BACE,KAAK,YACL,OAAO,OAAO,KAAK,OAAO,SAAS,MAAM,EAAE,SAC3C,KAAK,WAEP;;AAOR,OAAIQ,gCAAiB,UAAU,CAAC,MAAM,WAAW,OAC/C,MAAK,MAAM,CACT,CAAC,WAAW,GAAGrB,8BAAY,OAC3B,CAAC,UAAU,GAAGA,8BAAY;;AAIhC,SAAO;;CAGT,MAAM,WACJ,MACA,UACA,MACsD;AACtD,MACE,KAAK,gBAAgB,SAAS,KAC9Bc,6BAAgB,KAAK,YAAY,KAAK,gBAAgB,CAAC,QACvD;AACA,QAAK,YAAY,KAAK;AACtB;;EAGF,MAAM,SAASQ,gCACb;GAACpB;GAAM,KAAK,QAAQ;GAAI;GAAU,KAAK;GAAI;KAC3C,KAAK,YACL,KAAK,yBACL,KAAK,OACL,KAAK,UACL,KAAK,UAAU,IACf,MACA;GACE,MAAM,KAAK;GACX,cAAc,KAAK;GACnB,SAAS,KAAK;GACd,OAAO,KAAK;GACZ,QAAQ,KAAK;;AAIjB,MAAI,CAAC,OAAQ;AACb,MACE,KAAK,iBAAiB,SAAS,KAC/BY,6BAAgB,KAAK,YAAY,KAAK,iBAAiB,CAAC,UACxD;AACA,QAAK,YAAY,KAAK;AACtB;;AAGF,OAAK,MACHX,iCAAmBC,8BAAgBc,4BAAc,CAAC,UAAU;AAG9D,MAAI,KAAK,MAAO,8BAAe,KAAK,MAAM,CAAC;AAC3C,OAAK,MAAM,OAAO,MAAM;AACxB,MAAI,KAAK,cAAe,MAAK,aAAa,GAAG,OAAO,KAAK;EAEzD,MAAM,QAAQ,MAAM,KAAK;AACzB,OAAK,MAAM,EAAE,kBAAU,MACrB,MAAK,cAAcK,OAAK,IAAIA,OAAK,QAAQ;AAG3C,SAAO;;CAGT,AAAU,mBAAmB,GAAoB;AAC/C,SAAOF,gCAAiB,MAAM,CAAC,KAAK;;CAGtC,MAAgB,OAAO,WAA8B;EAOnD,MAAM,EAAE,iBAAiB,KAAK;EAG9B,MAAM,aAAa,eACjBjC;AAGF,MAAI,cAAc,WAAW,eAAe,OAC1C,MAAK,UAAUoC,gCAAc,CAAC,CAACP,0BAAQ,WAAW;AAIpD,MAAIhC,4BAAU,KAAK,QAAQ;GACzB,MAAM,YAAY,KAAK,MAAM,UAAU;AAEvC,OACE,KAAK,MAAM,UAAU,QACrB,OAAO,KAAK,MAAM,WAAW,YAC7B,OAAO,KAAK,KAAK,MAAM,QAAQ,MAAMwC,sBACrC;AACA,SAAK,OAAO,iBAAiB;AAC7B,SAAK,OAAO,aAAaC,2CAAyB,KAAK,MAAM;;AAG/D,OAAI,aAAa,KAAK,gBAAgB,KACpC,OAAM,IAAI,MAAM;GAGlB,MAAMC,SAA4C;AAGlD,QAAK,MAAM,CAAC,KAAK,KAAK,UAAUC,sBAC9B,KAAK,OACL,KAAK,0BACJ;AACD,WAAO,SAAS;AAChB,WAAO,KAAK,KAAK,CAAC,KAAK;;AAEzB,OAAI,OAAO,KAAK,QAAQ,WAAW,EACjC,OAAM,IAAIC,+BAAgB;AAI5B,QAAK,MAAM,CAAC,KAAK,OAAO,OAAO,QAAQ,QACrC,MAAK,UAAU,KAAK;;EAKxB,MAAM,cAAc,KAAK,2BAA2B,IACjD,QAAQ,MAAM,EAAE,OAAOL,gCACvB,KAAK,MAAM,EAAE,MAAM;AACtB,MAAI,WAAW,SAAS,EACtB,2BACE,KAAK,YACL,KAAK,UACL,CACE;GACE,MAAMM;GACN,QAAQ;GACR,UAAU;MAGd,KAAK,4BACL,KAAK;EAGT,MAAM,wBACJ7C,4BAAU,KAAK,UAAU,WAAW,SAAS;AAC/C,MAAI,KAAK,cAAc,uBAAuB;AAC5C,QAAK,MAAM,eAAe,KAAK,UAAU;AACvC,QAAI,CAAC,OAAO,UAAU,eAAe,KAAK,KAAK,UAAU,aACvD;AACF,QAAI,KAAK,WAAW,iBAAiB,iBAAiB,QAAW;KAC/D,MAAM,UAAU,KAAK,WAAW,iBAAiB;AACjD,UAAK,WAAW,cAAce,+BAAa;MACzC,GAAG,KAAK,WAAW,cAAcA;OAChC,cAAc;;;;GAKrB,MAAM,eAAe,MAAMY,6BACzBR,8BACES,2BAAgB,KAAK,YAAY,MAAM,KAAK,WAC5C;AAGJ,QAAK,MAAM;;AAEb,MAAI,KAAK,WACP,MAAK,QAAQ;WACJ,uBAAuB;AAIhC,SAAM,KAAK,eAAe,EAAE,QAAQ;AACpC,QAAK,QAAQ;SACR;GAEL,MAAM,cAAc,MAAMD,6BAAemB,oBAAS,WAAW,KAAK;AAClE,OAAI,YAAY,SAAS,GAAG;IAC1B,MAAM,eAAehB,+BACnB,KAAK,YACL,KAAK,yBACL,KAAK,OACL,KAAK,UACL,KAAK,QACL,MACA,EAAE,MAAM,KAAK;AAEf,SAAK,kBAAkBJ,0BACrB,KAAK,YACL,KAAK,UACJ,OAAO,OAAO,cAAmC,OAAO,CACvD;KACE,MAAMmB;KACN,QAAQ;KACR,UAAU;SAGd,KAAK,4BACL,KAAK;AAGP,UAAM,KAAK,eAAe,EAAE,QAAQ;AAEpC,SAAK,QAAQ;cACJ,EAAE9C,0CAAwB,KAAK,OAAO,gBAAgB,KAC/D,OAAM,IAAI6C,+BACR,gCAAgC,KAAK,UAAU,WAAW,MAAM;OAIlE,MAAK,QAAQ;;AAGjB,MAAI,CAAC,KAAK,SACR,MAAK,SAASxC,gCAAkB,KAAK,QAAQ,GAC1CL,wCAAsB,KAAK;;CAKlC,AAAU,MAAM,QAAiC;AAC/C,OAAK,MAAM,CAAC,MAAM,YAAY,QAAQ;AACpC,OAAI,KAAK,OAAO,MAAM,IAAI,MACxB,MAAK,OAAO,KAAK;IAAC,KAAK;IAAqB;IAAM;;AAKpD,QACG,SAAS,iBAAiB,SAAS,YACpC,KAAK,OAAO,MAAM,IAAI,UACtB;IACA,MAAM,OAAO,SAAS,gBAAgB,KAAK,OAAO,IAAI,KAAK;IAC3D,MAAM,6BAAY,IAAI,QAAO;IAC7B,MAAM,cAAc;AAClB,SAAI,SAAS,cACX,QAAO;cAEP,OAAO,YAAY,YACnB,WAAW,QACX,YAAY,QAEZ,QAAO;SAEP,QAAO;;AAIX,SAAK,OAAO,KAAK;KACf,KAAK;KACL;KACA;MAAE;MAAM;MAAM;MAAW;;;;;;CAMjC,AAAU,eACR,eACA;EACA,MAAM,UAAU,KAAK,uBAAuB;EAE5C,MAAM,eACJ,KAAK,gBAAgB,SAAS,KAAK,eAAe,UAAU;EAE9D,MAAM,mBAAmB,eAA2B;AAElD,QAAK,uBAAuB,KAAK,kBAAkB,cAC/C,gBACA,KAAK,mBACL;AAMJ,QAAK,mBAAmBK,gCAAkB,KAAK,kBAAkB,GAC9DC,6CACC,KAAK,OAAO,cAAc,iBAAiB;GAG/C,MAAM,kBAAkB,EAAE,GAAG,KAAK,WAAW;GAC7C,MAAM,cAAc0C,oCAClB,KAAK,4BACL;AAEF,QAAK,6BAA6B;AAIlC,GAAK,KAAK,8BAA8B;IACtC,QAAQ,EAAE,GAAG,KAAK;IAClB,iEAA2B;IAC3B,UAAU,EAAE,GAAG,KAAK;IACpB;;AAEF,QAAK,mBAAmB;IACtB,GAAG,KAAK;IACR,cAAc;KACZ,GAAG,KAAK,iBAAiB;KACzB,eAAe,KAAK,WAAW;;;;AAKrC,MAAI,CAAC,QACH,MAAK,qBAAqB;GACxB,GAAG;GACH,MAAM,KAAK;GACX,SAAS,KAAK,OAAO,eAAevC,gDAA8B;;AAKtE,OAAK,aAAawC,8BAChB,KAAK,YACL,eAAe,KAAK,WAAW,QAC/B,KAAK,MACL,UAAU,EAAE,IAAI,KAAK,WAAW,OAAO;AAIzC,MAAI,aAAc,iBAAgB,KAAK;AAEvC,MAAI,CAAC,QAEH,MAAK,QAAQ;;CAIjB,AAAU,sBAAsB;AAC9B,MAAI,KAAK,gBAAgB,KAAM;AAC/B,MAAI,KAAK,wBAAwB,WAAW,EAAG;EAG/C,MAAM,SAAS5C,gCAAkB,KAAK,kBAAkB;IACrDC,6CAA2B,KAAK,OAAO,cAAc,iBAAiB;IACtEQ,6CAA2B,KAAK,WAAW;;EAI9C,MAAMoC,SAAiD;AACvD,OAAK,MAAM,CAAC,KAAK,KAAK,UAAU,KAAK,yBAAyB;AAC5D,UAAO,SAAS;AAChB,UAAO,KAAK,KAAK,CAAC,KAAK;;AAIzB,OAAK,MAAM,CAAC,KAAK,OAAO,OAAO,QAAQ,QACrC,MAAK,qBAAqB,KACxB,KAAK,aAAa,UAAU,QAAQ,IAAI;;CAK9C,AAAU,aACR,OACA;AACA,OAAK,MAAM,CAAC,KAAK,GAAG,MAAM,KAAK,yBAAyB;AACtD,OAAI,MAAMnC,2BAAS,MAAMC,+BAAa,MAAMiB,yBAC1C;GAEF,MAAM,OAAO,OAAO,OAAO,OAAO,MAAM,MAAM,EAAE,OAAO;AACvD,OAAI,KACF,MAAK,OAAO,KAAK,CAAC,GAAG;;AAGzB,OAAK,MAAM,QAAQ,OAAO,OAAO,OAC/B,KAAI,KAAK,OAAO,SAAS,EACvB,MAAK,cAAc,KAAK,IAAI,KAAK,QAAQ"}
@@ -1,547 +0,0 @@
1
- import { EmptyInputError, GraphInterrupt, isGraphInterrupt } from "../errors.js";
2
- import { createCheckpoint, emptyChannels } from "../channels/base.js";
3
- import { CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, CONFIG_KEY_CHECKPOINT_ID, CONFIG_KEY_CHECKPOINT_MAP, CONFIG_KEY_CHECKPOINT_NS, CONFIG_KEY_READ, CONFIG_KEY_RESUME_MAP, CONFIG_KEY_RESUMING, CONFIG_KEY_SCRATCHPAD, CONFIG_KEY_STREAM, ERROR, INPUT, INTERRUPT, NULL_TASK_ID, PUSH, RESUME, START, TAG_HIDDEN, isCommand } from "../constants.js";
4
- import { gatherIterator, gatherIteratorSync, prefixGenerator } from "../utils.js";
5
- import { isXXH3 } from "../hash.js";
6
- import { mapCommand, mapInput, mapOutputUpdates, mapOutputValues, readChannels } from "./io.js";
7
- import { getNewChannelVersions, patchConfigurable } from "./utils/index.js";
8
- import { _applyWrites, _prepareNextTasks, _prepareSingleTask, increment, shouldInterrupt } from "./algo.js";
9
- import { mapDebugCheckpoint, mapDebugTaskResults, mapDebugTasks, printStepTasks } from "./debug.js";
10
- import { IterableReadableWritableStream } from "./stream.js";
11
- import { AsyncBatchedStore, BaseCache, WRITES_IDX_MAP, copyCheckpoint, emptyCheckpoint } from "@langchain/langgraph-checkpoint";
12
-
13
- //#region src/pregel/loop.ts
14
- const INPUT_DONE = Symbol.for("INPUT_DONE");
15
- const INPUT_RESUMING = Symbol.for("INPUT_RESUMING");
16
- const DEFAULT_LOOP_LIMIT = 25;
17
- function createDuplexStream(...streams) {
18
- return new IterableReadableWritableStream({
19
- passthroughFn: (value) => {
20
- for (const stream of streams) if (stream.modes.has(value[1])) stream.push(value);
21
- },
22
- modes: new Set(streams.flatMap((s) => Array.from(s.modes)))
23
- });
24
- }
25
- var AsyncBatchedCache = class extends BaseCache {
26
- cache;
27
- queue = Promise.resolve();
28
- constructor(cache) {
29
- super();
30
- this.cache = cache;
31
- }
32
- async get(keys) {
33
- return this.enqueueOperation("get", keys);
34
- }
35
- async set(pairs) {
36
- return this.enqueueOperation("set", pairs);
37
- }
38
- async clear(namespaces) {
39
- return this.enqueueOperation("clear", namespaces);
40
- }
41
- async stop() {
42
- await this.queue;
43
- }
44
- enqueueOperation(type, ...args) {
45
- const newPromise = this.queue.then(() => {
46
- return this.cache[type](...args);
47
- });
48
- this.queue = newPromise.then(() => void 0, () => void 0);
49
- return newPromise;
50
- }
51
- };
52
- var PregelLoop = class PregelLoop {
53
- input;
54
- output;
55
- config;
56
- checkpointer;
57
- checkpointerGetNextVersion;
58
- channels;
59
- checkpoint;
60
- checkpointIdSaved;
61
- checkpointConfig;
62
- checkpointMetadata;
63
- checkpointNamespace;
64
- checkpointPendingWrites = [];
65
- checkpointPreviousVersions;
66
- step;
67
- stop;
68
- durability;
69
- outputKeys;
70
- streamKeys;
71
- nodes;
72
- skipDoneTasks;
73
- prevCheckpointConfig;
74
- updatedChannels;
75
- status = "pending";
76
- tasks = {};
77
- stream;
78
- checkpointerPromises = [];
79
- isNested;
80
- _checkpointerChainedPromise = Promise.resolve();
81
- store;
82
- cache;
83
- manager;
84
- interruptAfter;
85
- interruptBefore;
86
- toInterrupt = [];
87
- debug = false;
88
- triggerToNodes;
89
- get isResuming() {
90
- let hasChannelVersions = false;
91
- if (START in this.checkpoint.channel_versions) hasChannelVersions = true;
92
- else for (const chan in this.checkpoint.channel_versions) if (Object.prototype.hasOwnProperty.call(this.checkpoint.channel_versions, chan)) {
93
- hasChannelVersions = true;
94
- break;
95
- }
96
- const configHasResumingFlag = this.config.configurable?.[CONFIG_KEY_RESUMING] !== void 0;
97
- const configIsResuming = configHasResumingFlag && this.config.configurable?.[CONFIG_KEY_RESUMING];
98
- const inputIsNullOrUndefined = this.input === null || this.input === void 0;
99
- const inputIsCommandResuming = isCommand(this.input) && this.input.resume != null;
100
- const inputIsResuming = this.input === INPUT_RESUMING;
101
- 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;
102
- return hasChannelVersions && (configIsResuming || inputIsNullOrUndefined || inputIsCommandResuming || inputIsResuming || runIdMatchesPrevious);
103
- }
104
- constructor(params) {
105
- this.input = params.input;
106
- this.checkpointer = params.checkpointer;
107
- if (this.checkpointer !== void 0) this.checkpointerGetNextVersion = this.checkpointer.getNextVersion.bind(this.checkpointer);
108
- else this.checkpointerGetNextVersion = increment;
109
- this.checkpoint = params.checkpoint;
110
- this.checkpointMetadata = params.checkpointMetadata;
111
- this.checkpointPreviousVersions = params.checkpointPreviousVersions;
112
- this.channels = params.channels;
113
- this.checkpointPendingWrites = params.checkpointPendingWrites;
114
- this.step = params.step;
115
- this.stop = params.stop;
116
- this.config = params.config;
117
- this.checkpointConfig = params.checkpointConfig;
118
- this.isNested = params.isNested;
119
- this.manager = params.manager;
120
- this.outputKeys = params.outputKeys;
121
- this.streamKeys = params.streamKeys;
122
- this.nodes = params.nodes;
123
- this.skipDoneTasks = params.skipDoneTasks;
124
- this.store = params.store;
125
- this.cache = params.cache ? new AsyncBatchedCache(params.cache) : void 0;
126
- this.stream = params.stream;
127
- this.checkpointNamespace = params.checkpointNamespace;
128
- this.prevCheckpointConfig = params.prevCheckpointConfig;
129
- this.interruptAfter = params.interruptAfter;
130
- this.interruptBefore = params.interruptBefore;
131
- this.durability = params.durability;
132
- this.debug = params.debug;
133
- this.triggerToNodes = params.triggerToNodes;
134
- }
135
- static async initialize(params) {
136
- let { config, stream } = params;
137
- if (stream !== void 0 && config.configurable?.[CONFIG_KEY_STREAM] !== void 0) stream = createDuplexStream(stream, config.configurable[CONFIG_KEY_STREAM]);
138
- const skipDoneTasks = config.configurable ? !("checkpoint_id" in config.configurable) : true;
139
- const scratchpad = config.configurable?.[CONFIG_KEY_SCRATCHPAD];
140
- if (config.configurable && scratchpad) {
141
- if (scratchpad.subgraphCounter > 0) config = patchConfigurable(config, { [CONFIG_KEY_CHECKPOINT_NS]: [config.configurable[CONFIG_KEY_CHECKPOINT_NS], scratchpad.subgraphCounter.toString()].join(CHECKPOINT_NAMESPACE_SEPARATOR) });
142
- scratchpad.subgraphCounter += 1;
143
- }
144
- const isNested = CONFIG_KEY_READ in (config.configurable ?? {});
145
- if (!isNested && config.configurable?.checkpoint_ns !== void 0 && config.configurable?.checkpoint_ns !== "") config = patchConfigurable(config, {
146
- checkpoint_ns: "",
147
- checkpoint_id: void 0
148
- });
149
- let checkpointConfig = config;
150
- if (config.configurable?.[CONFIG_KEY_CHECKPOINT_MAP] !== void 0 && config.configurable?.[CONFIG_KEY_CHECKPOINT_MAP]?.[config.configurable?.checkpoint_ns]) checkpointConfig = patchConfigurable(config, { checkpoint_id: config.configurable[CONFIG_KEY_CHECKPOINT_MAP][config.configurable?.checkpoint_ns] });
151
- const checkpointNamespace = config.configurable?.checkpoint_ns?.split(CHECKPOINT_NAMESPACE_SEPARATOR) ?? [];
152
- const saved = await params.checkpointer?.getTuple(checkpointConfig) ?? {
153
- config,
154
- checkpoint: emptyCheckpoint(),
155
- metadata: {
156
- source: "input",
157
- step: -2,
158
- parents: {}
159
- },
160
- pendingWrites: []
161
- };
162
- checkpointConfig = {
163
- ...config,
164
- ...saved.config,
165
- configurable: {
166
- checkpoint_ns: "",
167
- ...config.configurable,
168
- ...saved.config.configurable
169
- }
170
- };
171
- const prevCheckpointConfig = saved.parentConfig;
172
- const checkpoint = copyCheckpoint(saved.checkpoint);
173
- const checkpointMetadata = { ...saved.metadata };
174
- const checkpointPendingWrites = saved.pendingWrites ?? [];
175
- const channels = emptyChannels(params.channelSpecs, checkpoint);
176
- const step = (checkpointMetadata.step ?? 0) + 1;
177
- const stop = step + (config.recursionLimit ?? DEFAULT_LOOP_LIMIT) + 1;
178
- const checkpointPreviousVersions = { ...checkpoint.channel_versions };
179
- const store = params.store ? new AsyncBatchedStore(params.store) : void 0;
180
- if (store) await store.start();
181
- return new PregelLoop({
182
- input: params.input,
183
- config,
184
- checkpointer: params.checkpointer,
185
- checkpoint,
186
- checkpointMetadata,
187
- checkpointConfig,
188
- prevCheckpointConfig,
189
- checkpointNamespace,
190
- channels,
191
- isNested,
192
- manager: params.manager,
193
- skipDoneTasks,
194
- step,
195
- stop,
196
- checkpointPreviousVersions,
197
- checkpointPendingWrites,
198
- outputKeys: params.outputKeys ?? [],
199
- streamKeys: params.streamKeys ?? [],
200
- nodes: params.nodes,
201
- stream,
202
- store,
203
- cache: params.cache,
204
- interruptAfter: params.interruptAfter,
205
- interruptBefore: params.interruptBefore,
206
- durability: params.durability,
207
- debug: params.debug,
208
- triggerToNodes: params.triggerToNodes
209
- });
210
- }
211
- _checkpointerPutAfterPrevious(input) {
212
- this._checkpointerChainedPromise = this._checkpointerChainedPromise.then(() => {
213
- return this.checkpointer?.put(input.config, input.checkpoint, input.metadata, input.newVersions);
214
- });
215
- this.checkpointerPromises.push(this._checkpointerChainedPromise);
216
- }
217
- /**
218
- * Put writes for a task, to be read by the next tick.
219
- * @param taskId
220
- * @param writes
221
- */
222
- putWrites(taskId, writes) {
223
- let writesCopy = writes;
224
- if (writesCopy.length === 0) return;
225
- if (writesCopy.every(([key]) => key in WRITES_IDX_MAP)) writesCopy = Array.from(new Map(writesCopy.map((w) => [w[0], w])).values());
226
- this.checkpointPendingWrites = this.checkpointPendingWrites.filter((w) => w[0] !== taskId);
227
- for (const [c, v] of writesCopy) this.checkpointPendingWrites.push([
228
- taskId,
229
- c,
230
- v
231
- ]);
232
- const config = patchConfigurable(this.checkpointConfig, {
233
- [CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
234
- [CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id
235
- });
236
- if (this.durability !== "exit" && this.checkpointer != null) this.checkpointerPromises.push(this.checkpointer.putWrites(config, writesCopy, taskId));
237
- if (this.tasks) this._outputWrites(taskId, writesCopy);
238
- if (!writes.length || !this.cache || !this.tasks) return;
239
- const task = this.tasks[taskId];
240
- if (task == null || task.cache_key == null) return;
241
- if (writes[0][0] === ERROR || writes[0][0] === INTERRUPT) return;
242
- this.cache.set([{
243
- key: [task.cache_key.ns, task.cache_key.key],
244
- value: task.writes,
245
- ttl: task.cache_key.ttl
246
- }]);
247
- }
248
- _outputWrites(taskId, writes, cached = false) {
249
- const task = this.tasks[taskId];
250
- if (task !== void 0) {
251
- if (task.config !== void 0 && (task.config.tags ?? []).includes(TAG_HIDDEN)) return;
252
- if (writes.length > 0) {
253
- if (writes[0][0] === INTERRUPT) {
254
- if (task.path?.[0] === PUSH && task.path?.at(-1) === true) return;
255
- const interruptWrites = writes.filter((w) => w[0] === INTERRUPT).flatMap((w) => w[1]);
256
- this._emit([["updates", { [INTERRUPT]: interruptWrites }], ["values", { [INTERRUPT]: interruptWrites }]]);
257
- } else if (writes[0][0] !== ERROR) this._emit(gatherIteratorSync(prefixGenerator(mapOutputUpdates(this.outputKeys, [[task, writes]], cached), "updates")));
258
- }
259
- if (!cached) this._emit(gatherIteratorSync(prefixGenerator(mapDebugTaskResults([[task, writes]], this.streamKeys), "tasks")));
260
- }
261
- }
262
- async _matchCachedWrites() {
263
- if (!this.cache) return [];
264
- const matched = [];
265
- const serializeKey = ([ns, key]) => {
266
- return `ns:${ns.join(",")}|key:${key}`;
267
- };
268
- const keys = [];
269
- const keyMap = {};
270
- for (const task of Object.values(this.tasks)) if (task.cache_key != null && !task.writes.length) {
271
- keys.push([task.cache_key.ns, task.cache_key.key]);
272
- keyMap[serializeKey([task.cache_key.ns, task.cache_key.key])] = task;
273
- }
274
- if (keys.length === 0) return [];
275
- const cache = await this.cache.get(keys);
276
- for (const { key, value } of cache) {
277
- const task = keyMap[serializeKey(key)];
278
- if (task != null) {
279
- task.writes.push(...value);
280
- matched.push({
281
- task,
282
- result: value
283
- });
284
- }
285
- }
286
- return matched;
287
- }
288
- /**
289
- * Execute a single iteration of the Pregel loop.
290
- * Returns true if more iterations are needed.
291
- * @param params
292
- */
293
- async tick(params) {
294
- if (this.store && !this.store.isRunning) await this.store?.start();
295
- const { inputKeys = [] } = params;
296
- if (this.status !== "pending") throw new Error(`Cannot tick when status is no longer "pending". Current status: "${this.status}"`);
297
- if (![INPUT_DONE, INPUT_RESUMING].includes(this.input)) await this._first(inputKeys);
298
- else if (this.toInterrupt.length > 0) {
299
- this.status = "interrupt_before";
300
- throw new GraphInterrupt();
301
- } else if (Object.values(this.tasks).every((task) => task.writes.length > 0)) {
302
- const writes = Object.values(this.tasks).flatMap((t) => t.writes);
303
- this.updatedChannels = _applyWrites(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
304
- const valuesOutput = await gatherIterator(prefixGenerator(mapOutputValues(this.outputKeys, writes, this.channels), "values"));
305
- this._emit(valuesOutput);
306
- this.checkpointPendingWrites = [];
307
- await this._putCheckpoint({ source: "loop" });
308
- if (shouldInterrupt(this.checkpoint, this.interruptAfter, Object.values(this.tasks))) {
309
- this.status = "interrupt_after";
310
- throw new GraphInterrupt();
311
- }
312
- if (this.config.configurable?.[CONFIG_KEY_RESUMING] !== void 0) delete this.config.configurable?.[CONFIG_KEY_RESUMING];
313
- } else return false;
314
- if (this.step > this.stop) {
315
- this.status = "out_of_steps";
316
- return false;
317
- }
318
- const nextTasks = _prepareNextTasks(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, {
319
- step: this.step,
320
- checkpointer: this.checkpointer,
321
- isResuming: this.isResuming,
322
- manager: this.manager,
323
- store: this.store,
324
- stream: this.stream,
325
- triggerToNodes: this.triggerToNodes,
326
- updatedChannels: this.updatedChannels
327
- });
328
- this.tasks = nextTasks;
329
- if (this.checkpointer) this._emit(await gatherIterator(prefixGenerator(mapDebugCheckpoint(this.checkpointConfig, this.channels, this.streamKeys, this.checkpointMetadata, Object.values(this.tasks), this.checkpointPendingWrites, this.prevCheckpointConfig, this.outputKeys), "checkpoints")));
330
- if (Object.values(this.tasks).length === 0) {
331
- this.status = "done";
332
- return false;
333
- }
334
- if (this.skipDoneTasks && this.checkpointPendingWrites.length > 0) {
335
- for (const [tid, k, v] of this.checkpointPendingWrites) {
336
- if (k === ERROR || k === INTERRUPT || k === RESUME) continue;
337
- const task = Object.values(this.tasks).find((t) => t.id === tid);
338
- if (task) task.writes.push([k, v]);
339
- }
340
- for (const task of Object.values(this.tasks)) if (task.writes.length > 0) this._outputWrites(task.id, task.writes, true);
341
- }
342
- if (Object.values(this.tasks).every((task) => task.writes.length > 0)) return this.tick({ inputKeys });
343
- if (shouldInterrupt(this.checkpoint, this.interruptBefore, Object.values(this.tasks))) {
344
- this.status = "interrupt_before";
345
- throw new GraphInterrupt();
346
- }
347
- const debugOutput = await gatherIterator(prefixGenerator(mapDebugTasks(Object.values(this.tasks)), "tasks"));
348
- this._emit(debugOutput);
349
- return true;
350
- }
351
- async finishAndHandleError(error) {
352
- if (this.durability === "exit" && (!this.isNested || typeof error !== "undefined" || this.checkpointNamespace.every((part) => !part.includes(CHECKPOINT_NAMESPACE_END)))) {
353
- this._putCheckpoint(this.checkpointMetadata);
354
- this._flushPendingWrites();
355
- }
356
- const suppress = this._suppressInterrupt(error);
357
- if (suppress || error === void 0) this.output = readChannels(this.channels, this.outputKeys);
358
- if (suppress) {
359
- if (this.tasks !== void 0 && this.checkpointPendingWrites.length > 0 && Object.values(this.tasks).some((task) => task.writes.length > 0)) {
360
- this.updatedChannels = _applyWrites(this.checkpoint, this.channels, Object.values(this.tasks), this.checkpointerGetNextVersion, this.triggerToNodes);
361
- this._emit(gatherIteratorSync(prefixGenerator(mapOutputValues(this.outputKeys, Object.values(this.tasks).flatMap((t) => t.writes), this.channels), "values")));
362
- }
363
- if (isGraphInterrupt(error) && !error.interrupts.length) this._emit([["updates", { [INTERRUPT]: [] }], ["values", { [INTERRUPT]: [] }]]);
364
- }
365
- return suppress;
366
- }
367
- async acceptPush(task, writeIdx, call) {
368
- if (this.interruptAfter?.length > 0 && shouldInterrupt(this.checkpoint, this.interruptAfter, [task])) {
369
- this.toInterrupt.push(task);
370
- return;
371
- }
372
- const pushed = _prepareSingleTask([
373
- PUSH,
374
- task.path ?? [],
375
- writeIdx,
376
- task.id,
377
- call
378
- ], this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, task.config ?? {}, true, {
379
- step: this.step,
380
- checkpointer: this.checkpointer,
381
- manager: this.manager,
382
- store: this.store,
383
- stream: this.stream
384
- });
385
- if (!pushed) return;
386
- if (this.interruptBefore?.length > 0 && shouldInterrupt(this.checkpoint, this.interruptBefore, [pushed])) {
387
- this.toInterrupt.push(pushed);
388
- return;
389
- }
390
- this._emit(gatherIteratorSync(prefixGenerator(mapDebugTasks([pushed]), "tasks")));
391
- if (this.debug) printStepTasks(this.step, [pushed]);
392
- this.tasks[pushed.id] = pushed;
393
- if (this.skipDoneTasks) this._matchWrites({ [pushed.id]: pushed });
394
- const tasks = await this._matchCachedWrites();
395
- for (const { task: task$1 } of tasks) this._outputWrites(task$1.id, task$1.writes, true);
396
- return pushed;
397
- }
398
- _suppressInterrupt(e) {
399
- return isGraphInterrupt(e) && !this.isNested;
400
- }
401
- async _first(inputKeys) {
402
- const { configurable } = this.config;
403
- const scratchpad = configurable?.[CONFIG_KEY_SCRATCHPAD];
404
- if (scratchpad && scratchpad.nullResume !== void 0) this.putWrites(NULL_TASK_ID, [[RESUME, scratchpad.nullResume]]);
405
- if (isCommand(this.input)) {
406
- const hasResume = this.input.resume != null;
407
- if (this.input.resume != null && typeof this.input.resume === "object" && Object.keys(this.input.resume).every(isXXH3)) {
408
- this.config.configurable ??= {};
409
- this.config.configurable[CONFIG_KEY_RESUME_MAP] = this.input.resume;
410
- }
411
- if (hasResume && this.checkpointer == null) throw new Error("Cannot use Command(resume=...) without checkpointer");
412
- const writes = {};
413
- for (const [tid, key, value] of mapCommand(this.input, this.checkpointPendingWrites)) {
414
- writes[tid] ??= [];
415
- writes[tid].push([key, value]);
416
- }
417
- if (Object.keys(writes).length === 0) throw new EmptyInputError("Received empty Command input");
418
- for (const [tid, ws] of Object.entries(writes)) this.putWrites(tid, ws);
419
- }
420
- const nullWrites = (this.checkpointPendingWrites ?? []).filter((w) => w[0] === NULL_TASK_ID).map((w) => w.slice(1));
421
- if (nullWrites.length > 0) _applyWrites(this.checkpoint, this.channels, [{
422
- name: INPUT,
423
- writes: nullWrites,
424
- triggers: []
425
- }], this.checkpointerGetNextVersion, this.triggerToNodes);
426
- const isCommandUpdateOrGoto = isCommand(this.input) && nullWrites.length > 0;
427
- if (this.isResuming || isCommandUpdateOrGoto) {
428
- for (const channelName in this.channels) {
429
- if (!Object.prototype.hasOwnProperty.call(this.channels, channelName)) continue;
430
- if (this.checkpoint.channel_versions[channelName] !== void 0) {
431
- const version = this.checkpoint.channel_versions[channelName];
432
- this.checkpoint.versions_seen[INTERRUPT] = {
433
- ...this.checkpoint.versions_seen[INTERRUPT],
434
- [channelName]: version
435
- };
436
- }
437
- }
438
- const valuesOutput = await gatherIterator(prefixGenerator(mapOutputValues(this.outputKeys, true, this.channels), "values"));
439
- this._emit(valuesOutput);
440
- }
441
- if (this.isResuming) this.input = INPUT_RESUMING;
442
- else if (isCommandUpdateOrGoto) {
443
- await this._putCheckpoint({ source: "input" });
444
- this.input = INPUT_DONE;
445
- } else {
446
- const inputWrites = await gatherIterator(mapInput(inputKeys, this.input));
447
- if (inputWrites.length > 0) {
448
- const discardTasks = _prepareNextTasks(this.checkpoint, this.checkpointPendingWrites, this.nodes, this.channels, this.config, true, { step: this.step });
449
- this.updatedChannels = _applyWrites(this.checkpoint, this.channels, Object.values(discardTasks).concat([{
450
- name: INPUT,
451
- writes: inputWrites,
452
- triggers: []
453
- }]), this.checkpointerGetNextVersion, this.triggerToNodes);
454
- await this._putCheckpoint({ source: "input" });
455
- this.input = INPUT_DONE;
456
- } else if (!(CONFIG_KEY_RESUMING in (this.config.configurable ?? {}))) throw new EmptyInputError(`Received no input writes for ${JSON.stringify(inputKeys, null, 2)}`);
457
- else this.input = INPUT_DONE;
458
- }
459
- if (!this.isNested) this.config = patchConfigurable(this.config, { [CONFIG_KEY_RESUMING]: this.isResuming });
460
- }
461
- _emit(values) {
462
- for (const [mode, payload] of values) {
463
- if (this.stream.modes.has(mode)) this.stream.push([
464
- this.checkpointNamespace,
465
- mode,
466
- payload
467
- ]);
468
- if ((mode === "checkpoints" || mode === "tasks") && this.stream.modes.has("debug")) {
469
- const step = mode === "checkpoints" ? this.step - 1 : this.step;
470
- const timestamp = (/* @__PURE__ */ new Date()).toISOString();
471
- const type = (() => {
472
- if (mode === "checkpoints") return "checkpoint";
473
- else if (typeof payload === "object" && payload != null && "result" in payload) return "task_result";
474
- else return "task";
475
- })();
476
- this.stream.push([
477
- this.checkpointNamespace,
478
- "debug",
479
- {
480
- step,
481
- type,
482
- timestamp,
483
- payload
484
- }
485
- ]);
486
- }
487
- }
488
- }
489
- _putCheckpoint(inputMetadata) {
490
- const exiting = this.checkpointMetadata === inputMetadata;
491
- const doCheckpoint = this.checkpointer != null && (this.durability !== "exit" || exiting);
492
- const storeCheckpoint = (checkpoint) => {
493
- this.prevCheckpointConfig = this.checkpointConfig?.configurable?.checkpoint_id ? this.checkpointConfig : void 0;
494
- this.checkpointConfig = patchConfigurable(this.checkpointConfig, { [CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "" });
495
- const channelVersions = { ...this.checkpoint.channel_versions };
496
- const newVersions = getNewChannelVersions(this.checkpointPreviousVersions, channelVersions);
497
- this.checkpointPreviousVersions = channelVersions;
498
- this._checkpointerPutAfterPrevious({
499
- config: { ...this.checkpointConfig },
500
- checkpoint: copyCheckpoint(checkpoint),
501
- metadata: { ...this.checkpointMetadata },
502
- newVersions
503
- });
504
- this.checkpointConfig = {
505
- ...this.checkpointConfig,
506
- configurable: {
507
- ...this.checkpointConfig.configurable,
508
- checkpoint_id: this.checkpoint.id
509
- }
510
- };
511
- };
512
- if (!exiting) this.checkpointMetadata = {
513
- ...inputMetadata,
514
- step: this.step,
515
- parents: this.config.configurable?.[CONFIG_KEY_CHECKPOINT_MAP] ?? {}
516
- };
517
- this.checkpoint = createCheckpoint(this.checkpoint, doCheckpoint ? this.channels : void 0, this.step, exiting ? { id: this.checkpoint.id } : void 0);
518
- if (doCheckpoint) storeCheckpoint(this.checkpoint);
519
- if (!exiting) this.step += 1;
520
- }
521
- _flushPendingWrites() {
522
- if (this.checkpointer == null) return;
523
- if (this.checkpointPendingWrites.length === 0) return;
524
- const config = patchConfigurable(this.checkpointConfig, {
525
- [CONFIG_KEY_CHECKPOINT_NS]: this.config.configurable?.checkpoint_ns ?? "",
526
- [CONFIG_KEY_CHECKPOINT_ID]: this.checkpoint.id
527
- });
528
- const byTask = {};
529
- for (const [tid, key, value] of this.checkpointPendingWrites) {
530
- byTask[tid] ??= [];
531
- byTask[tid].push([key, value]);
532
- }
533
- for (const [tid, ws] of Object.entries(byTask)) this.checkpointerPromises.push(this.checkpointer.putWrites(config, ws, tid));
534
- }
535
- _matchWrites(tasks) {
536
- for (const [tid, k, v] of this.checkpointPendingWrites) {
537
- if (k === ERROR || k === INTERRUPT || k === RESUME) continue;
538
- const task = Object.values(tasks).find((t) => t.id === tid);
539
- if (task) task.writes.push([k, v]);
540
- }
541
- for (const task of Object.values(tasks)) if (task.writes.length > 0) this._outputWrites(task.id, task.writes, true);
542
- }
543
- };
544
-
545
- //#endregion
546
- export { PregelLoop };
547
- //# sourceMappingURL=loop.js.map