@librechat/agents 3.4.0 → 3.4.2

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 (204) hide show
  1. package/dist/cjs/agents/AgentContext.cjs.map +1 -1
  2. package/dist/cjs/graphs/Graph.cjs +165 -57
  3. package/dist/cjs/graphs/Graph.cjs.map +1 -1
  4. package/dist/cjs/graphs/MultiAgentGraph.cjs +81 -33
  5. package/dist/cjs/graphs/MultiAgentGraph.cjs.map +1 -1
  6. package/dist/cjs/graphs/applyGraphRuntimeConfig.cjs +14 -0
  7. package/dist/cjs/graphs/applyGraphRuntimeConfig.cjs.map +1 -0
  8. package/dist/cjs/graphs/createGraph.cjs +15 -0
  9. package/dist/cjs/graphs/createGraph.cjs.map +1 -0
  10. package/dist/cjs/graphs/index.cjs +1 -0
  11. package/dist/cjs/hooks/HookRegistry.cjs +83 -0
  12. package/dist/cjs/hooks/HookRegistry.cjs.map +1 -1
  13. package/dist/cjs/hooks/createWorkspacePolicyHook.cjs +1 -1
  14. package/dist/cjs/hooks/executeHooks.cjs +60 -14
  15. package/dist/cjs/hooks/executeHooks.cjs.map +1 -1
  16. package/dist/cjs/hooks/index.cjs.map +1 -1
  17. package/dist/cjs/hooks/types.cjs +2 -0
  18. package/dist/cjs/hooks/types.cjs.map +1 -1
  19. package/dist/cjs/llm/init.cjs +3 -3
  20. package/dist/cjs/llm/invoke.cjs +2 -2
  21. package/dist/cjs/main.cjs +34 -24
  22. package/dist/cjs/messages/format.cjs +40 -23
  23. package/dist/cjs/messages/format.cjs.map +1 -1
  24. package/dist/cjs/run.cjs +130 -53
  25. package/dist/cjs/run.cjs.map +1 -1
  26. package/dist/cjs/session/AgentSession.cjs +37 -2
  27. package/dist/cjs/session/AgentSession.cjs.map +1 -1
  28. package/dist/cjs/session/JsonlSessionStore.cjs +2 -2
  29. package/dist/cjs/stream.cjs +1 -1
  30. package/dist/cjs/summarization/node.cjs +6 -3
  31. package/dist/cjs/summarization/node.cjs.map +1 -1
  32. package/dist/cjs/tools/BashExecutor.cjs +1 -1
  33. package/dist/cjs/tools/CodeExecutor.cjs +1 -1
  34. package/dist/cjs/tools/ProgrammaticToolCalling.cjs +1 -1
  35. package/dist/cjs/tools/SubagentTool.cjs +4 -4
  36. package/dist/cjs/tools/SubagentTool.cjs.map +1 -1
  37. package/dist/cjs/tools/ToolNode.cjs +321 -95
  38. package/dist/cjs/tools/ToolNode.cjs.map +1 -1
  39. package/dist/cjs/tools/ToolSearch.cjs +1 -1
  40. package/dist/cjs/tools/eagerEventExecution.cjs +1 -0
  41. package/dist/cjs/tools/local/LocalExecutionEngine.cjs +3 -3
  42. package/dist/cjs/tools/local/LocalProgrammaticToolCalling.cjs +2 -2
  43. package/dist/cjs/tools/search/rerankers.cjs +154 -1
  44. package/dist/cjs/tools/search/rerankers.cjs.map +1 -1
  45. package/dist/cjs/tools/search/tool.cjs +4 -1
  46. package/dist/cjs/tools/search/tool.cjs.map +1 -1
  47. package/dist/cjs/tools/subagent/SubagentExecutionRegistry.cjs +601 -0
  48. package/dist/cjs/tools/subagent/SubagentExecutionRegistry.cjs.map +1 -0
  49. package/dist/cjs/tools/subagent/SubagentExecutor.cjs +1114 -233
  50. package/dist/cjs/tools/subagent/SubagentExecutor.cjs.map +1 -1
  51. package/dist/cjs/tools/subagent/SubagentReplay.cjs +177 -0
  52. package/dist/cjs/tools/subagent/SubagentReplay.cjs.map +1 -0
  53. package/dist/cjs/tools/subagent/childGraphConfig.cjs +278 -0
  54. package/dist/cjs/tools/subagent/childGraphConfig.cjs.map +1 -0
  55. package/dist/cjs/tools/subagent/index.cjs +1 -0
  56. package/dist/cjs/tools/subagent/runtimeLimits.cjs +5 -0
  57. package/dist/cjs/tools/subagent/runtimeLimits.cjs.map +1 -0
  58. package/dist/cjs/tools/toolOutputReferences.cjs +20 -0
  59. package/dist/cjs/tools/toolOutputReferences.cjs.map +1 -1
  60. package/dist/cjs/utils/index.cjs +2 -2
  61. package/dist/esm/agents/AgentContext.mjs.map +1 -1
  62. package/dist/esm/graphs/Graph.mjs +165 -57
  63. package/dist/esm/graphs/Graph.mjs.map +1 -1
  64. package/dist/esm/graphs/MultiAgentGraph.mjs +81 -33
  65. package/dist/esm/graphs/MultiAgentGraph.mjs.map +1 -1
  66. package/dist/esm/graphs/applyGraphRuntimeConfig.mjs +14 -0
  67. package/dist/esm/graphs/applyGraphRuntimeConfig.mjs.map +1 -0
  68. package/dist/esm/graphs/createGraph.mjs +15 -0
  69. package/dist/esm/graphs/createGraph.mjs.map +1 -0
  70. package/dist/esm/graphs/index.mjs +1 -0
  71. package/dist/esm/hooks/HookRegistry.mjs +83 -0
  72. package/dist/esm/hooks/HookRegistry.mjs.map +1 -1
  73. package/dist/esm/hooks/createWorkspacePolicyHook.mjs +1 -1
  74. package/dist/esm/hooks/executeHooks.mjs +60 -14
  75. package/dist/esm/hooks/executeHooks.mjs.map +1 -1
  76. package/dist/esm/hooks/index.mjs.map +1 -1
  77. package/dist/esm/hooks/types.mjs +2 -1
  78. package/dist/esm/hooks/types.mjs.map +1 -1
  79. package/dist/esm/llm/init.mjs +1 -1
  80. package/dist/esm/llm/invoke.mjs +2 -2
  81. package/dist/esm/main.mjs +21 -19
  82. package/dist/esm/messages/format.mjs +40 -23
  83. package/dist/esm/messages/format.mjs.map +1 -1
  84. package/dist/esm/run.mjs +131 -54
  85. package/dist/esm/run.mjs.map +1 -1
  86. package/dist/esm/session/AgentSession.mjs +37 -2
  87. package/dist/esm/session/AgentSession.mjs.map +1 -1
  88. package/dist/esm/session/JsonlSessionStore.mjs +2 -2
  89. package/dist/esm/stream.mjs +1 -1
  90. package/dist/esm/summarization/node.mjs +6 -3
  91. package/dist/esm/summarization/node.mjs.map +1 -1
  92. package/dist/esm/tools/BashExecutor.mjs +1 -1
  93. package/dist/esm/tools/CodeExecutor.mjs +1 -1
  94. package/dist/esm/tools/ProgrammaticToolCalling.mjs +1 -1
  95. package/dist/esm/tools/SubagentTool.mjs +4 -4
  96. package/dist/esm/tools/SubagentTool.mjs.map +1 -1
  97. package/dist/esm/tools/ToolNode.mjs +322 -96
  98. package/dist/esm/tools/ToolNode.mjs.map +1 -1
  99. package/dist/esm/tools/ToolSearch.mjs +1 -1
  100. package/dist/esm/tools/eagerEventExecution.mjs +1 -1
  101. package/dist/esm/tools/local/LocalExecutionEngine.mjs +3 -3
  102. package/dist/esm/tools/local/LocalProgrammaticToolCalling.mjs +2 -2
  103. package/dist/esm/tools/search/rerankers.mjs +154 -1
  104. package/dist/esm/tools/search/rerankers.mjs.map +1 -1
  105. package/dist/esm/tools/search/tool.mjs +4 -1
  106. package/dist/esm/tools/search/tool.mjs.map +1 -1
  107. package/dist/esm/tools/subagent/SubagentExecutionRegistry.mjs +597 -0
  108. package/dist/esm/tools/subagent/SubagentExecutionRegistry.mjs.map +1 -0
  109. package/dist/esm/tools/subagent/SubagentExecutor.mjs +1114 -233
  110. package/dist/esm/tools/subagent/SubagentExecutor.mjs.map +1 -1
  111. package/dist/esm/tools/subagent/SubagentReplay.mjs +170 -0
  112. package/dist/esm/tools/subagent/SubagentReplay.mjs.map +1 -0
  113. package/dist/esm/tools/subagent/childGraphConfig.mjs +272 -0
  114. package/dist/esm/tools/subagent/childGraphConfig.mjs.map +1 -0
  115. package/dist/esm/tools/subagent/index.mjs +1 -0
  116. package/dist/esm/tools/subagent/runtimeLimits.mjs +5 -0
  117. package/dist/esm/tools/subagent/runtimeLimits.mjs.map +1 -0
  118. package/dist/esm/tools/toolOutputReferences.mjs +20 -0
  119. package/dist/esm/tools/toolOutputReferences.mjs.map +1 -1
  120. package/dist/esm/utils/index.mjs +2 -2
  121. package/dist/types/agents/AgentContext.d.ts +1 -1
  122. package/dist/types/graphs/Graph.d.ts +22 -2
  123. package/dist/types/graphs/MultiAgentGraph.d.ts +7 -1
  124. package/dist/types/graphs/applyGraphRuntimeConfig.d.ts +3 -0
  125. package/dist/types/graphs/createGraph.d.ts +13 -0
  126. package/dist/types/graphs/graphFactory.d.ts +13 -0
  127. package/dist/types/graphs/index.d.ts +2 -0
  128. package/dist/types/hooks/HookRegistry.d.ts +10 -1
  129. package/dist/types/hooks/executeHooks.d.ts +5 -1
  130. package/dist/types/hooks/index.d.ts +2 -2
  131. package/dist/types/hooks/types.d.ts +12 -0
  132. package/dist/types/run.d.ts +6 -3
  133. package/dist/types/session/AgentSession.d.ts +1 -0
  134. package/dist/types/tools/SubagentTool.d.ts +1 -1
  135. package/dist/types/tools/ToolNode.d.ts +23 -15
  136. package/dist/types/tools/search/rerankers.d.ts +28 -0
  137. package/dist/types/tools/search/types.d.ts +38 -1
  138. package/dist/types/tools/subagent/SubagentExecutionRegistry.d.ts +171 -0
  139. package/dist/types/tools/subagent/SubagentExecutor.d.ts +70 -36
  140. package/dist/types/tools/subagent/SubagentReplay.d.ts +85 -0
  141. package/dist/types/tools/subagent/childGraphConfig.d.ts +37 -0
  142. package/dist/types/tools/subagent/index.d.ts +1 -1
  143. package/dist/types/tools/subagent/runtimeLimits.d.ts +4 -0
  144. package/dist/types/tools/toolOutputReferences.d.ts +12 -0
  145. package/dist/types/types/graph.d.ts +189 -24
  146. package/dist/types/types/hitl.d.ts +15 -0
  147. package/package.json +8 -2
  148. package/src/agents/AgentContext.ts +1 -1
  149. package/src/graphs/Graph.ts +349 -104
  150. package/src/graphs/MultiAgentGraph.ts +149 -47
  151. package/src/graphs/__tests__/Graph.breakerLifecycle.test.ts +48 -9
  152. package/src/graphs/__tests__/Graph.contextOverflow.test.ts +1 -1
  153. package/src/graphs/__tests__/Graph.subagentResumeState.test.ts +80 -0
  154. package/src/graphs/__tests__/MultiAgentGraph.test.ts +86 -0
  155. package/src/graphs/__tests__/composition.smoke.test.ts +152 -0
  156. package/src/graphs/__tests__/createGraph.test.ts +206 -0
  157. package/src/graphs/applyGraphRuntimeConfig.ts +25 -0
  158. package/src/graphs/createGraph.ts +32 -0
  159. package/src/graphs/graphFactory.ts +12 -0
  160. package/src/graphs/index.ts +2 -0
  161. package/src/hooks/HookRegistry.ts +151 -1
  162. package/src/hooks/__tests__/HookRegistry.test.ts +96 -0
  163. package/src/hooks/__tests__/executeHooks.test.ts +160 -0
  164. package/src/hooks/executeHooks.ts +133 -15
  165. package/src/hooks/index.ts +3 -1
  166. package/src/hooks/types.ts +16 -0
  167. package/src/messages/format.ts +54 -26
  168. package/src/messages/formatAgentMessages.reducer.test.ts +162 -0
  169. package/src/messages/formatAgentMessages.steer.test.ts +16 -11
  170. package/src/messages/formatAgentMessages.test.ts +7 -2
  171. package/src/run.ts +214 -56
  172. package/src/session/AgentSession.ts +52 -2
  173. package/src/session/__tests__/JsonlSessionStore.test.ts +53 -0
  174. package/src/specs/graph-subagent.live.test.ts +242 -0
  175. package/src/specs/graph-subagent.test.ts +322 -0
  176. package/src/specs/langfuse-routing.integration.test.ts +120 -1
  177. package/src/specs/subagent.test.ts +130 -4
  178. package/src/summarization/__tests__/node.test.ts +32 -14
  179. package/src/summarization/node.ts +15 -11
  180. package/src/tools/SubagentTool.ts +4 -4
  181. package/src/tools/ToolNode.ts +643 -157
  182. package/src/tools/__tests__/SubagentExecutor.test.ts +740 -15
  183. package/src/tools/__tests__/SubagentReplay.test.ts +329 -0
  184. package/src/tools/__tests__/ToolNode.breakerSignal.test.ts +208 -2
  185. package/src/tools/__tests__/directToolHITLResumeScope.test.ts +512 -2
  186. package/src/tools/__tests__/hitl.test.ts +234 -5
  187. package/src/tools/__tests__/subagentHooks.test.ts +1126 -61
  188. package/src/tools/__tests__/toolOutputReferences.test.ts +19 -2
  189. package/src/tools/search/rag-api-reranker.test.ts +891 -0
  190. package/src/tools/search/rerankers.ts +260 -0
  191. package/src/tools/search/tool.ts +6 -0
  192. package/src/tools/search/types.ts +45 -1
  193. package/src/tools/subagent/SubagentExecutionRegistry.ts +1159 -0
  194. package/src/tools/subagent/SubagentExecutor.ts +2232 -312
  195. package/src/tools/subagent/SubagentReplay.ts +584 -0
  196. package/src/tools/subagent/__tests__/SubagentExecutionRegistry.test.ts +637 -0
  197. package/src/tools/subagent/__tests__/SubagentExecutor.lazy.test.ts +3054 -0
  198. package/src/tools/subagent/__tests__/childGraphConfig.test.ts +521 -0
  199. package/src/tools/subagent/childGraphConfig.ts +694 -0
  200. package/src/tools/subagent/index.ts +6 -0
  201. package/src/tools/subagent/runtimeLimits.ts +7 -0
  202. package/src/tools/toolOutputReferences.ts +40 -1
  203. package/src/types/graph.ts +222 -24
  204. package/src/types/hitl.ts +16 -0
@@ -1,3 +1,4 @@
1
+ const require_SubagentReplay = require("./subagent/SubagentReplay.cjs");
1
2
  const require_misc = require("../utils/misc.cjs");
2
3
  const require_langfuseRuntimeScope = require("../langfuseRuntimeScope.cjs");
3
4
  const require_enum = require("../common/enum.cjs");
@@ -12,9 +13,10 @@ const require_intentArg = require("./intentArg.cjs");
12
13
  const require_toolOutputReferences = require("./toolOutputReferences.cjs");
13
14
  const require_run = require("../utils/run.cjs");
14
15
  require("../utils/index.cjs");
15
- const require_CodeSessionFileSummary = require("./CodeSessionFileSummary.cjs");
16
16
  const require_executeHooks = require("../hooks/executeHooks.cjs");
17
+ const require_types = require("../hooks/types.cjs");
17
18
  require("../hooks/index.cjs");
19
+ const require_CodeSessionFileSummary = require("./CodeSessionFileSummary.cjs");
18
20
  const require_resolveLocalExecutionTools = require("./local/resolveLocalExecutionTools.cjs");
19
21
  require("./local/index.cjs");
20
22
  let nanoid = require("nanoid");
@@ -29,6 +31,32 @@ function stripRunBreakerScope(configurable) {
29
31
  const { [require_streamLimits.RUN_BREAKER_SCOPE_CONFIG_KEY]: _scope, ...rest } = configurable;
30
32
  return rest;
31
33
  }
34
+ function createToolApprovalReplayKey(config, agentId, toolUseId) {
35
+ const configuredScope = config.configurable?.[require_types.TOOL_APPROVAL_EXECUTION_SCOPE_CONFIG_KEY];
36
+ const threadId = config.configurable?.thread_id;
37
+ let executionScope = "";
38
+ if (typeof configuredScope === "string") executionScope = configuredScope;
39
+ else if (typeof threadId === "string") executionScope = threadId;
40
+ return {
41
+ executionScope,
42
+ agentId,
43
+ toolUseId
44
+ };
45
+ }
46
+ function getToolApprovalReplaySessionId(config, hookSessionId) {
47
+ const configuredScope = config.configurable?.[require_types.TOOL_APPROVAL_EXECUTION_SCOPE_CONFIG_KEY];
48
+ return typeof configuredScope === "string" && configuredScope.length > 0 ? configuredScope : hookSessionId;
49
+ }
50
+ function withSubagentReplayBatch(config, replayBatchKey) {
51
+ if (replayBatchKey == null) return config;
52
+ return {
53
+ ...config,
54
+ configurable: {
55
+ ...config.configurable,
56
+ [require_SubagentReplay.SUBAGENT_PARENT_BATCH_CONFIG_KEY]: replayBatchKey
57
+ }
58
+ };
59
+ }
32
60
  function createToolErrorOwnership() {
33
61
  return {
34
62
  handlerOwned: /* @__PURE__ */ new Set(),
@@ -99,6 +127,25 @@ function describeOfferedShape(value) {
99
127
  if (Array.isArray(value)) return "array";
100
128
  return typeof value;
101
129
  }
130
+ function findAssistantBatch(messages) {
131
+ for (let i = messages.length - 1; i >= 0; i--) {
132
+ const message = messages[i];
133
+ if ((0, _langchain_core_messages.isAIMessage)(message)) return {
134
+ message,
135
+ index: i,
136
+ messageCount: messages.length
137
+ };
138
+ }
139
+ }
140
+ function getAssistantBatchReplayKey(batch, runId, threadId) {
141
+ return JSON.stringify([
142
+ runId ?? null,
143
+ threadId ?? null,
144
+ batch.message.id ?? null,
145
+ batch.index,
146
+ batch.messageCount
147
+ ]);
148
+ }
102
149
  /**
103
150
  * Build the `tool_approval` interrupt payload from the set of pending
104
151
  * `ask`-decision entries collected during PreToolUse hook handling.
@@ -106,9 +153,10 @@ function describeOfferedShape(value) {
106
153
  * scope. The interrupt itself is raised by the caller (which still
107
154
  * needs `interrupt()` plus the AsyncLocalStorage anchoring shim).
108
155
  */
109
- function buildToolApprovalInterruptPayload(askEntries) {
156
+ function buildToolApprovalInterruptPayload(askEntries, hookSessionId) {
110
157
  return {
111
158
  type: "tool_approval",
159
+ ...hookSessionId == null || hookSessionId === "" ? {} : { hook_session_id: hookSessionId },
112
160
  action_requests: askEntries.map(({ entry, reason }) => {
113
161
  const request = {
114
162
  tool_call_id: entry.call.id,
@@ -287,6 +335,10 @@ var ToolNode = class extends require_run.RunnableCallable {
287
335
  * the Run ends.
288
336
  */
289
337
  directPathTurns = /* @__PURE__ */ new Map();
338
+ /** Terminal results from interrupting siblings that must survive a
339
+ * LangGraph replay of the containing ToolNode. Includes the sidecar data
340
+ * the fresh batch needs for hook-context injection and completion events. */
341
+ settledInterruptingResults = /* @__PURE__ */ new Map();
290
342
  /** Tool registry for filtering (lazy computation of programmatic maps) */
291
343
  toolRegistry;
292
344
  /** Cached programmatic tools (computed once on first PTC call) */
@@ -520,16 +572,16 @@ var ToolNode = class extends require_run.RunnableCallable {
520
572
  return `Tool "${callName}" not found. Did you mean "${suggestion}"? Handoff tool names must match exactly.`;
521
573
  }
522
574
  /**
523
- * Flush the per-Run direct-path turn cache. Called by the Graph at
524
- * end-of-Run via `clearHeavyState`. The map intentionally survives
525
- * `run()` re-entry so an interrupt + resume reuses the original
526
- * slot (Codex P2 #30), but it would otherwise grow linearly with
527
- * tool calls and could collide across Runs if a provider reused
528
- * call IDs (Codex P2 #33). Hosts can also call this directly if
575
+ * Flush per-Run direct replay state. Called by the Graph at end-of-Run via
576
+ * `clearHeavyState`. The state intentionally survives `run()` re-entry so
577
+ * interrupt + resume keeps both original turn slots and terminal sibling
578
+ * outputs, but it would otherwise grow linearly and could collide across
579
+ * Runs if a provider reused call IDs. Hosts can also call this directly if
529
580
  * they reuse a ToolNode across batches outside of a Graph.
530
581
  */
531
582
  clearDirectPathTurns() {
532
583
  this.directPathTurns.clear();
584
+ this.settledInterruptingResults.clear();
533
585
  }
534
586
  /**
535
587
  * Returns cached programmatic tools, computing once on first access.
@@ -559,6 +611,29 @@ var ToolNode = class extends require_run.RunnableCallable {
559
611
  getToolUsageCounts() {
560
612
  return new Map(this.toolUsageCount);
561
613
  }
614
+ createSubagentResumeState() {
615
+ return {
616
+ stateKey: JSON.stringify([
617
+ this.executingAgentId ?? "",
618
+ this.agentId ?? "",
619
+ this.name
620
+ ]),
621
+ toolUsageCounts: [...this.toolUsageCount].map(([toolName, count]) => ({
622
+ toolName,
623
+ count
624
+ })),
625
+ directPathTurns: [...this.directPathTurns].map(([toolCallId, turn]) => ({
626
+ toolCallId,
627
+ turn
628
+ }))
629
+ };
630
+ }
631
+ restoreSubagentResumeState(state) {
632
+ this.toolUsageCount.clear();
633
+ for (const { toolName, count } of state.toolUsageCounts) this.toolUsageCount.set(toolName, count);
634
+ this.directPathTurns.clear();
635
+ for (const { toolCallId, turn } of state.directPathTurns) this.directPathTurns.set(toolCallId, turn);
636
+ }
562
637
  recordToolUsageTurn(toolName, turn, callId) {
563
638
  this.toolUsageCount.set(toolName, Math.max(this.toolUsageCount.get(toolName) ?? 0, turn + 1));
564
639
  if (callId != null && callId !== "") this.toolCallTurns.set(callId, turn);
@@ -860,9 +935,9 @@ var ToolNode = class extends require_run.RunnableCallable {
860
935
  * `blockDirectCall`, `respond` returns the host-supplied
861
936
  * `responseText` as a synthetic success ToolMessage,
862
937
  * `edit` re-runs with edited args. LangGraph re-enters
863
- * ToolNode.run from the start on resume; the hook fires
864
- * again and the resume value distinguishes "first ask" from
865
- * "second pass with decision".
938
+ * ToolNode.run from the start on resume. Reusable hooks fire
939
+ * again; a consumed one-shot hook replays its pending approval
940
+ * result. In both cases `interrupt()` consumes the resume value.
866
941
  * • When HITL is off: collapses to a fail-closed deny (matches
867
942
  * the rest of the SDK's HITL-disabled default). One-time
868
943
  * warning logged so hosts notice the gap.
@@ -882,13 +957,43 @@ var ToolNode = class extends require_run.RunnableCallable {
882
957
  * scopes and is left to a follow-up.
883
958
  */
884
959
  async runDirectToolWithLifecycleHooks(call, config, batchContext = {}) {
960
+ const replayController = this.toolMap.get(call.name)?.[require_SubagentReplay.SUBAGENT_REPLAY_CONTROLLER];
961
+ const replayConfig = replayController == null ? config : withSubagentReplayBatch(config, batchContext.replayBatchKey);
962
+ const settledOutput = await replayController?.getSettledOutput(call, replayConfig);
963
+ if (settledOutput != null) {
964
+ if (batchContext.additionalContextsSink != null && settledOutput.additionalContexts.length > 0) batchContext.additionalContextsSink.push(...settledOutput.additionalContexts);
965
+ if (batchContext.resolvedArgsByCallId != null && call.id != null && settledOutput.resolvedArgs != null) batchContext.resolvedArgsByCallId.set(call.id, settledOutput.resolvedArgs);
966
+ const refMeta = settledOutput.output.additional_kwargs;
967
+ if (this.toolOutputRegistry != null && refMeta?._refKey != null && settledOutput.referenceContent != null) this.toolOutputRegistry.set(refMeta._refScope, refMeta._refKey, settledOutput.referenceContent);
968
+ return settledOutput.output;
969
+ }
970
+ const replayAdditionalContexts = [];
971
+ const persistOutput = async (output, terminalArgs) => {
972
+ const refMeta = output.additional_kwargs;
973
+ if (terminalArgs != null && call.id != null && batchContext.resolvedArgsByCallId != null) batchContext.resolvedArgsByCallId.set(call.id, terminalArgs);
974
+ const resolvedArgs = terminalArgs ?? (call.id == null ? void 0 : batchContext.resolvedArgsByCallId?.get(call.id));
975
+ const referenceContent = this.toolOutputRegistry == null || refMeta?._refKey == null ? void 0 : this.toolOutputRegistry.get(refMeta._refScope, refMeta._refKey);
976
+ await replayController?.persistSettledOutput(call, replayConfig, {
977
+ output,
978
+ additionalContexts: replayAdditionalContexts,
979
+ ...resolvedArgs == null ? {} : { resolvedArgs },
980
+ ...referenceContent == null ? {} : { referenceContent }
981
+ });
982
+ return output;
983
+ };
885
984
  const runId = config.configurable?.run_id ?? "";
985
+ const threadId = config.configurable?.thread_id;
886
986
  const hookRegistry = this.hookRegistry;
887
987
  const hasPreHook = hookRegistry?.hasHookFor("PreToolUse", runId) === true;
988
+ const approvalReplayKey = call.id == null || call.id === "" ? void 0 : createToolApprovalReplayKey(config, this.executingAgentId ?? this.agentId ?? "", call.id);
989
+ const approvalReplaySessionId = getToolApprovalReplaySessionId(config, runId);
990
+ const pendingApproval = approvalReplayKey == null ? void 0 : hookRegistry?.getPendingToolApproval(approvalReplaySessionId, approvalReplayKey);
888
991
  const hasPostHook = hookRegistry?.hasHookFor("PostToolUse", runId) === true;
889
992
  const hasFailureHook = hookRegistry?.hasHookFor("PostToolUseFailure", runId) === true;
890
- if (hookRegistry == null || !hasPreHook && !hasPostHook && !hasFailureHook) return this.runTool(call, config, batchContext);
891
- const threadId = config.configurable?.thread_id;
993
+ if (hookRegistry == null || !hasPreHook && pendingApproval == null && !hasPostHook && !hasFailureHook) {
994
+ const output = await this.runTool(call, replayConfig, batchContext);
995
+ return output instanceof _langchain_core_messages.ToolMessage ? persistOutput(output) : output;
996
+ }
892
997
  const registryRunId = batchContext.batchScopeId ?? config.configurable?.run_id;
893
998
  let usageCount;
894
999
  const cachedTurn = call.id != null && call.id !== "" ? this.directPathTurns.get(call.id) ?? this.toolCallTurns.get(call.id) : void 0;
@@ -913,7 +1018,7 @@ var ToolNode = class extends require_run.RunnableCallable {
913
1018
  resolvedArgs = resolved;
914
1019
  }
915
1020
  let effectiveCall = call;
916
- if (hasPreHook) {
1021
+ if (hasPreHook || pendingApproval != null) {
917
1022
  const preResult = await require_executeHooks.executeHooks({
918
1023
  registry: hookRegistry,
919
1024
  input: {
@@ -929,34 +1034,48 @@ var ToolNode = class extends require_run.RunnableCallable {
929
1034
  turn
930
1035
  },
931
1036
  sessionId: runId,
932
- matchQuery: call.name
1037
+ matchQuery: call.name,
1038
+ onceReplayKey: approvalReplayKey,
1039
+ onceReplaySessionId: approvalReplaySessionId
933
1040
  }).catch(() => void 0);
934
1041
  if (preResult != null) {
935
- if (batchContext.additionalContextsSink != null && preResult.additionalContexts.length > 0) batchContext.additionalContextsSink.push(...preResult.additionalContexts);
1042
+ if (preResult.additionalContexts.length > 0) {
1043
+ replayAdditionalContexts.push(...preResult.additionalContexts);
1044
+ batchContext.additionalContextsSink?.push(...preResult.additionalContexts);
1045
+ }
936
1046
  if (preResult.updatedInput != null) effectiveCall = {
937
1047
  ...call,
938
1048
  args: preResult.updatedInput
939
1049
  };
940
- if (preResult.decision === "deny") return this.blockDirectCall({
1050
+ if (preResult.decision === "deny") return persistOutput(this.blockDirectCall({
941
1051
  call,
942
1052
  resolvedArgs,
943
1053
  reason: preResult.reason ?? "Blocked by hook",
944
1054
  hookRegistry,
945
1055
  runId,
946
1056
  threadId
947
- });
1057
+ }), effectiveCall.args);
948
1058
  if (preResult.decision === "ask") {
949
1059
  if (this.humanInTheLoop?.enabled !== true) {
950
1060
  const reason = this.resolveAskDecisionForDirectTool(preResult.reason, call.name);
951
- return this.blockDirectCall({
1061
+ return persistOutput(this.blockDirectCall({
952
1062
  call,
953
1063
  resolvedArgs,
954
1064
  reason,
955
1065
  hookRegistry,
956
1066
  runId,
957
1067
  threadId
958
- });
1068
+ }), effectiveCall.args);
959
1069
  }
1070
+ const toolCallId = call.id;
1071
+ if (toolCallId == null || toolCallId === "" || approvalReplayKey == null) return persistOutput(this.blockDirectCall({
1072
+ call,
1073
+ resolvedArgs,
1074
+ reason: "Tool approval requires a non-empty tool call ID — failing closed",
1075
+ hookRegistry,
1076
+ runId,
1077
+ threadId
1078
+ }), effectiveCall.args);
960
1079
  const payload = buildToolApprovalInterruptPayload([{
961
1080
  entry: {
962
1081
  call: effectiveCall,
@@ -965,59 +1084,71 @@ var ToolNode = class extends require_run.RunnableCallable {
965
1084
  },
966
1085
  reason: preResult.reason,
967
1086
  allowedDecisions: preResult.allowedDecisions
968
- }]);
1087
+ }], runId);
969
1088
  const resumeValue = _langchain_core_singletons.AsyncLocalStorageProviderSingleton.runWithConfig(config, () => (0, _langchain_langgraph.interrupt)(payload));
970
- const decision = normalizeApprovalDecisions([call.id], resumeValue).get(call.id) ?? {
1089
+ hookRegistry.clearPendingToolApproval(approvalReplaySessionId, approvalReplayKey);
1090
+ const decision = normalizeApprovalDecisions([toolCallId], resumeValue).get(toolCallId) ?? {
971
1091
  type: "reject",
972
1092
  reason: "No decision provided for tool approval"
973
1093
  };
974
1094
  const declaredType = decision.type;
975
- if (preResult.allowedDecisions != null && (typeof declaredType !== "string" || !preResult.allowedDecisions.includes(declaredType))) return this.blockDirectCall({
1095
+ if (preResult.allowedDecisions != null && (typeof declaredType !== "string" || !preResult.allowedDecisions.includes(declaredType))) return persistOutput(this.blockDirectCall({
976
1096
  call,
977
1097
  resolvedArgs,
978
1098
  reason: `Decision "${typeof declaredType === "string" ? declaredType : "<missing>"}" not in allowedDecisions [${preResult.allowedDecisions.join(", ")}] — failing closed`,
979
1099
  hookRegistry,
980
1100
  runId,
981
1101
  threadId
982
- });
983
- if (decision.type === "reject") return this.blockDirectCall({
1102
+ }), effectiveCall.args);
1103
+ if (decision.type === "reject") return persistOutput(this.blockDirectCall({
984
1104
  call,
985
1105
  resolvedArgs,
986
1106
  reason: decision.reason ?? preResult.reason ?? "Rejected by user",
987
1107
  hookRegistry,
988
1108
  runId,
989
1109
  threadId
990
- });
1110
+ }), effectiveCall.args);
991
1111
  if (decision.type === "respond") {
992
1112
  const responseText = decision.responseText;
993
- if (typeof responseText !== "string") return this.blockDirectCall({
1113
+ if (typeof responseText !== "string") return persistOutput(this.blockDirectCall({
994
1114
  call,
995
1115
  resolvedArgs,
996
1116
  reason: "Approval payload `respond` was missing a string `responseText`",
997
1117
  hookRegistry,
998
1118
  runId,
999
1119
  threadId
1000
- });
1001
- return new _langchain_core_messages.ToolMessage({
1120
+ }), effectiveCall.args);
1121
+ return persistOutput(new _langchain_core_messages.ToolMessage({
1002
1122
  status: "success",
1003
- content: responseText,
1123
+ content: require_truncation.truncateToolResultContent(responseText, this.maxToolResultChars),
1004
1124
  name: call.name,
1005
1125
  tool_call_id: call.id ?? ""
1006
- });
1126
+ }), effectiveCall.args);
1007
1127
  }
1008
1128
  if (decision.type === "edit") {
1009
1129
  const updatedInput = decision.updatedInput;
1010
- if (updatedInput === null || typeof updatedInput !== "object" || Array.isArray(updatedInput)) return new _langchain_core_messages.ToolMessage({
1130
+ if (updatedInput === null || typeof updatedInput !== "object" || Array.isArray(updatedInput)) return persistOutput(new _langchain_core_messages.ToolMessage({
1011
1131
  status: "error",
1012
1132
  content: "Decision \"edit\" missing object updatedInput — failing closed.",
1013
1133
  name: call.name,
1014
1134
  tool_call_id: call.id ?? ""
1015
- });
1135
+ }), effectiveCall.args);
1016
1136
  effectiveCall = {
1017
1137
  ...call,
1018
1138
  args: updatedInput
1019
1139
  };
1020
1140
  }
1141
+ if (declaredType !== "approve" && declaredType !== "edit") {
1142
+ const unknownType = typeof declaredType === "string" ? declaredType : "<missing>";
1143
+ return persistOutput(this.blockDirectCall({
1144
+ call,
1145
+ resolvedArgs,
1146
+ reason: `Unknown approval decision type "${unknownType}" — failing closed`,
1147
+ hookRegistry,
1148
+ runId,
1149
+ threadId
1150
+ }), effectiveCall.args);
1151
+ }
1021
1152
  }
1022
1153
  }
1023
1154
  }
@@ -1031,7 +1162,7 @@ var ToolNode = class extends require_run.RunnableCallable {
1031
1162
  * which is strictly more accurate.
1032
1163
  */
1033
1164
  if (effectiveCall !== call && batchContext.resolvedArgsByCallId != null && call.id != null && call.id !== "") batchContext.resolvedArgsByCallId.set(call.id, effectiveCall.args);
1034
- const output = await this.runTool(effectiveCall, config, {
1165
+ const output = await this.runTool(effectiveCall, replayConfig, {
1035
1166
  ...batchContext,
1036
1167
  usageCount
1037
1168
  });
@@ -1055,8 +1186,11 @@ var ToolNode = class extends require_run.RunnableCallable {
1055
1186
  sessionId: runId,
1056
1187
  matchQuery: call.name
1057
1188
  }).catch(() => void 0);
1058
- if (failureResult != null && batchContext.additionalContextsSink != null && failureResult.additionalContexts.length > 0) batchContext.additionalContextsSink.push(...failureResult.additionalContexts);
1059
- return output;
1189
+ if (failureResult != null && failureResult.additionalContexts.length > 0) {
1190
+ replayAdditionalContexts.push(...failureResult.additionalContexts);
1191
+ batchContext.additionalContextsSink?.push(...failureResult.additionalContexts);
1192
+ }
1193
+ return persistOutput(output);
1060
1194
  }
1061
1195
  if (output.status !== "error" && hasPostHook) {
1062
1196
  const postResult = await require_executeHooks.executeHooks({
@@ -1077,21 +1211,24 @@ var ToolNode = class extends require_run.RunnableCallable {
1077
1211
  sessionId: runId,
1078
1212
  matchQuery: call.name
1079
1213
  }).catch(() => void 0);
1080
- if (postResult != null && batchContext.additionalContextsSink != null && postResult.additionalContexts.length > 0) batchContext.additionalContextsSink.push(...postResult.additionalContexts);
1214
+ if (postResult != null && postResult.additionalContexts.length > 0) {
1215
+ replayAdditionalContexts.push(...postResult.additionalContexts);
1216
+ batchContext.additionalContextsSink?.push(...postResult.additionalContexts);
1217
+ }
1081
1218
  if (postResult?.updatedOutput != null) {
1082
1219
  if (require_toolContent.hasComputerCallOutputMarker(output)) {
1083
1220
  if (!require_toolContent.isComputerCallOutputContent(postResult.updatedOutput)) throw new Error("PostToolUse updatedOutput for a computer call must be a valid screenshot URL or screenshot content block.");
1084
- return require_toolContent.cloneToolMessageWithContent(output, postResult.updatedOutput);
1221
+ return persistOutput(require_toolContent.cloneToolMessageWithContent(output, postResult.updatedOutput));
1085
1222
  }
1086
1223
  const refMeta = output.additional_kwargs;
1087
1224
  const refKey = refMeta?._refKey;
1088
1225
  const refScope = refMeta?._refScope;
1089
1226
  const replaced = serializeToolOutputWithinLimits(postResult.updatedOutput, this.maxToolResultChars, this.toolOutputRegistry != null && refKey != null ? this.toolOutputRegistry.perOutputLimit : 0);
1090
1227
  if (this.toolOutputRegistry != null && refKey != null) this.toolOutputRegistry.set(refScope, refKey, replaced.registryContent);
1091
- return require_toolContent.cloneToolMessageWithContent(output, replaced.content);
1228
+ return persistOutput(require_toolContent.cloneToolMessageWithContent(output, replaced.content));
1092
1229
  }
1093
1230
  }
1094
- return output;
1231
+ return persistOutput(output);
1095
1232
  }
1096
1233
  /**
1097
1234
  * `ask` decisions on direct-path tools collapse to fail-closed deny
@@ -1383,13 +1520,10 @@ var ToolNode = class extends require_run.RunnableCallable {
1383
1520
  injectedMessages: [],
1384
1521
  errors: []
1385
1522
  });
1386
- if (this.hookRegistry?.hasHookFor("PreToolUse", runId) === true) {
1387
- /**
1388
- * Capture as a non-null local so the inner `blockEntry` closure
1389
- * doesn't lose narrowing on `this.hookRegistry` and we don't have
1390
- * to defensively `?.` it across every reference inside.
1391
- */
1392
- const hookRegistry = this.hookRegistry;
1523
+ const hookRegistry = this.hookRegistry;
1524
+ const approvalReplaySessionId = getToolApprovalReplaySessionId(config, runId);
1525
+ const hasPendingApproval = preToolCalls.some((entry) => entry.call.id != null && hookRegistry?.getPendingToolApproval(approvalReplaySessionId, createToolApprovalReplayKey(config, this.executingAgentId ?? this.agentId ?? "", entry.call.id)) != null);
1526
+ if (hookRegistry != null && (hookRegistry.hasHookFor("PreToolUse", runId) || hasPendingApproval)) {
1393
1527
  /**
1394
1528
  * Pull each call's prestarted eager record BEFORE awaiting the hooks:
1395
1529
  * an async deny must never race the eager host promise into emitting a
@@ -1409,23 +1543,29 @@ var ToolNode = class extends require_run.RunnableCallable {
1409
1543
  this.eagerEventToolExecutions.delete(callId);
1410
1544
  }
1411
1545
  }
1412
- const preResults = await Promise.all(preToolCalls.map((entry) => require_executeHooks.executeHooks({
1413
- registry: hookRegistry,
1414
- input: {
1415
- hook_event_name: "PreToolUse",
1416
- runId,
1417
- threadId,
1418
- agentId: this.agentId,
1419
- executingAgentId: this.executingAgentId,
1420
- toolName: entry.call.name,
1421
- toolInput: entry.args,
1422
- toolUseId: entry.call.id,
1423
- stepId: entry.stepId,
1424
- turn: this.toolUsageCount.get(entry.call.name) ?? 0
1425
- },
1426
- sessionId: runId,
1427
- matchQuery: entry.call.name
1428
- }).catch(() => HOOK_FALLBACK)));
1546
+ const preResults = await Promise.all(preToolCalls.map((entry) => {
1547
+ const toolUseId = entry.call.id;
1548
+ const approvalReplayKey = toolUseId == null ? void 0 : createToolApprovalReplayKey(config, this.executingAgentId ?? this.agentId ?? "", toolUseId);
1549
+ return require_executeHooks.executeHooks({
1550
+ registry: hookRegistry,
1551
+ input: {
1552
+ hook_event_name: "PreToolUse",
1553
+ runId,
1554
+ threadId,
1555
+ agentId: this.agentId,
1556
+ executingAgentId: this.executingAgentId,
1557
+ toolName: entry.call.name,
1558
+ toolInput: entry.args,
1559
+ toolUseId: entry.call.id,
1560
+ stepId: entry.stepId,
1561
+ turn: this.toolUsageCount.get(entry.call.name) ?? 0
1562
+ },
1563
+ sessionId: runId,
1564
+ matchQuery: entry.call.name,
1565
+ onceReplayKey: approvalReplayKey,
1566
+ onceReplaySessionId: approvalReplaySessionId
1567
+ }).catch(() => HOOK_FALLBACK);
1568
+ }));
1429
1569
  /**
1430
1570
  * Side effects deferred from `blockEntry` until after any pending
1431
1571
  * `interrupt()` resolves. Without deferral, a batch that mixes a
@@ -1582,12 +1722,11 @@ var ToolNode = class extends require_run.RunnableCallable {
1582
1722
  * pauses, gathers human input, and resumes the run with one
1583
1723
  * decision per request. On resume LangGraph re-executes this node
1584
1724
  * from the start; `interrupt()` then returns the resume value
1585
- * instead of throwing, so the loop above re-runs and the same
1586
- * `askEntries` list is rebuilt deterministically (assuming hooks
1587
- * are pure — see `humanInTheLoop` docs).
1725
+ * instead of throwing. Reusable hooks rebuild their entries, while
1726
+ * consumed one-shot hooks replay their pending approval aggregate.
1588
1727
  */
1589
1728
  if (askEntries.length > 0) {
1590
- const payload = buildToolApprovalInterruptPayload(askEntries);
1729
+ const payload = buildToolApprovalInterruptPayload(askEntries, runId);
1591
1730
  /**
1592
1731
  * `interrupt()` reads the current `RunnableConfig` from
1593
1732
  * AsyncLocalStorage. ToolNode usually runs with tracing disabled
@@ -1598,6 +1737,7 @@ var ToolNode = class extends require_run.RunnableCallable {
1598
1737
  * suspend and resume.
1599
1738
  */
1600
1739
  const resumeValue = _langchain_core_singletons.AsyncLocalStorageProviderSingleton.runWithConfig(config, () => (0, _langchain_langgraph.interrupt)(payload));
1740
+ for (const { entry } of askEntries) hookRegistry.clearPendingToolApproval(approvalReplaySessionId, createToolApprovalReplayKey(config, this.executingAgentId ?? this.agentId ?? "", entry.call.id));
1601
1741
  const decisionByCallId = normalizeApprovalDecisions(askEntries.map(({ entry }) => entry.call.id), resumeValue);
1602
1742
  for (const { entry, reason: askReason, allowedDecisions } of askEntries) {
1603
1743
  const decision = decisionByCallId.get(entry.call.id) ?? {
@@ -2189,16 +2329,16 @@ var ToolNode = class extends require_run.RunnableCallable {
2189
2329
  * call throws a `GraphInterrupt`, the `await` below rejects and unwinds the
2190
2330
  * whole ToolNode *before* any non-interrupting sibling has started — so a
2191
2331
  * sibling with real side effects (send_email, billing) never executes on
2192
- * the first pass. On the resume pass LangGraph re-runs the batch from the
2193
- * top; the interrupting tool resolves with the host's answer instead of
2194
- * throwing, and the siblings execute for the FIRST time, exactly once.
2332
+ * the first pass. Terminal interrupting siblings are cached by call id, so
2333
+ * LangGraph replay reuses their complete lifecycle output instead of
2334
+ * repeating model calls, hooks, or side effects.
2195
2335
  *
2196
2336
  * Without this ordering, a flat `Promise.all` starts every sibling
2197
2337
  * concurrently, so a non-idempotent sibling can complete its side effect
2198
2338
  * before the interrupt unwinds and then run a SECOND time on resume — the
2199
- * duplicate side effect this method exists to prevent. Interrupting tools
2200
- * are expected to be side-effect-free (they only suspend), so running them
2201
- * as a group and re-running them on resume is harmless.
2339
+ * duplicate side effect this method exists to prevent. A tool that actually
2340
+ * suspends re-enters until it reaches a terminal result; siblings that
2341
+ * already settled do not re-enter.
2202
2342
  *
2203
2343
  * `batchIndices[i]` is `directCalls[i]`'s position within the parent
2204
2344
  * ToolNode batch (used for `{{tool<i>turn<n>}}` registration); it is
@@ -2207,28 +2347,95 @@ var ToolNode = class extends require_run.RunnableCallable {
2207
2347
  * per-call here.
2208
2348
  */
2209
2349
  async runDirectBatchInterruptSafe(directCalls, batchIndices, config, baseContext) {
2210
- const runOne = (call, position) => this.runDirectToolWithLifecycleHooks(call, config, {
2211
- ...baseContext,
2212
- batchIndex: batchIndices[position]
2213
- });
2350
+ const settledBatchResults = baseContext.replayBatchKey == null ? void 0 : this.settledInterruptingResults.get(baseContext.replayBatchKey);
2351
+ const restoreResult = (call, result) => {
2352
+ baseContext.additionalContextsSink?.push(...result.additionalContexts);
2353
+ if (call.id != null && result.resolvedArgs != null && baseContext.resolvedArgsByCallId != null) baseContext.resolvedArgsByCallId.set(call.id, result.resolvedArgs);
2354
+ return result;
2355
+ };
2356
+ const runOne = async (call, position) => {
2357
+ const cachedResult = typeof call.id === "string" ? settledBatchResults?.get(call.id) : void 0;
2358
+ if (cachedResult != null) return restoreResult(call, cachedResult);
2359
+ const additionalContexts = [];
2360
+ const output = await this.runDirectToolWithLifecycleHooks(call, config, {
2361
+ ...baseContext,
2362
+ batchIndex: batchIndices[position],
2363
+ additionalContextsSink: additionalContexts
2364
+ });
2365
+ const resolvedArgs = call.id == null ? void 0 : baseContext.resolvedArgsByCallId?.get(call.id);
2366
+ return restoreResult(call, {
2367
+ output,
2368
+ additionalContexts,
2369
+ ...resolvedArgs == null ? {} : { resolvedArgs }
2370
+ });
2371
+ };
2214
2372
  const interrupting = this.interruptingToolNames;
2215
- if (!(interrupting != null && directCalls.some((call) => interrupting.has(call.name)))) return Promise.all(directCalls.map((call, i) => runOne(call, i)));
2373
+ if (!(interrupting != null && directCalls.some((call) => interrupting.has(call.name)))) return (await Promise.all(directCalls.map((call, i) => runOne(call, i)))).map((result) => result.output);
2216
2374
  const outputs = new Array(directCalls.length);
2217
2375
  const interruptingPositions = [];
2218
2376
  const regularPositions = [];
2219
2377
  for (let i = 0; i < directCalls.length; i++) if (interrupting.has(directCalls[i].name)) interruptingPositions.push(i);
2220
2378
  else regularPositions.push(i);
2221
- const interruptingOutputs = await Promise.all(interruptingPositions.map((i) => runOne(directCalls[i], i)));
2222
- interruptingPositions.forEach((i, k) => {
2223
- outputs[i] = interruptingOutputs[k];
2224
- });
2379
+ const interruptingResults = await Promise.allSettled(interruptingPositions.map((i) => runOne(directCalls[i], i)));
2380
+ /** A sibling may have tripped the run-wide breaker while another raised
2381
+ * a GraphInterrupt. Safety failures take precedence over approval pauses
2382
+ * regardless of input order. */
2383
+ this.throwIfBreakerTripped(config);
2384
+ let hasInterruptingError = false;
2385
+ let interruptingError;
2386
+ for (let i = 0; i < interruptingResults.length; i++) {
2387
+ const result = interruptingResults[i];
2388
+ if (result.status === "rejected") {
2389
+ if (!hasInterruptingError) {
2390
+ hasInterruptingError = true;
2391
+ interruptingError = result.reason;
2392
+ }
2393
+ continue;
2394
+ }
2395
+ const position = interruptingPositions[i];
2396
+ outputs[position] = result.value.output;
2397
+ const callId = directCalls[position].id;
2398
+ if (baseContext.replayBatchKey != null && typeof callId === "string" && callId !== "") {
2399
+ const batchResults = this.settledInterruptingResults.get(baseContext.replayBatchKey) ?? /* @__PURE__ */ new Map();
2400
+ batchResults.set(callId, result.value);
2401
+ this.settledInterruptingResults.set(baseContext.replayBatchKey, batchResults);
2402
+ }
2403
+ }
2404
+ if (hasInterruptingError) {
2405
+ if ((0, _langchain_langgraph.isGraphInterrupt)(interruptingError)) {
2406
+ const controllers = /* @__PURE__ */ new Set();
2407
+ const parentToolCallIds = /* @__PURE__ */ new Set();
2408
+ for (const call of directCalls) {
2409
+ const controller = this.toolMap.get(call.name)?.[require_SubagentReplay.SUBAGENT_REPLAY_CONTROLLER];
2410
+ if (controller?.getResumeManifest != null) controllers.add(controller);
2411
+ if (call.id != null && call.id !== "") parentToolCallIds.add(call.id);
2412
+ }
2413
+ const executionsByParentCall = /* @__PURE__ */ new Map();
2414
+ for (const controller of controllers) {
2415
+ const manifest = await controller.getResumeManifest?.(parentToolCallIds, withSubagentReplayBatch(config, baseContext.replayBatchKey));
2416
+ if (manifest != null) for (const execution of manifest.executions) executionsByParentCall.set(execution.parentToolCallId, execution);
2417
+ }
2418
+ const executions = [...executionsByParentCall.values()];
2419
+ if (executions.length > 0) {
2420
+ const manifest = {
2421
+ version: 1,
2422
+ executions
2423
+ };
2424
+ throw new _langchain_langgraph.GraphInterrupt(interruptingError.interrupts.map((pendingInterrupt) => ({
2425
+ ...pendingInterrupt,
2426
+ value: require_SubagentReplay.attachSubagentResumeManifest(pendingInterrupt.value, manifest)
2427
+ })));
2428
+ }
2429
+ }
2430
+ throw interruptingError;
2431
+ }
2225
2432
  /** The breaker can trip while the interrupting group is awaited — e.g.
2226
2433
  * a tool that ignores cancellation and completes normally. Recheck
2227
2434
  * before starting the non-idempotent siblings. */
2228
2435
  this.throwIfBreakerTripped(config);
2229
2436
  const regularOutputs = await Promise.all(regularPositions.map((i) => runOne(directCalls[i], i)));
2230
2437
  regularPositions.forEach((i, k) => {
2231
- outputs[i] = regularOutputs[k];
2438
+ outputs[i] = regularOutputs[k].output;
2232
2439
  });
2233
2440
  return outputs;
2234
2441
  }
@@ -2284,9 +2491,23 @@ var ToolNode = class extends require_run.RunnableCallable {
2284
2491
  const resolvedArgsByCallId = /* @__PURE__ */ new Map();
2285
2492
  /** Per-invocation error-completion ownership — see `ToolErrorOwnership`. */
2286
2493
  const errorOwnership = createToolErrorOwnership();
2287
- const batchScopeId = config.configurable?.run_id ?? `\0anon-${this.anonBatchCounter++}`;
2494
+ /**
2495
+ * Claim this batch's turn synchronously from the registry (or
2496
+ * fall back to 0 when the feature is disabled). The registry is
2497
+ * partitioned by scope id so overlapping batches cannot
2498
+ * overwrite each other's state even under a shared registry.
2499
+ *
2500
+ * For anonymous callers (no `run_id` in config), mint a unique
2501
+ * per-batch scope id so two concurrent anonymous invocations
2502
+ * don't target the same bucket. The scope is threaded down to
2503
+ * every subsequent registry call on this batch.
2504
+ */
2505
+ const incomingRunId = config.configurable?.run_id;
2506
+ const incomingThreadId = config.configurable?.thread_id;
2507
+ const batchScopeId = incomingRunId ?? `\0anon-${this.anonBatchCounter++}`;
2288
2508
  const turn = this.toolOutputRegistry?.nextTurn(batchScopeId) ?? 0;
2289
2509
  let outputs;
2510
+ let replayBatchKey;
2290
2511
  /** Hoisted from the messages-state branch so the Command tail can carry
2291
2512
  * the promotion into handoff updates (same-id state copies there would
2292
2513
  * otherwise overwrite the replacement message). */
@@ -2301,6 +2522,9 @@ var ToolNode = class extends require_run.RunnableCallable {
2301
2522
  });
2302
2523
  const directAdditionalContexts = [];
2303
2524
  const { lg_tool_call: _sendToolCall, ...sendState } = input;
2525
+ const sendMessages = sendState.messages;
2526
+ const sendAssistantBatch = Array.isArray(sendMessages) ? findAssistantBatch(sendMessages) : void 0;
2527
+ const sendReplayBatchKey = sendAssistantBatch == null ? void 0 : getAssistantBatchReplayKey(sendAssistantBatch, incomingRunId, incomingThreadId);
2304
2528
  const sendOutput = await this.runDirectToolWithLifecycleHooks(input.lg_tool_call, config, {
2305
2529
  batchIndex: 0,
2306
2530
  turn,
@@ -2308,6 +2532,7 @@ var ToolNode = class extends require_run.RunnableCallable {
2308
2532
  resolvedArgsByCallId,
2309
2533
  errorOwnership,
2310
2534
  additionalContextsSink: directAdditionalContexts,
2535
+ replayBatchKey: sendReplayBatchKey,
2311
2536
  runInput: sendState
2312
2537
  });
2313
2538
  outputs = directAdditionalContexts.length > 0 ? [sendOutput, new _langchain_core_messages.HumanMessage({
@@ -2324,15 +2549,10 @@ var ToolNode = class extends require_run.RunnableCallable {
2324
2549
  else if (this.isMessagesState(input)) messages = input.messages;
2325
2550
  else throw new Error("ToolNode only accepts BaseMessage[] or { messages: BaseMessage[] } as input.");
2326
2551
  const toolMessageIds = new Set(messages.filter((msg) => msg._getType() === "tool").map((msg) => msg.tool_call_id));
2327
- let aiMessage;
2328
- for (let i = messages.length - 1; i >= 0; i--) {
2329
- const message = messages[i];
2330
- if ((0, _langchain_core_messages.isAIMessage)(message)) {
2331
- aiMessage = message;
2332
- break;
2333
- }
2334
- }
2335
- if (aiMessage == null || !(0, _langchain_core_messages.isAIMessage)(aiMessage)) throw new Error("ToolNode only accepts AIMessages as input.");
2552
+ const assistantBatch = findAssistantBatch(messages);
2553
+ if (assistantBatch == null) throw new Error("ToolNode only accepts AIMessages as input.");
2554
+ const aiMessage = assistantBatch.message;
2555
+ replayBatchKey = getAssistantBatchReplayKey(assistantBatch, incomingRunId, incomingThreadId);
2336
2556
  if (this.loadRuntimeTools) {
2337
2557
  const { tools, toolMap } = this.loadRuntimeTools(aiMessage.tool_calls ?? []);
2338
2558
  this.toolMap = toolMap ?? new Map(tools.map((tool) => [tool.name, tool]));
@@ -2447,6 +2667,7 @@ var ToolNode = class extends require_run.RunnableCallable {
2447
2667
  errorOwnership,
2448
2668
  preBatchSnapshot,
2449
2669
  additionalContextsSink: directAdditionalContexts,
2670
+ replayBatchKey,
2450
2671
  runInput: input
2451
2672
  }) : [];
2452
2673
  if (directCalls.length > 0 && directOutputs.length > 0) await this.handleRunToolCompletions(directCalls, directOutputs, config, resolvedArgsByCallId, errorOwnership);
@@ -2485,6 +2706,7 @@ var ToolNode = class extends require_run.RunnableCallable {
2485
2706
  errorOwnership,
2486
2707
  preBatchSnapshot,
2487
2708
  additionalContextsSink: directAdditionalContexts,
2709
+ replayBatchKey,
2488
2710
  runInput: input
2489
2711
  });
2490
2712
  await this.handleRunToolCompletions(filteredCalls, toolOutputs, config, resolvedArgsByCallId, errorOwnership);
@@ -2521,7 +2743,10 @@ var ToolNode = class extends require_run.RunnableCallable {
2521
2743
  await this.dispatchStepCompleted(result.tool_call_id, result.name ?? "unknown", {}, typeof result.content === "string" ? result.content : "", config);
2522
2744
  }
2523
2745
  }
2524
- if (!outputs.some(_langchain_langgraph.isCommand)) return Array.isArray(input) ? outputs : { messages: outputs };
2746
+ if (!outputs.some(_langchain_langgraph.isCommand)) {
2747
+ if (replayBatchKey != null) this.settledInterruptingResults.delete(replayBatchKey);
2748
+ return Array.isArray(input) ? outputs : { messages: outputs };
2749
+ }
2525
2750
  /**
2526
2751
  * Carry the invalid-call promotion into handoff commands. A handoff
2527
2752
  * tool's Command snapshots `update.messages` from the PRE-promotion
@@ -2608,6 +2833,7 @@ var ToolNode = class extends require_run.RunnableCallable {
2608
2833
  /** Single handoff - pass through as-is */
2609
2834
  combinedOutputs.push(handoffCommands[0]);
2610
2835
  if (parentCommand) combinedOutputs.push(parentCommand);
2836
+ if (replayBatchKey != null) this.settledInterruptingResults.delete(replayBatchKey);
2611
2837
  return combinedOutputs;
2612
2838
  }
2613
2839
  isSendInput(input) {