@illuma-ai/agents 1.0.89 → 1.0.93

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 (838) hide show
  1. package/dist/cjs/agents/AgentContext.cjs +98 -49
  2. package/dist/cjs/agents/AgentContext.cjs.map +1 -1
  3. package/dist/cjs/common/constants.cjs +25 -0
  4. package/dist/cjs/common/constants.cjs.map +1 -0
  5. package/dist/cjs/common/enum.cjs +30 -0
  6. package/dist/cjs/common/enum.cjs.map +1 -1
  7. package/dist/cjs/events.cjs +9 -4
  8. package/dist/cjs/events.cjs.map +1 -1
  9. package/dist/cjs/graphs/Graph.cjs +397 -92
  10. package/dist/cjs/graphs/Graph.cjs.map +1 -1
  11. package/dist/cjs/graphs/MultiAgentGraph.cjs +223 -92
  12. package/dist/cjs/graphs/MultiAgentGraph.cjs.map +1 -1
  13. package/dist/cjs/instrumentation.cjs +30 -14
  14. package/dist/cjs/instrumentation.cjs.map +1 -1
  15. package/dist/cjs/llm/anthropic/index.cjs +43 -11
  16. package/dist/cjs/llm/anthropic/index.cjs.map +1 -1
  17. package/dist/cjs/llm/anthropic/types.cjs.map +1 -1
  18. package/dist/cjs/llm/anthropic/utils/message_inputs.cjs +10 -7
  19. package/dist/cjs/llm/anthropic/utils/message_inputs.cjs.map +1 -1
  20. package/dist/cjs/llm/anthropic/utils/message_outputs.cjs +32 -0
  21. package/dist/cjs/llm/anthropic/utils/message_outputs.cjs.map +1 -1
  22. package/dist/cjs/llm/anthropic/utils/tools.cjs.map +1 -1
  23. package/dist/cjs/llm/bedrock/index.cjs +129 -101
  24. package/dist/cjs/llm/bedrock/index.cjs.map +1 -1
  25. package/dist/cjs/llm/bedrock/utils/message_inputs.cjs +489 -0
  26. package/dist/cjs/llm/bedrock/utils/message_inputs.cjs.map +1 -0
  27. package/dist/cjs/llm/bedrock/utils/message_outputs.cjs +176 -0
  28. package/dist/cjs/llm/bedrock/utils/message_outputs.cjs.map +1 -0
  29. package/dist/cjs/llm/fake.cjs.map +1 -1
  30. package/dist/cjs/llm/google/index.cjs.map +1 -1
  31. package/dist/cjs/llm/google/utils/common.cjs.map +1 -1
  32. package/dist/cjs/llm/openai/index.cjs +1 -1
  33. package/dist/cjs/llm/openai/index.cjs.map +1 -1
  34. package/dist/cjs/llm/openai/utils/index.cjs.map +1 -1
  35. package/dist/cjs/llm/openrouter/index.cjs +59 -5
  36. package/dist/cjs/llm/openrouter/index.cjs.map +1 -1
  37. package/dist/cjs/llm/providers.cjs.map +1 -1
  38. package/dist/cjs/llm/text.cjs.map +1 -1
  39. package/dist/cjs/llm/vertexai/index.cjs +80 -2
  40. package/dist/cjs/llm/vertexai/index.cjs.map +1 -1
  41. package/dist/cjs/main.cjs +60 -27
  42. package/dist/cjs/main.cjs.map +1 -1
  43. package/dist/cjs/messages/cache.cjs +131 -108
  44. package/dist/cjs/messages/cache.cjs.map +1 -1
  45. package/dist/cjs/messages/content.cjs.map +1 -1
  46. package/dist/cjs/messages/core.cjs +3 -0
  47. package/dist/cjs/messages/core.cjs.map +1 -1
  48. package/dist/cjs/messages/format.cjs +265 -47
  49. package/dist/cjs/messages/format.cjs.map +1 -1
  50. package/dist/cjs/messages/ids.cjs.map +1 -1
  51. package/dist/cjs/messages/prune.cjs +55 -2
  52. package/dist/cjs/messages/prune.cjs.map +1 -1
  53. package/dist/cjs/messages/summarize.cjs +170 -0
  54. package/dist/cjs/messages/summarize.cjs.map +1 -0
  55. package/dist/cjs/messages/tools.cjs.map +1 -1
  56. package/dist/cjs/run.cjs +87 -30
  57. package/dist/cjs/run.cjs.map +1 -1
  58. package/dist/cjs/schemas/validate.cjs.map +1 -1
  59. package/dist/cjs/splitStream.cjs.map +1 -1
  60. package/dist/cjs/stream.cjs +59 -25
  61. package/dist/cjs/stream.cjs.map +1 -1
  62. package/dist/cjs/tools/AskUser.cjs +131 -0
  63. package/dist/cjs/tools/AskUser.cjs.map +1 -0
  64. package/dist/cjs/tools/BrowserTools.cjs +11 -7
  65. package/dist/cjs/tools/BrowserTools.cjs.map +1 -1
  66. package/dist/cjs/tools/Calculator.cjs.map +1 -1
  67. package/dist/cjs/tools/CodeExecutor.cjs +46 -4
  68. package/dist/cjs/tools/CodeExecutor.cjs.map +1 -1
  69. package/dist/cjs/tools/ProgrammaticToolCalling.cjs +36 -53
  70. package/dist/cjs/tools/ProgrammaticToolCalling.cjs.map +1 -1
  71. package/dist/cjs/tools/StreamingToolCallBuffer.cjs +208 -0
  72. package/dist/cjs/tools/StreamingToolCallBuffer.cjs.map +1 -0
  73. package/dist/cjs/tools/ToolNode.cjs +333 -30
  74. package/dist/cjs/tools/ToolNode.cjs.map +1 -1
  75. package/dist/cjs/tools/ToolSearch.cjs +66 -30
  76. package/dist/cjs/tools/ToolSearch.cjs.map +1 -1
  77. package/dist/cjs/tools/handlers.cjs +94 -8
  78. package/dist/cjs/tools/handlers.cjs.map +1 -1
  79. package/dist/cjs/tools/schema.cjs.map +1 -1
  80. package/dist/cjs/tools/search/content.cjs.map +1 -1
  81. package/dist/cjs/tools/search/firecrawl.cjs.map +1 -1
  82. package/dist/cjs/tools/search/format.cjs.map +1 -1
  83. package/dist/cjs/tools/search/highlights.cjs.map +1 -1
  84. package/dist/cjs/tools/search/rerankers.cjs.map +1 -1
  85. package/dist/cjs/tools/search/schema.cjs.map +1 -1
  86. package/dist/cjs/tools/search/search.cjs +1 -0
  87. package/dist/cjs/tools/search/search.cjs.map +1 -1
  88. package/dist/cjs/tools/search/serper-scraper.cjs.map +1 -1
  89. package/dist/cjs/tools/search/tool.cjs.map +1 -1
  90. package/dist/cjs/tools/search/utils.cjs.map +1 -1
  91. package/dist/cjs/types/graph.cjs +1 -1
  92. package/dist/cjs/types/graph.cjs.map +1 -1
  93. package/dist/cjs/utils/contextAnalytics.cjs +23 -6
  94. package/dist/cjs/utils/contextAnalytics.cjs.map +1 -1
  95. package/dist/cjs/utils/events.cjs.map +1 -1
  96. package/dist/cjs/utils/graph.cjs.map +1 -1
  97. package/dist/cjs/utils/handlers.cjs.map +1 -1
  98. package/dist/cjs/utils/llm.cjs.map +1 -1
  99. package/dist/cjs/utils/misc.cjs.map +1 -1
  100. package/dist/cjs/utils/run.cjs +3 -1
  101. package/dist/cjs/utils/run.cjs.map +1 -1
  102. package/dist/cjs/utils/schema.cjs.map +1 -1
  103. package/dist/cjs/utils/title.cjs.map +1 -1
  104. package/dist/cjs/utils/tokens.cjs +33 -58
  105. package/dist/cjs/utils/tokens.cjs.map +1 -1
  106. package/dist/cjs/utils/toolCallContinuation.cjs +55 -0
  107. package/dist/cjs/utils/toolCallContinuation.cjs.map +1 -0
  108. package/dist/cjs/utils/toonFormat.cjs.map +1 -1
  109. package/dist/esm/agents/AgentContext.mjs +98 -49
  110. package/dist/esm/agents/AgentContext.mjs.map +1 -1
  111. package/dist/esm/common/constants.mjs +22 -0
  112. package/dist/esm/common/constants.mjs.map +1 -0
  113. package/dist/esm/common/enum.mjs +31 -1
  114. package/dist/esm/common/enum.mjs.map +1 -1
  115. package/dist/esm/events.mjs +9 -4
  116. package/dist/esm/events.mjs.map +1 -1
  117. package/dist/esm/graphs/Graph.mjs +393 -88
  118. package/dist/esm/graphs/Graph.mjs.map +1 -1
  119. package/dist/esm/graphs/MultiAgentGraph.mjs +224 -93
  120. package/dist/esm/graphs/MultiAgentGraph.mjs.map +1 -1
  121. package/dist/esm/instrumentation.mjs +30 -14
  122. package/dist/esm/instrumentation.mjs.map +1 -1
  123. package/dist/esm/llm/anthropic/index.mjs +43 -11
  124. package/dist/esm/llm/anthropic/index.mjs.map +1 -1
  125. package/dist/esm/llm/anthropic/types.mjs.map +1 -1
  126. package/dist/esm/llm/anthropic/utils/message_inputs.mjs +10 -7
  127. package/dist/esm/llm/anthropic/utils/message_inputs.mjs.map +1 -1
  128. package/dist/esm/llm/anthropic/utils/message_outputs.mjs +32 -0
  129. package/dist/esm/llm/anthropic/utils/message_outputs.mjs.map +1 -1
  130. package/dist/esm/llm/anthropic/utils/tools.mjs.map +1 -1
  131. package/dist/esm/llm/bedrock/index.mjs +128 -101
  132. package/dist/esm/llm/bedrock/index.mjs.map +1 -1
  133. package/dist/esm/llm/bedrock/utils/message_inputs.mjs +484 -0
  134. package/dist/esm/llm/bedrock/utils/message_inputs.mjs.map +1 -0
  135. package/dist/esm/llm/bedrock/utils/message_outputs.mjs +171 -0
  136. package/dist/esm/llm/bedrock/utils/message_outputs.mjs.map +1 -0
  137. package/dist/esm/llm/fake.mjs.map +1 -1
  138. package/dist/esm/llm/google/index.mjs.map +1 -1
  139. package/dist/esm/llm/google/utils/common.mjs.map +1 -1
  140. package/dist/esm/llm/openai/index.mjs +1 -1
  141. package/dist/esm/llm/openai/index.mjs.map +1 -1
  142. package/dist/esm/llm/openai/utils/index.mjs.map +1 -1
  143. package/dist/esm/llm/openrouter/index.mjs +59 -5
  144. package/dist/esm/llm/openrouter/index.mjs.map +1 -1
  145. package/dist/esm/llm/providers.mjs.map +1 -1
  146. package/dist/esm/llm/text.mjs.map +1 -1
  147. package/dist/esm/llm/vertexai/index.mjs +80 -2
  148. package/dist/esm/llm/vertexai/index.mjs.map +1 -1
  149. package/dist/esm/main.mjs +8 -3
  150. package/dist/esm/main.mjs.map +1 -1
  151. package/dist/esm/messages/cache.mjs +131 -108
  152. package/dist/esm/messages/cache.mjs.map +1 -1
  153. package/dist/esm/messages/content.mjs.map +1 -1
  154. package/dist/esm/messages/core.mjs +4 -1
  155. package/dist/esm/messages/core.mjs.map +1 -1
  156. package/dist/esm/messages/format.mjs +267 -49
  157. package/dist/esm/messages/format.mjs.map +1 -1
  158. package/dist/esm/messages/ids.mjs.map +1 -1
  159. package/dist/esm/messages/prune.mjs +56 -4
  160. package/dist/esm/messages/prune.mjs.map +1 -1
  161. package/dist/esm/messages/summarize.mjs +161 -0
  162. package/dist/esm/messages/summarize.mjs.map +1 -0
  163. package/dist/esm/messages/tools.mjs.map +1 -1
  164. package/dist/esm/run.mjs +88 -31
  165. package/dist/esm/run.mjs.map +1 -1
  166. package/dist/esm/schemas/validate.mjs.map +1 -1
  167. package/dist/esm/splitStream.mjs.map +1 -1
  168. package/dist/esm/stream.mjs +60 -26
  169. package/dist/esm/stream.mjs.map +1 -1
  170. package/dist/esm/tools/AskUser.mjs +125 -0
  171. package/dist/esm/tools/AskUser.mjs.map +1 -0
  172. package/dist/esm/tools/BrowserTools.mjs +11 -7
  173. package/dist/esm/tools/BrowserTools.mjs.map +1 -1
  174. package/dist/esm/tools/Calculator.mjs.map +1 -1
  175. package/dist/esm/tools/CodeExecutor.mjs +46 -4
  176. package/dist/esm/tools/CodeExecutor.mjs.map +1 -1
  177. package/dist/esm/tools/ProgrammaticToolCalling.mjs +37 -54
  178. package/dist/esm/tools/ProgrammaticToolCalling.mjs.map +1 -1
  179. package/dist/esm/tools/StreamingToolCallBuffer.mjs +206 -0
  180. package/dist/esm/tools/StreamingToolCallBuffer.mjs.map +1 -0
  181. package/dist/esm/tools/ToolNode.mjs +333 -30
  182. package/dist/esm/tools/ToolNode.mjs.map +1 -1
  183. package/dist/esm/tools/ToolSearch.mjs +66 -30
  184. package/dist/esm/tools/ToolSearch.mjs.map +1 -1
  185. package/dist/esm/tools/handlers.mjs +95 -9
  186. package/dist/esm/tools/handlers.mjs.map +1 -1
  187. package/dist/esm/tools/schema.mjs.map +1 -1
  188. package/dist/esm/tools/search/content.mjs.map +1 -1
  189. package/dist/esm/tools/search/firecrawl.mjs.map +1 -1
  190. package/dist/esm/tools/search/format.mjs.map +1 -1
  191. package/dist/esm/tools/search/highlights.mjs.map +1 -1
  192. package/dist/esm/tools/search/rerankers.mjs.map +1 -1
  193. package/dist/esm/tools/search/schema.mjs.map +1 -1
  194. package/dist/esm/tools/search/search.mjs +1 -0
  195. package/dist/esm/tools/search/search.mjs.map +1 -1
  196. package/dist/esm/tools/search/serper-scraper.mjs.map +1 -1
  197. package/dist/esm/tools/search/tool.mjs.map +1 -1
  198. package/dist/esm/tools/search/utils.mjs.map +1 -1
  199. package/dist/esm/types/graph.mjs +1 -1
  200. package/dist/esm/types/graph.mjs.map +1 -1
  201. package/dist/esm/utils/contextAnalytics.mjs +23 -6
  202. package/dist/esm/utils/contextAnalytics.mjs.map +1 -1
  203. package/dist/esm/utils/events.mjs.map +1 -1
  204. package/dist/esm/utils/graph.mjs.map +1 -1
  205. package/dist/esm/utils/handlers.mjs.map +1 -1
  206. package/dist/esm/utils/llm.mjs.map +1 -1
  207. package/dist/esm/utils/misc.mjs.map +1 -1
  208. package/dist/esm/utils/run.mjs +3 -1
  209. package/dist/esm/utils/run.mjs.map +1 -1
  210. package/dist/esm/utils/schema.mjs.map +1 -1
  211. package/dist/esm/utils/title.mjs.map +1 -1
  212. package/dist/esm/utils/tokens.mjs +33 -59
  213. package/dist/esm/utils/tokens.mjs.map +1 -1
  214. package/dist/esm/utils/toolCallContinuation.mjs +52 -0
  215. package/dist/esm/utils/toolCallContinuation.mjs.map +1 -0
  216. package/dist/esm/utils/toonFormat.mjs.map +1 -1
  217. package/dist/types/agents/AgentContext.d.ts +14 -7
  218. package/dist/types/common/constants.d.ts +18 -0
  219. package/dist/types/common/enum.d.ts +28 -0
  220. package/dist/types/common/index.d.ts +1 -0
  221. package/dist/types/events.d.ts +10 -3
  222. package/dist/types/graphs/Graph.d.ts +37 -0
  223. package/dist/types/index.d.ts +4 -0
  224. package/dist/types/llm/anthropic/index.d.ts +7 -1
  225. package/dist/types/llm/anthropic/types.d.ts +5 -2
  226. package/dist/types/llm/anthropic/utils/message_outputs.d.ts +1 -1
  227. package/dist/types/llm/bedrock/index.d.ts +40 -33
  228. package/dist/types/llm/bedrock/utils/message_outputs.d.ts +1 -1
  229. package/dist/types/llm/google/index.d.ts +2 -3
  230. package/dist/types/llm/openrouter/index.d.ts +21 -1
  231. package/dist/types/llm/vertexai/index.d.ts +3 -2
  232. package/dist/types/messages/cache.d.ts +1 -1
  233. package/dist/types/messages/index.d.ts +1 -0
  234. package/dist/types/messages/prune.d.ts +2 -7
  235. package/dist/types/messages/summarize.d.ts +33 -0
  236. package/dist/types/run.d.ts +6 -0
  237. package/dist/types/tools/AskUser.d.ts +408 -0
  238. package/dist/types/tools/BrowserTools.d.ts +2 -2
  239. package/dist/types/tools/CodeExecutor.d.ts +2 -2
  240. package/dist/types/tools/StreamingToolCallBuffer.d.ts +106 -0
  241. package/dist/types/tools/ToolNode.d.ts +55 -3
  242. package/dist/types/tools/ToolSearch.d.ts +9 -5
  243. package/dist/types/tools/handlers.d.ts +2 -2
  244. package/dist/types/types/graph.d.ts +9 -2
  245. package/dist/types/types/llm.d.ts +8 -3
  246. package/dist/types/types/run.d.ts +2 -0
  247. package/dist/types/types/tools.d.ts +20 -2
  248. package/dist/types/utils/contextAnalytics.d.ts +5 -4
  249. package/dist/types/utils/index.d.ts +1 -0
  250. package/dist/types/utils/tokens.d.ts +6 -19
  251. package/dist/types/utils/toolCallContinuation.d.ts +30 -0
  252. package/package.json +15 -8
  253. package/src/agents/AgentContext.js +782 -0
  254. package/src/agents/AgentContext.js.map +1 -0
  255. package/src/agents/AgentContext.test.js +421 -0
  256. package/src/agents/AgentContext.test.js.map +1 -0
  257. package/src/agents/AgentContext.ts +132 -64
  258. package/src/agents/__tests__/AgentContext.test.js +678 -0
  259. package/src/agents/__tests__/AgentContext.test.js.map +1 -0
  260. package/src/agents/__tests__/AgentContext.test.ts +25 -4
  261. package/src/agents/__tests__/resolveStructuredOutputMode.test.js +117 -0
  262. package/src/agents/__tests__/resolveStructuredOutputMode.test.js.map +1 -0
  263. package/src/common/__tests__/enum.test.ts +135 -0
  264. package/src/common/constants.ts +21 -0
  265. package/src/common/enum.js +192 -0
  266. package/src/common/enum.js.map +1 -0
  267. package/src/common/enum.ts +30 -0
  268. package/src/common/index.js +3 -0
  269. package/src/common/index.js.map +1 -0
  270. package/src/common/index.ts +2 -1
  271. package/src/events.js +166 -0
  272. package/src/events.js.map +1 -0
  273. package/src/events.ts +11 -14
  274. package/src/graphs/Graph.js +1857 -0
  275. package/src/graphs/Graph.js.map +1 -0
  276. package/src/graphs/Graph.ts +580 -162
  277. package/src/graphs/MultiAgentGraph.js +1092 -0
  278. package/src/graphs/MultiAgentGraph.js.map +1 -0
  279. package/src/graphs/MultiAgentGraph.ts +331 -112
  280. package/src/graphs/__tests__/adaptive-thinking.test.ts +369 -0
  281. package/src/graphs/__tests__/graph-direct-tool-names.test.ts +210 -0
  282. package/src/graphs/__tests__/multi-agent-edges.test.ts +237 -0
  283. package/src/graphs/__tests__/structured-output.integration.test.js +624 -0
  284. package/src/graphs/__tests__/structured-output.integration.test.js.map +1 -0
  285. package/src/graphs/__tests__/structured-output.test.js +144 -0
  286. package/src/graphs/__tests__/structured-output.test.js.map +1 -0
  287. package/src/graphs/contextManagement.e2e.test.js +718 -0
  288. package/src/graphs/contextManagement.e2e.test.js.map +1 -0
  289. package/src/graphs/contextManagement.e2e.test.ts +990 -0
  290. package/src/graphs/contextManagement.test.js +485 -0
  291. package/src/graphs/contextManagement.test.js.map +1 -0
  292. package/src/graphs/contextManagement.test.ts +625 -0
  293. package/src/graphs/handoffValidation.test.js +276 -0
  294. package/src/graphs/handoffValidation.test.js.map +1 -0
  295. package/src/graphs/handoffValidation.test.ts +353 -0
  296. package/src/graphs/index.js +3 -0
  297. package/src/graphs/index.js.map +1 -0
  298. package/src/index.js +28 -0
  299. package/src/index.js.map +1 -0
  300. package/src/index.ts +13 -0
  301. package/src/instrumentation.js +21 -0
  302. package/src/instrumentation.js.map +1 -0
  303. package/src/instrumentation.ts +38 -17
  304. package/src/llm/anthropic/index.js +319 -0
  305. package/src/llm/anthropic/index.js.map +1 -0
  306. package/src/llm/anthropic/index.ts +68 -15
  307. package/src/llm/anthropic/llm.spec.ts +402 -0
  308. package/src/llm/anthropic/types.js +46 -0
  309. package/src/llm/anthropic/types.js.map +1 -0
  310. package/src/llm/anthropic/types.ts +8 -2
  311. package/src/llm/anthropic/utils/message_inputs.js +627 -0
  312. package/src/llm/anthropic/utils/message_inputs.js.map +1 -0
  313. package/src/llm/anthropic/utils/message_inputs.ts +16 -33
  314. package/src/llm/anthropic/utils/message_outputs.js +290 -0
  315. package/src/llm/anthropic/utils/message_outputs.js.map +1 -0
  316. package/src/llm/anthropic/utils/message_outputs.ts +40 -1
  317. package/src/llm/anthropic/utils/output_parsers.js +89 -0
  318. package/src/llm/anthropic/utils/output_parsers.js.map +1 -0
  319. package/src/llm/anthropic/utils/tools.js +25 -0
  320. package/src/llm/anthropic/utils/tools.js.map +1 -0
  321. package/src/llm/bedrock/__tests__/bedrock-caching.test.js +392 -0
  322. package/src/llm/bedrock/__tests__/bedrock-caching.test.js.map +1 -0
  323. package/src/llm/bedrock/__tests__/bedrock-caching.test.ts +24 -40
  324. package/src/llm/bedrock/index.js +303 -0
  325. package/src/llm/bedrock/index.js.map +1 -0
  326. package/src/llm/bedrock/index.ts +171 -134
  327. package/src/llm/bedrock/llm.spec.ts +395 -52
  328. package/src/llm/bedrock/types.js +2 -0
  329. package/src/llm/bedrock/types.js.map +1 -0
  330. package/src/llm/bedrock/utils/index.js +6 -0
  331. package/src/llm/bedrock/utils/index.js.map +1 -0
  332. package/src/llm/bedrock/utils/message_inputs.js +463 -0
  333. package/src/llm/bedrock/utils/message_inputs.js.map +1 -0
  334. package/src/llm/bedrock/utils/message_inputs.ts +30 -5
  335. package/src/llm/bedrock/utils/message_outputs.js +269 -0
  336. package/src/llm/bedrock/utils/message_outputs.js.map +1 -0
  337. package/src/llm/bedrock/utils/message_outputs.ts +70 -22
  338. package/src/llm/fake.js +92 -0
  339. package/src/llm/fake.js.map +1 -0
  340. package/src/llm/google/index.js +215 -0
  341. package/src/llm/google/index.js.map +1 -0
  342. package/src/llm/google/index.ts +2 -3
  343. package/src/llm/google/types.js +12 -0
  344. package/src/llm/google/types.js.map +1 -0
  345. package/src/llm/google/utils/common.js +670 -0
  346. package/src/llm/google/utils/common.js.map +1 -0
  347. package/src/llm/google/utils/tools.js +111 -0
  348. package/src/llm/google/utils/tools.js.map +1 -0
  349. package/src/llm/google/utils/zod_to_genai_parameters.js +47 -0
  350. package/src/llm/google/utils/zod_to_genai_parameters.js.map +1 -0
  351. package/src/llm/openai/index.js +1033 -0
  352. package/src/llm/openai/index.js.map +1 -0
  353. package/src/llm/openai/types.js +2 -0
  354. package/src/llm/openai/types.js.map +1 -0
  355. package/src/llm/openai/utils/index.js +756 -0
  356. package/src/llm/openai/utils/index.js.map +1 -0
  357. package/src/llm/openai/utils/isReasoningModel.test.js +79 -0
  358. package/src/llm/openai/utils/isReasoningModel.test.js.map +1 -0
  359. package/src/llm/openrouter/index.js +261 -0
  360. package/src/llm/openrouter/index.js.map +1 -0
  361. package/src/llm/openrouter/index.ts +117 -6
  362. package/src/llm/openrouter/reasoning.test.js +181 -0
  363. package/src/llm/openrouter/reasoning.test.js.map +1 -0
  364. package/src/llm/openrouter/reasoning.test.ts +207 -0
  365. package/src/llm/providers.js +36 -0
  366. package/src/llm/providers.js.map +1 -0
  367. package/src/llm/text.js +65 -0
  368. package/src/llm/text.js.map +1 -0
  369. package/src/llm/vertexai/index.js +402 -0
  370. package/src/llm/vertexai/index.js.map +1 -0
  371. package/src/llm/vertexai/index.ts +115 -5
  372. package/src/llm/vertexai/llm.spec.ts +114 -0
  373. package/src/messages/__tests__/tools.test.js +392 -0
  374. package/src/messages/__tests__/tools.test.js.map +1 -0
  375. package/src/messages/cache.js +404 -0
  376. package/src/messages/cache.js.map +1 -0
  377. package/src/messages/cache.test.js +1167 -0
  378. package/src/messages/cache.test.js.map +1 -0
  379. package/src/messages/cache.test.ts +178 -16
  380. package/src/messages/cache.ts +152 -147
  381. package/src/messages/content.js +48 -0
  382. package/src/messages/content.js.map +1 -0
  383. package/src/messages/content.test.js +314 -0
  384. package/src/messages/content.test.js.map +1 -0
  385. package/src/messages/core.js +359 -0
  386. package/src/messages/core.js.map +1 -0
  387. package/src/messages/core.ts +5 -0
  388. package/src/messages/ensureThinkingBlock.test.js +997 -0
  389. package/src/messages/ensureThinkingBlock.test.js.map +1 -0
  390. package/src/messages/ensureThinkingBlock.test.ts +751 -10
  391. package/src/messages/format.js +973 -0
  392. package/src/messages/format.js.map +1 -0
  393. package/src/messages/format.ts +334 -57
  394. package/src/messages/formatAgentMessages.test.js +2278 -0
  395. package/src/messages/formatAgentMessages.test.js.map +1 -0
  396. package/src/messages/formatAgentMessages.test.ts +1175 -1
  397. package/src/messages/formatAgentMessages.tools.test.js +362 -0
  398. package/src/messages/formatAgentMessages.tools.test.js.map +1 -0
  399. package/src/messages/formatMessage.test.js +608 -0
  400. package/src/messages/formatMessage.test.js.map +1 -0
  401. package/src/messages/ids.js +18 -0
  402. package/src/messages/ids.js.map +1 -0
  403. package/src/messages/index.js +9 -0
  404. package/src/messages/index.js.map +1 -0
  405. package/src/messages/index.ts +1 -0
  406. package/src/messages/labelContentByAgent.test.js +725 -0
  407. package/src/messages/labelContentByAgent.test.js.map +1 -0
  408. package/src/messages/prune.js +438 -0
  409. package/src/messages/prune.js.map +1 -0
  410. package/src/messages/prune.ts +87 -25
  411. package/src/messages/reducer.js +60 -0
  412. package/src/messages/reducer.js.map +1 -0
  413. package/src/messages/shiftIndexTokenCountMap.test.js +63 -0
  414. package/src/messages/shiftIndexTokenCountMap.test.js.map +1 -0
  415. package/src/messages/summarize.js +146 -0
  416. package/src/messages/summarize.js.map +1 -0
  417. package/src/messages/summarize.test.js +332 -0
  418. package/src/messages/summarize.test.js.map +1 -0
  419. package/src/messages/summarize.test.ts +466 -0
  420. package/src/messages/summarize.ts +222 -0
  421. package/src/messages/tools.js +90 -0
  422. package/src/messages/tools.js.map +1 -0
  423. package/src/mockStream.js +81 -0
  424. package/src/mockStream.js.map +1 -0
  425. package/src/prompts/collab.js +7 -0
  426. package/src/prompts/collab.js.map +1 -0
  427. package/src/prompts/index.js +3 -0
  428. package/src/prompts/index.js.map +1 -0
  429. package/src/prompts/taskmanager.js +58 -0
  430. package/src/prompts/taskmanager.js.map +1 -0
  431. package/src/run.js +427 -0
  432. package/src/run.js.map +1 -0
  433. package/src/run.ts +101 -33
  434. package/src/schemas/index.js +3 -0
  435. package/src/schemas/index.js.map +1 -0
  436. package/src/schemas/schema-preparation.test.js +370 -0
  437. package/src/schemas/schema-preparation.test.js.map +1 -0
  438. package/src/schemas/validate.js +314 -0
  439. package/src/schemas/validate.js.map +1 -0
  440. package/src/schemas/validate.test.js +264 -0
  441. package/src/schemas/validate.test.js.map +1 -0
  442. package/src/scripts/abort.js +127 -0
  443. package/src/scripts/abort.js.map +1 -0
  444. package/src/scripts/ant_web_search.js +130 -0
  445. package/src/scripts/ant_web_search.js.map +1 -0
  446. package/src/scripts/ant_web_search.ts +1 -0
  447. package/src/scripts/ant_web_search_edge_case.js +133 -0
  448. package/src/scripts/ant_web_search_edge_case.js.map +1 -0
  449. package/src/scripts/ant_web_search_edge_case.ts +1 -0
  450. package/src/scripts/ant_web_search_error_edge_case.js +119 -0
  451. package/src/scripts/ant_web_search_error_edge_case.js.map +1 -0
  452. package/src/scripts/ant_web_search_error_edge_case.ts +1 -0
  453. package/src/scripts/args.js +41 -0
  454. package/src/scripts/args.js.map +1 -0
  455. package/src/scripts/bedrock-cache-debug.js +186 -0
  456. package/src/scripts/bedrock-cache-debug.js.map +1 -0
  457. package/src/scripts/bedrock-cache-debug.ts +250 -0
  458. package/src/scripts/bedrock-content-aggregation-test.js +195 -0
  459. package/src/scripts/bedrock-content-aggregation-test.js.map +1 -0
  460. package/src/scripts/bedrock-content-aggregation-test.ts +266 -0
  461. package/src/scripts/bedrock-merge-test.js +80 -0
  462. package/src/scripts/bedrock-merge-test.js.map +1 -0
  463. package/src/scripts/bedrock-merge-test.ts +107 -0
  464. package/src/scripts/bedrock-parallel-tools-test.js +150 -0
  465. package/src/scripts/bedrock-parallel-tools-test.js.map +1 -0
  466. package/src/scripts/bedrock-parallel-tools-test.ts +204 -0
  467. package/src/scripts/caching.js +106 -0
  468. package/src/scripts/caching.js.map +1 -0
  469. package/src/scripts/caching.ts +1 -0
  470. package/src/scripts/cli.js +152 -0
  471. package/src/scripts/cli.js.map +1 -0
  472. package/src/scripts/cli2.js +119 -0
  473. package/src/scripts/cli2.js.map +1 -0
  474. package/src/scripts/cli3.js +163 -0
  475. package/src/scripts/cli3.js.map +1 -0
  476. package/src/scripts/cli4.js +165 -0
  477. package/src/scripts/cli4.js.map +1 -0
  478. package/src/scripts/cli5.js +165 -0
  479. package/src/scripts/cli5.js.map +1 -0
  480. package/src/scripts/code_exec.js +171 -0
  481. package/src/scripts/code_exec.js.map +1 -0
  482. package/src/scripts/code_exec.ts +1 -0
  483. package/src/scripts/code_exec_files.js +180 -0
  484. package/src/scripts/code_exec_files.js.map +1 -0
  485. package/src/scripts/code_exec_files.ts +1 -0
  486. package/src/scripts/code_exec_multi_session.js +185 -0
  487. package/src/scripts/code_exec_multi_session.js.map +1 -0
  488. package/src/scripts/code_exec_multi_session.ts +9 -13
  489. package/src/scripts/code_exec_ptc.js +265 -0
  490. package/src/scripts/code_exec_ptc.js.map +1 -0
  491. package/src/scripts/code_exec_ptc.ts +1 -0
  492. package/src/scripts/code_exec_session.js +217 -0
  493. package/src/scripts/code_exec_session.js.map +1 -0
  494. package/src/scripts/code_exec_session.ts +1 -0
  495. package/src/scripts/code_exec_simple.js +120 -0
  496. package/src/scripts/code_exec_simple.js.map +1 -0
  497. package/src/scripts/code_exec_simple.ts +1 -0
  498. package/src/scripts/content.js +111 -0
  499. package/src/scripts/content.js.map +1 -0
  500. package/src/scripts/content.ts +1 -0
  501. package/src/scripts/empty_input.js +125 -0
  502. package/src/scripts/empty_input.js.map +1 -0
  503. package/src/scripts/handoff-test.js +96 -0
  504. package/src/scripts/handoff-test.js.map +1 -0
  505. package/src/scripts/image.js +138 -0
  506. package/src/scripts/image.js.map +1 -0
  507. package/src/scripts/image.ts +3 -1
  508. package/src/scripts/memory.js +83 -0
  509. package/src/scripts/memory.js.map +1 -0
  510. package/src/scripts/memory.ts +16 -6
  511. package/src/scripts/multi-agent-chain.js +271 -0
  512. package/src/scripts/multi-agent-chain.js.map +1 -0
  513. package/src/scripts/multi-agent-chain.ts +1 -0
  514. package/src/scripts/multi-agent-conditional.js +185 -0
  515. package/src/scripts/multi-agent-conditional.js.map +1 -0
  516. package/src/scripts/multi-agent-conditional.ts +1 -0
  517. package/src/scripts/multi-agent-document-review-chain.js +171 -0
  518. package/src/scripts/multi-agent-document-review-chain.js.map +1 -0
  519. package/src/scripts/multi-agent-document-review-chain.ts +1 -0
  520. package/src/scripts/multi-agent-hybrid-flow.js +264 -0
  521. package/src/scripts/multi-agent-hybrid-flow.js.map +1 -0
  522. package/src/scripts/multi-agent-hybrid-flow.ts +1 -0
  523. package/src/scripts/multi-agent-parallel-start.js +214 -0
  524. package/src/scripts/multi-agent-parallel-start.js.map +1 -0
  525. package/src/scripts/multi-agent-parallel-start.ts +4 -4
  526. package/src/scripts/multi-agent-parallel.js +346 -0
  527. package/src/scripts/multi-agent-parallel.js.map +1 -0
  528. package/src/scripts/multi-agent-parallel.ts +1 -0
  529. package/src/scripts/multi-agent-sequence.js +184 -0
  530. package/src/scripts/multi-agent-sequence.js.map +1 -0
  531. package/src/scripts/multi-agent-sequence.ts +4 -4
  532. package/src/scripts/multi-agent-supervisor.js +324 -0
  533. package/src/scripts/multi-agent-supervisor.js.map +1 -0
  534. package/src/scripts/multi-agent-supervisor.ts +1 -0
  535. package/src/scripts/multi-agent-test.js +147 -0
  536. package/src/scripts/multi-agent-test.js.map +1 -0
  537. package/src/scripts/multi-agent-test.ts +1 -0
  538. package/src/scripts/parallel-asymmetric-tools-test.js +202 -0
  539. package/src/scripts/parallel-asymmetric-tools-test.js.map +1 -0
  540. package/src/scripts/parallel-asymmetric-tools-test.ts +1 -0
  541. package/src/scripts/parallel-full-metadata-test.js +176 -0
  542. package/src/scripts/parallel-full-metadata-test.js.map +1 -0
  543. package/src/scripts/parallel-full-metadata-test.ts +1 -0
  544. package/src/scripts/parallel-tools-test.js +256 -0
  545. package/src/scripts/parallel-tools-test.js.map +1 -0
  546. package/src/scripts/parallel-tools-test.ts +1 -0
  547. package/src/scripts/poc-multi-agent-comprehensive.ts +1222 -0
  548. package/src/scripts/programmatic_exec.js +277 -0
  549. package/src/scripts/programmatic_exec.js.map +1 -0
  550. package/src/scripts/programmatic_exec_agent.js +168 -0
  551. package/src/scripts/programmatic_exec_agent.js.map +1 -0
  552. package/src/scripts/programmatic_exec_agent.ts +1 -0
  553. package/src/scripts/search.js +118 -0
  554. package/src/scripts/search.js.map +1 -0
  555. package/src/scripts/search.ts +1 -0
  556. package/src/scripts/sequential-full-metadata-test.js +143 -0
  557. package/src/scripts/sequential-full-metadata-test.js.map +1 -0
  558. package/src/scripts/sequential-full-metadata-test.ts +1 -0
  559. package/src/scripts/simple.js +174 -0
  560. package/src/scripts/simple.js.map +1 -0
  561. package/src/scripts/simple.ts +2 -1
  562. package/src/scripts/single-agent-metadata-test.js +152 -0
  563. package/src/scripts/single-agent-metadata-test.js.map +1 -0
  564. package/src/scripts/single-agent-metadata-test.ts +4 -6
  565. package/src/scripts/stream.js +113 -0
  566. package/src/scripts/stream.js.map +1 -0
  567. package/src/scripts/stream.ts +1 -0
  568. package/src/scripts/test-custom-prompt-key.js +132 -0
  569. package/src/scripts/test-custom-prompt-key.js.map +1 -0
  570. package/src/scripts/test-handoff-input.js +143 -0
  571. package/src/scripts/test-handoff-input.js.map +1 -0
  572. package/src/scripts/test-handoff-preamble.js +227 -0
  573. package/src/scripts/test-handoff-preamble.js.map +1 -0
  574. package/src/scripts/test-handoff-preamble.ts +1 -0
  575. package/src/scripts/test-handoff-steering.js +353 -0
  576. package/src/scripts/test-handoff-steering.js.map +1 -0
  577. package/src/scripts/test-handoff-steering.ts +430 -0
  578. package/src/scripts/test-multi-agent-list-handoff.js +318 -0
  579. package/src/scripts/test-multi-agent-list-handoff.js.map +1 -0
  580. package/src/scripts/test-multi-agent-list-handoff.ts +1 -0
  581. package/src/scripts/test-parallel-agent-labeling.js +253 -0
  582. package/src/scripts/test-parallel-agent-labeling.js.map +1 -0
  583. package/src/scripts/test-parallel-agent-labeling.ts +2 -0
  584. package/src/scripts/test-parallel-handoffs.js +229 -0
  585. package/src/scripts/test-parallel-handoffs.js.map +1 -0
  586. package/src/scripts/test-parallel-handoffs.ts +1 -0
  587. package/src/scripts/test-thinking-handoff-bedrock.js +132 -0
  588. package/src/scripts/test-thinking-handoff-bedrock.js.map +1 -0
  589. package/src/scripts/test-thinking-handoff-bedrock.ts +1 -0
  590. package/src/scripts/test-thinking-handoff.js +132 -0
  591. package/src/scripts/test-thinking-handoff.js.map +1 -0
  592. package/src/scripts/test-thinking-handoff.ts +1 -0
  593. package/src/scripts/test-thinking-to-thinking-handoff-bedrock.js +140 -0
  594. package/src/scripts/test-thinking-to-thinking-handoff-bedrock.js.map +1 -0
  595. package/src/scripts/test-thinking-to-thinking-handoff-bedrock.ts +166 -0
  596. package/src/scripts/test-tool-before-handoff-role-order.js +223 -0
  597. package/src/scripts/test-tool-before-handoff-role-order.js.map +1 -0
  598. package/src/scripts/test-tool-before-handoff-role-order.ts +276 -0
  599. package/src/scripts/test-tools-before-handoff.js +187 -0
  600. package/src/scripts/test-tools-before-handoff.js.map +1 -0
  601. package/src/scripts/test-tools-before-handoff.ts +4 -8
  602. package/src/scripts/test_code_api.js +263 -0
  603. package/src/scripts/test_code_api.js.map +1 -0
  604. package/src/scripts/thinking-bedrock.js +128 -0
  605. package/src/scripts/thinking-bedrock.js.map +1 -0
  606. package/src/scripts/thinking-bedrock.ts +1 -0
  607. package/src/scripts/thinking-vertexai.js +130 -0
  608. package/src/scripts/thinking-vertexai.js.map +1 -0
  609. package/src/scripts/thinking-vertexai.ts +168 -0
  610. package/src/scripts/thinking.js +134 -0
  611. package/src/scripts/thinking.js.map +1 -0
  612. package/src/scripts/thinking.ts +1 -0
  613. package/src/scripts/tool_search.js +114 -0
  614. package/src/scripts/tool_search.js.map +1 -0
  615. package/src/scripts/tools.js +125 -0
  616. package/src/scripts/tools.js.map +1 -0
  617. package/src/scripts/tools.ts +5 -19
  618. package/src/specs/agent-handoffs-bedrock.integration.test.js +280 -0
  619. package/src/specs/agent-handoffs-bedrock.integration.test.js.map +1 -0
  620. package/src/specs/agent-handoffs-bedrock.integration.test.ts +412 -375
  621. package/src/specs/agent-handoffs.test.js +924 -0
  622. package/src/specs/agent-handoffs.test.js.map +1 -0
  623. package/src/specs/agent-handoffs.test.ts +152 -39
  624. package/src/specs/anthropic.simple.test.js +287 -0
  625. package/src/specs/anthropic.simple.test.js.map +1 -0
  626. package/src/specs/anthropic.simple.test.ts +7 -4
  627. package/src/specs/azure.simple.test.js +381 -0
  628. package/src/specs/azure.simple.test.js.map +1 -0
  629. package/src/specs/azure.simple.test.ts +143 -5
  630. package/src/specs/cache.simple.test.js +282 -0
  631. package/src/specs/cache.simple.test.js.map +1 -0
  632. package/src/specs/cache.simple.test.ts +9 -2
  633. package/src/specs/custom-event-await.test.js +148 -0
  634. package/src/specs/custom-event-await.test.js.map +1 -0
  635. package/src/specs/custom-event-await.test.ts +215 -0
  636. package/src/specs/deepseek.simple.test.js +189 -0
  637. package/src/specs/deepseek.simple.test.js.map +1 -0
  638. package/src/specs/deepseek.simple.test.ts +4 -2
  639. package/src/specs/emergency-prune.test.js +308 -0
  640. package/src/specs/emergency-prune.test.js.map +1 -0
  641. package/src/specs/moonshot.simple.test.js +237 -0
  642. package/src/specs/moonshot.simple.test.js.map +1 -0
  643. package/src/specs/moonshot.simple.test.ts +6 -2
  644. package/src/specs/observability.integration.test.js +1337 -0
  645. package/src/specs/observability.integration.test.js.map +1 -0
  646. package/src/specs/observability.integration.test.ts +2223 -0
  647. package/src/specs/openai.simple.test.js +233 -0
  648. package/src/specs/openai.simple.test.js.map +1 -0
  649. package/src/specs/openai.simple.test.ts +4 -2
  650. package/src/specs/openrouter.simple.test.js +202 -0
  651. package/src/specs/openrouter.simple.test.js.map +1 -0
  652. package/src/specs/openrouter.simple.test.ts +165 -4
  653. package/src/specs/prune.test.js +733 -0
  654. package/src/specs/prune.test.js.map +1 -0
  655. package/src/specs/prune.test.ts +1 -0
  656. package/src/specs/reasoning.test.js +144 -0
  657. package/src/specs/reasoning.test.js.map +1 -0
  658. package/src/specs/reasoning.test.ts +2 -2
  659. package/src/specs/spec.utils.js +4 -0
  660. package/src/specs/spec.utils.js.map +1 -0
  661. package/src/specs/thinking-handoff.test.js +486 -0
  662. package/src/specs/thinking-handoff.test.js.map +1 -0
  663. package/src/specs/thinking-handoff.test.ts +3 -2
  664. package/src/specs/thinking-prune.test.js +600 -0
  665. package/src/specs/thinking-prune.test.js.map +1 -0
  666. package/src/specs/token-distribution-edge-case.test.js +246 -0
  667. package/src/specs/token-distribution-edge-case.test.js.map +1 -0
  668. package/src/specs/token-memoization.test.js +32 -0
  669. package/src/specs/token-memoization.test.js.map +1 -0
  670. package/src/specs/token-memoization.test.ts +14 -5
  671. package/src/specs/tokens.test.js +49 -0
  672. package/src/specs/tokens.test.js.map +1 -0
  673. package/src/specs/tokens.test.ts +64 -0
  674. package/src/specs/tool-error.test.js +139 -0
  675. package/src/specs/tool-error.test.js.map +1 -0
  676. package/src/specs/tool-error.test.ts +2 -2
  677. package/src/splitStream.js +204 -0
  678. package/src/splitStream.js.map +1 -0
  679. package/src/splitStream.test.js +504 -0
  680. package/src/splitStream.test.js.map +1 -0
  681. package/src/stream.js +650 -0
  682. package/src/stream.js.map +1 -0
  683. package/src/stream.test.js +225 -0
  684. package/src/stream.test.js.map +1 -0
  685. package/src/stream.test.ts +25 -15
  686. package/src/stream.ts +82 -32
  687. package/src/test/mockTools.js +340 -0
  688. package/src/test/mockTools.js.map +1 -0
  689. package/src/tools/AskUser.ts +159 -0
  690. package/src/tools/BrowserTools.js +245 -0
  691. package/src/tools/BrowserTools.js.map +1 -0
  692. package/src/tools/BrowserTools.ts +12 -8
  693. package/src/tools/Calculator.js +38 -0
  694. package/src/tools/Calculator.js.map +1 -0
  695. package/src/tools/Calculator.test.js +225 -0
  696. package/src/tools/Calculator.test.js.map +1 -0
  697. package/src/tools/CodeExecutor.js +233 -0
  698. package/src/tools/CodeExecutor.js.map +1 -0
  699. package/src/tools/CodeExecutor.selfhealing.test.ts +435 -0
  700. package/src/tools/CodeExecutor.ts +62 -4
  701. package/src/tools/ProgrammaticToolCalling.js +602 -0
  702. package/src/tools/ProgrammaticToolCalling.js.map +1 -0
  703. package/src/tools/ProgrammaticToolCalling.ts +40 -52
  704. package/src/tools/StreamingToolCallBuffer.js +179 -0
  705. package/src/tools/StreamingToolCallBuffer.js.map +1 -0
  706. package/src/tools/StreamingToolCallBuffer.ts +218 -0
  707. package/src/tools/ToolNode.js +930 -0
  708. package/src/tools/ToolNode.js.map +1 -0
  709. package/src/tools/ToolNode.ts +454 -41
  710. package/src/tools/ToolSearch.js +904 -0
  711. package/src/tools/ToolSearch.js.map +1 -0
  712. package/src/tools/ToolSearch.ts +84 -33
  713. package/src/tools/__tests__/AskUser.test.ts +537 -0
  714. package/src/tools/__tests__/BrowserTools.test.js +306 -0
  715. package/src/tools/__tests__/BrowserTools.test.js.map +1 -0
  716. package/src/tools/__tests__/BrowserTools.test.ts +131 -6
  717. package/src/tools/__tests__/ProgrammaticToolCalling.integration.test.js +276 -0
  718. package/src/tools/__tests__/ProgrammaticToolCalling.integration.test.js.map +1 -0
  719. package/src/tools/__tests__/ProgrammaticToolCalling.test.js +807 -0
  720. package/src/tools/__tests__/ProgrammaticToolCalling.test.js.map +1 -0
  721. package/src/tools/__tests__/StreamingToolCallBuffer.test.js +175 -0
  722. package/src/tools/__tests__/StreamingToolCallBuffer.test.js.map +1 -0
  723. package/src/tools/__tests__/StreamingToolCallBuffer.test.ts +263 -0
  724. package/src/tools/__tests__/ToolApproval.test.js +675 -0
  725. package/src/tools/__tests__/ToolApproval.test.js.map +1 -0
  726. package/src/tools/__tests__/ToolApproval.test.ts +194 -20
  727. package/src/tools/__tests__/ToolNode.hitl.test.ts +267 -0
  728. package/src/tools/__tests__/ToolNode.recovery.test.js +200 -0
  729. package/src/tools/__tests__/ToolNode.recovery.test.js.map +1 -0
  730. package/src/tools/__tests__/ToolNode.recovery.test.ts +276 -0
  731. package/src/tools/__tests__/ToolNode.session.test.js +319 -0
  732. package/src/tools/__tests__/ToolNode.session.test.js.map +1 -0
  733. package/src/tools/__tests__/ToolNode.session.test.ts +465 -0
  734. package/src/tools/__tests__/ToolSearch.integration.test.js +125 -0
  735. package/src/tools/__tests__/ToolSearch.integration.test.js.map +1 -0
  736. package/src/tools/__tests__/ToolSearch.test.js +812 -0
  737. package/src/tools/__tests__/ToolSearch.test.js.map +1 -0
  738. package/src/tools/__tests__/ToolSearch.test.ts +78 -5
  739. package/src/tools/__tests__/handlers.test.js +799 -0
  740. package/src/tools/__tests__/handlers.test.js.map +1 -0
  741. package/src/tools/__tests__/handlers.test.ts +1100 -0
  742. package/src/tools/__tests__/truncation-recovery.integration.test.js +362 -0
  743. package/src/tools/__tests__/truncation-recovery.integration.test.js.map +1 -0
  744. package/src/tools/__tests__/truncation-recovery.integration.test.ts +560 -0
  745. package/src/tools/handlers.js +306 -0
  746. package/src/tools/handlers.js.map +1 -0
  747. package/src/tools/handlers.ts +119 -16
  748. package/src/tools/schema.js +25 -0
  749. package/src/tools/schema.js.map +1 -0
  750. package/src/tools/search/anthropic.js +34 -0
  751. package/src/tools/search/anthropic.js.map +1 -0
  752. package/src/tools/search/content.js +116 -0
  753. package/src/tools/search/content.js.map +1 -0
  754. package/src/tools/search/content.test.js +133 -0
  755. package/src/tools/search/content.test.js.map +1 -0
  756. package/src/tools/search/firecrawl.js +173 -0
  757. package/src/tools/search/firecrawl.js.map +1 -0
  758. package/src/tools/search/format.js +198 -0
  759. package/src/tools/search/format.js.map +1 -0
  760. package/src/tools/search/highlights.js +241 -0
  761. package/src/tools/search/highlights.js.map +1 -0
  762. package/src/tools/search/index.js +3 -0
  763. package/src/tools/search/index.js.map +1 -0
  764. package/src/tools/search/jina-reranker.test.js +106 -0
  765. package/src/tools/search/jina-reranker.test.js.map +1 -0
  766. package/src/tools/search/rerankers.js +165 -0
  767. package/src/tools/search/rerankers.js.map +1 -0
  768. package/src/tools/search/schema.js +102 -0
  769. package/src/tools/search/schema.js.map +1 -0
  770. package/src/tools/search/search.js +561 -0
  771. package/src/tools/search/search.js.map +1 -0
  772. package/src/tools/search/serper-scraper.js +126 -0
  773. package/src/tools/search/serper-scraper.js.map +1 -0
  774. package/src/tools/search/test.js +129 -0
  775. package/src/tools/search/test.js.map +1 -0
  776. package/src/tools/search/tool.js +453 -0
  777. package/src/tools/search/tool.js.map +1 -0
  778. package/src/tools/search/types.js +2 -0
  779. package/src/tools/search/types.js.map +1 -0
  780. package/src/tools/search/utils.js +59 -0
  781. package/src/tools/search/utils.js.map +1 -0
  782. package/src/types/graph.js +24 -0
  783. package/src/types/graph.js.map +1 -0
  784. package/src/types/graph.test.js +192 -0
  785. package/src/types/graph.test.js.map +1 -0
  786. package/src/types/graph.ts +26 -6
  787. package/src/types/index.js +7 -0
  788. package/src/types/index.js.map +1 -0
  789. package/src/types/llm.js +2 -0
  790. package/src/types/llm.js.map +1 -0
  791. package/src/types/llm.ts +8 -3
  792. package/src/types/messages.js +2 -0
  793. package/src/types/messages.js.map +1 -0
  794. package/src/types/run.js +2 -0
  795. package/src/types/run.js.map +1 -0
  796. package/src/types/run.ts +2 -0
  797. package/src/types/stream.js +2 -0
  798. package/src/types/stream.js.map +1 -0
  799. package/src/types/tools.js +2 -0
  800. package/src/types/tools.js.map +1 -0
  801. package/src/types/tools.ts +21 -2
  802. package/src/utils/contextAnalytics.js +79 -0
  803. package/src/utils/contextAnalytics.js.map +1 -0
  804. package/src/utils/contextAnalytics.test.js +166 -0
  805. package/src/utils/contextAnalytics.test.js.map +1 -0
  806. package/src/utils/contextAnalytics.test.ts +222 -0
  807. package/src/utils/contextAnalytics.ts +27 -9
  808. package/src/utils/events.js +26 -0
  809. package/src/utils/events.js.map +1 -0
  810. package/src/utils/graph.js +11 -0
  811. package/src/utils/graph.js.map +1 -0
  812. package/src/utils/handlers.js +65 -0
  813. package/src/utils/handlers.js.map +1 -0
  814. package/src/utils/index.js +10 -0
  815. package/src/utils/index.js.map +1 -0
  816. package/src/utils/index.ts +1 -0
  817. package/src/utils/llm.js +21 -0
  818. package/src/utils/llm.js.map +1 -0
  819. package/src/utils/llmConfig.js +205 -0
  820. package/src/utils/llmConfig.js.map +1 -0
  821. package/src/utils/llmConfig.ts +5 -5
  822. package/src/utils/logging.js +37 -0
  823. package/src/utils/logging.js.map +1 -0
  824. package/src/utils/misc.js +51 -0
  825. package/src/utils/misc.js.map +1 -0
  826. package/src/utils/run.js +69 -0
  827. package/src/utils/run.js.map +1 -0
  828. package/src/utils/run.ts +108 -106
  829. package/src/utils/schema.js +21 -0
  830. package/src/utils/schema.js.map +1 -0
  831. package/src/utils/title.js +119 -0
  832. package/src/utils/title.js.map +1 -0
  833. package/src/utils/tokens.js +92 -0
  834. package/src/utils/tokens.js.map +1 -0
  835. package/src/utils/tokens.ts +118 -142
  836. package/src/utils/toolCallContinuation.ts +55 -0
  837. package/src/utils/toonFormat.js +379 -0
  838. package/src/utils/toonFormat.js.map +1 -0
@@ -0,0 +1,186 @@
1
+ /**
2
+ * Debug script to investigate cache token omission in Bedrock responses.
3
+ *
4
+ * This script:
5
+ * 1. Makes a streaming call to Bedrock and logs the raw metadata event
6
+ * 2. Shows exactly what fields the AWS SDK returns in usage (including cache tokens)
7
+ * 3. Shows what our handleConverseStreamMetadata produces vs what it should produce
8
+ * 4. Makes a multi-turn call to trigger caching and verify cache tokens appear
9
+ */
10
+ import { config } from 'dotenv';
11
+ config();
12
+ import { concat } from '@langchain/core/utils/stream';
13
+ import { HumanMessage } from '@langchain/core/messages';
14
+ import { BedrockRuntimeClient, ConverseStreamCommand, } from '@aws-sdk/client-bedrock-runtime';
15
+ import { CustomChatBedrockConverse } from '@/llm/bedrock';
16
+ const region = process.env.BEDROCK_AWS_REGION ?? 'us-east-1';
17
+ const credentials = {
18
+ accessKeyId: process.env.BEDROCK_AWS_ACCESS_KEY_ID,
19
+ secretAccessKey: process.env.BEDROCK_AWS_SECRET_ACCESS_KEY,
20
+ };
21
+ const MODEL_ID = 'us.anthropic.claude-sonnet-4-5-20250929-v1:0';
22
+ // A long system prompt to increase likelihood of cache usage
23
+ // Bedrock requires minimum 1024 tokens for prompt caching to activate
24
+ const SYSTEM_PROMPT = `You are an expert assistant. Here is a large context block to help trigger cache behavior:
25
+
26
+ ${Array(200).fill('This is padding content to make the prompt large enough to trigger Bedrock prompt caching. The minimum requirement for Anthropic models on Bedrock is 1024 tokens in the cached prefix. We need to ensure this prompt is well above that threshold. ').join('')}
27
+
28
+ When answering, be brief and direct.`;
29
+ async function rawSdkCall() {
30
+ console.log('='.repeat(60));
31
+ console.log('TEST 1: Raw AWS SDK call - inspect metadata.usage directly');
32
+ console.log('='.repeat(60));
33
+ const client = new BedrockRuntimeClient({ region, credentials });
34
+ // First call - should create cache
35
+ // Use cachePoint block to explicitly enable prompt caching
36
+ console.log('\n--- Call 1 (cache write expected) ---');
37
+ const command1 = new ConverseStreamCommand({
38
+ modelId: MODEL_ID,
39
+ system: [{ text: SYSTEM_PROMPT }, { cachePoint: { type: 'default' } }],
40
+ messages: [{ role: 'user', content: [{ text: 'What is 2+2?' }] }],
41
+ inferenceConfig: { maxTokens: 100 },
42
+ });
43
+ const response1 = await client.send(command1);
44
+ if (response1.stream) {
45
+ for await (const event of response1.stream) {
46
+ if (event.metadata != null) {
47
+ console.log('\nRAW metadata event (Call 1):');
48
+ console.dir(event.metadata, { depth: null });
49
+ console.log('\nRAW metadata.usage:');
50
+ console.dir(event.metadata.usage, { depth: null });
51
+ console.log('\nSpecific cache fields:');
52
+ console.log(' cacheReadInputTokens:', event.metadata.usage
53
+ ?.cacheReadInputTokens);
54
+ console.log(' cacheWriteInputTokens:', event.metadata.usage
55
+ ?.cacheWriteInputTokens);
56
+ }
57
+ }
58
+ }
59
+ // Second call - should read from cache
60
+ console.log('\n--- Call 2 (cache read expected) ---');
61
+ const command2 = new ConverseStreamCommand({
62
+ modelId: MODEL_ID,
63
+ system: [{ text: SYSTEM_PROMPT }, { cachePoint: { type: 'default' } }],
64
+ messages: [
65
+ { role: 'user', content: [{ text: 'What is 2+2?' }] },
66
+ { role: 'assistant', content: [{ text: '4' }] },
67
+ { role: 'user', content: [{ text: 'And what is 3+3?' }] },
68
+ ],
69
+ inferenceConfig: { maxTokens: 100 },
70
+ });
71
+ const response2 = await client.send(command2);
72
+ if (response2.stream) {
73
+ for await (const event of response2.stream) {
74
+ if (event.metadata != null) {
75
+ console.log('\nRAW metadata event (Call 2):');
76
+ console.dir(event.metadata, { depth: null });
77
+ console.log('\nRAW metadata.usage:');
78
+ console.dir(event.metadata.usage, { depth: null });
79
+ console.log('\nSpecific cache fields:');
80
+ console.log(' cacheReadInputTokens:', event.metadata.usage
81
+ ?.cacheReadInputTokens);
82
+ console.log(' cacheWriteInputTokens:', event.metadata.usage
83
+ ?.cacheWriteInputTokens);
84
+ }
85
+ }
86
+ }
87
+ }
88
+ async function wrapperStreamCallNoCachePoint() {
89
+ console.log('\n' + '='.repeat(60));
90
+ console.log('TEST 2: CustomChatBedrockConverse stream (NO cachePoint) - check usage_metadata');
91
+ console.log('='.repeat(60));
92
+ console.log('(Without cachePoint, Bedrock does NOT return cache tokens)');
93
+ const model = new CustomChatBedrockConverse({
94
+ model: MODEL_ID,
95
+ region,
96
+ credentials,
97
+ maxTokens: 100,
98
+ streaming: true,
99
+ streamUsage: true,
100
+ });
101
+ console.log('\n--- Wrapper Call (no cachePoint) ---');
102
+ const messages1 = [new HumanMessage(SYSTEM_PROMPT + '\n\nWhat is 2+2?')];
103
+ let finalChunk1;
104
+ for await (const chunk of await model.stream(messages1)) {
105
+ finalChunk1 = finalChunk1 ? concat(finalChunk1, chunk) : chunk;
106
+ }
107
+ console.log('\nFinal usage_metadata:', JSON.stringify(finalChunk1.usage_metadata));
108
+ console.log('(No cache tokens expected since no cachePoint block was sent)');
109
+ }
110
+ async function wrapperStreamCallWithCachePoint() {
111
+ console.log('\n' + '='.repeat(60));
112
+ console.log('TEST 3: Raw SDK with cachePoint -> verify handleConverseStreamMetadata extracts cache tokens');
113
+ console.log('='.repeat(60));
114
+ // We use the raw SDK with cachePoint to trigger caching, then verify
115
+ // that our handleConverseStreamMetadata function properly extracts cache fields
116
+ const { handleConverseStreamMetadata } = await import('@/llm/bedrock/utils/message_outputs');
117
+ const client = new BedrockRuntimeClient({ region, credentials });
118
+ // Call 1 - establish cache
119
+ console.log('\n--- Call 1 (cache write) ---');
120
+ const command1 = new ConverseStreamCommand({
121
+ modelId: MODEL_ID,
122
+ system: [{ text: SYSTEM_PROMPT }, { cachePoint: { type: 'default' } }],
123
+ messages: [{ role: 'user', content: [{ text: 'What is 2+2?' }] }],
124
+ inferenceConfig: { maxTokens: 100 },
125
+ });
126
+ const response1 = await client.send(command1);
127
+ if (response1.stream) {
128
+ for await (const event of response1.stream) {
129
+ if (event.metadata != null) {
130
+ console.log('Raw usage:', JSON.stringify(event.metadata.usage));
131
+ // Test our handler
132
+ const chunk = handleConverseStreamMetadata(event.metadata, {
133
+ streamUsage: true,
134
+ });
135
+ const msg = chunk.message;
136
+ console.log('handleConverseStreamMetadata output usage_metadata:', JSON.stringify(msg.usage_metadata));
137
+ const hasDetails = msg.usage_metadata?.input_token_details != null;
138
+ console.log(`Has input_token_details: ${hasDetails}`, hasDetails
139
+ ? JSON.stringify(msg.usage_metadata.input_token_details)
140
+ : '(MISSING - BUG!)');
141
+ }
142
+ }
143
+ }
144
+ // Call 2 - read from cache
145
+ console.log('\n--- Call 2 (cache read) ---');
146
+ const command2 = new ConverseStreamCommand({
147
+ modelId: MODEL_ID,
148
+ system: [{ text: SYSTEM_PROMPT }, { cachePoint: { type: 'default' } }],
149
+ messages: [
150
+ { role: 'user', content: [{ text: 'What is 2+2?' }] },
151
+ { role: 'assistant', content: [{ text: '4' }] },
152
+ { role: 'user', content: [{ text: 'What is 3+3?' }] },
153
+ ],
154
+ inferenceConfig: { maxTokens: 100 },
155
+ });
156
+ const response2 = await client.send(command2);
157
+ if (response2.stream) {
158
+ for await (const event of response2.stream) {
159
+ if (event.metadata != null) {
160
+ console.log('Raw usage:', JSON.stringify(event.metadata.usage));
161
+ const chunk = handleConverseStreamMetadata(event.metadata, {
162
+ streamUsage: true,
163
+ });
164
+ const msg = chunk.message;
165
+ console.log('handleConverseStreamMetadata output usage_metadata:', JSON.stringify(msg.usage_metadata));
166
+ const hasDetails = msg.usage_metadata?.input_token_details != null;
167
+ console.log(`Has input_token_details: ${hasDetails}`, hasDetails
168
+ ? JSON.stringify(msg.usage_metadata.input_token_details)
169
+ : '(MISSING - BUG!)');
170
+ }
171
+ }
172
+ }
173
+ }
174
+ async function main() {
175
+ console.log('Bedrock Cache Token Debug Script');
176
+ console.log(`Model: ${MODEL_ID}`);
177
+ console.log(`Region: ${region}\n`);
178
+ await rawSdkCall();
179
+ await wrapperStreamCallNoCachePoint();
180
+ await wrapperStreamCallWithCachePoint();
181
+ }
182
+ main().catch((err) => {
183
+ console.error('Fatal error:', err);
184
+ process.exit(1);
185
+ });
186
+ //# sourceMappingURL=bedrock-cache-debug.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"bedrock-cache-debug.js","sourceRoot":"","sources":["bedrock-cache-debug.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,MAAM,EAAE,MAAM,8BAA8B,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EACL,oBAAoB,EACpB,qBAAqB,GACtB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,EAAE,yBAAyB,EAAE,MAAM,eAAe,CAAC;AAE1D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,WAAW,CAAC;AAC7D,MAAM,WAAW,GAAG;IAClB,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,yBAA0B;IACnD,eAAe,EAAE,OAAO,CAAC,GAAG,CAAC,6BAA8B;CAC5D,CAAC;AAEF,MAAM,QAAQ,GAAG,8CAA8C,CAAC;AAEhE,6DAA6D;AAC7D,sEAAsE;AACtE,MAAM,aAAa,GAAG;;EAEpB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,sPAAsP,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;;qCAE7O,CAAC;AAEtC,KAAK,UAAU,UAAU;IACvB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,OAAO,CAAC,GAAG,CAAC,4DAA4D,CAAC,CAAC;IAC1E,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IAEjE,mCAAmC;IACnC,2DAA2D;IAC3D,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;IACvD,MAAM,QAAQ,GAAG,IAAI,qBAAqB,CAAC;QACzC,OAAO,EAAE,QAAQ;QACjB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC;QACtE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC;QACjE,eAAe,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE;KACpC,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;gBAC3B,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;gBAC9C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC7C,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;gBACrC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnD,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,OAAO,CAAC,GAAG,CACT,yBAAyB,EACxB,KAAK,CAAC,QAAQ,CAAC,KAA4C;oBAC1D,EAAE,oBAAoB,CACzB,CAAC;gBACF,OAAO,CAAC,GAAG,CACT,0BAA0B,EACzB,KAAK,CAAC,QAAQ,CAAC,KAA4C;oBAC1D,EAAE,qBAAqB,CAC1B,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,uCAAuC;IACvC,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IACtD,MAAM,QAAQ,GAAG,IAAI,qBAAqB,CAAC;QACzC,OAAO,EAAE,QAAQ;QACjB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC;QACtE,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,EAAE;YACrD,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE;YAC/C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,EAAE;SAC1D;QACD,eAAe,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE;KACpC,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;gBAC3B,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;gBAC9C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC7C,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;gBACrC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACnD,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBACxC,OAAO,CAAC,GAAG,CACT,yBAAyB,EACxB,KAAK,CAAC,QAAQ,CAAC,KAA4C;oBAC1D,EAAE,oBAAoB,CACzB,CAAC;gBACF,OAAO,CAAC,GAAG,CACT,0BAA0B,EACzB,KAAK,CAAC,QAAQ,CAAC,KAA4C;oBAC1D,EAAE,qBAAqB,CAC1B,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,6BAA6B;IAC1C,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CACT,iFAAiF,CAClF,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5B,OAAO,CAAC,GAAG,CAAC,4DAA4D,CAAC,CAAC;IAE1E,MAAM,KAAK,GAAG,IAAI,yBAAyB,CAAC;QAC1C,KAAK,EAAE,QAAQ;QACf,MAAM;QACN,WAAW;QACX,SAAS,EAAE,GAAG;QACd,SAAS,EAAE,IAAI;QACf,WAAW,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IACtD,MAAM,SAAS,GAAG,CAAC,IAAI,YAAY,CAAC,aAAa,GAAG,kBAAkB,CAAC,CAAC,CAAC;IACzE,IAAI,WAAuC,CAAC;IAE5C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;QACxD,WAAW,GAAG,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IACjE,CAAC;IAED,OAAO,CAAC,GAAG,CACT,yBAAyB,EACzB,IAAI,CAAC,SAAS,CAAC,WAAY,CAAC,cAAc,CAAC,CAC5C,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,+DAA+D,CAAC,CAAC;AAC/E,CAAC;AAED,KAAK,UAAU,+BAA+B;IAC5C,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACnC,OAAO,CAAC,GAAG,CACT,8FAA8F,CAC/F,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAE5B,qEAAqE;IACrE,gFAAgF;IAChF,MAAM,EAAE,4BAA4B,EAAE,GAAG,MAAM,MAAM,CACnD,qCAAqC,CACtC,CAAC;IAEF,MAAM,MAAM,GAAG,IAAI,oBAAoB,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IAEjE,2BAA2B;IAC3B,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;IAC9C,MAAM,QAAQ,GAAG,IAAI,qBAAqB,CAAC;QACzC,OAAO,EAAE,QAAQ;QACjB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC;QACtE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC;QACjE,eAAe,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE;KACpC,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;gBAC3B,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;gBAEhE,mBAAmB;gBACnB,MAAM,KAAK,GAAG,4BAA4B,CAAC,KAAK,CAAC,QAAQ,EAAE;oBACzD,WAAW,EAAE,IAAI;iBAClB,CAAC,CAAC;gBACH,MAAM,GAAG,GAAG,KAAK,CAAC,OAAyB,CAAC;gBAC5C,OAAO,CAAC,GAAG,CACT,qDAAqD,EACrD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC,CACnC,CAAC;gBAEF,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,EAAE,mBAAmB,IAAI,IAAI,CAAC;gBACnE,OAAO,CAAC,GAAG,CACT,4BAA4B,UAAU,EAAE,EACxC,UAAU;oBACR,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAe,CAAC,mBAAmB,CAAC;oBACzD,CAAC,CAAC,kBAAkB,CACvB,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,2BAA2B;IAC3B,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,IAAI,qBAAqB,CAAC;QACzC,OAAO,EAAE,QAAQ;QACjB,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC;QACtE,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,EAAE;YACrD,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE;YAC/C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,EAAE;SACtD;QACD,eAAe,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE;KACpC,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;QACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;gBAC3B,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;gBAEhE,MAAM,KAAK,GAAG,4BAA4B,CAAC,KAAK,CAAC,QAAQ,EAAE;oBACzD,WAAW,EAAE,IAAI;iBAClB,CAAC,CAAC;gBACH,MAAM,GAAG,GAAG,KAAK,CAAC,OAAyB,CAAC;gBAC5C,OAAO,CAAC,GAAG,CACT,qDAAqD,EACrD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAc,CAAC,CACnC,CAAC;gBAEF,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,EAAE,mBAAmB,IAAI,IAAI,CAAC;gBACnE,OAAO,CAAC,GAAG,CACT,4BAA4B,UAAU,EAAE,EACxC,UAAU;oBACR,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,cAAe,CAAC,mBAAmB,CAAC;oBACzD,CAAC,CAAC,kBAAkB,CACvB,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,IAAI;IACjB,OAAO,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;IAChD,OAAO,CAAC,GAAG,CAAC,UAAU,QAAQ,EAAE,CAAC,CAAC;IAClC,OAAO,CAAC,GAAG,CAAC,WAAW,MAAM,IAAI,CAAC,CAAC;IAEnC,MAAM,UAAU,EAAE,CAAC;IACnB,MAAM,6BAA6B,EAAE,CAAC;IACtC,MAAM,+BAA+B,EAAE,CAAC;AAC1C,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;IACnC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}
@@ -0,0 +1,250 @@
1
+ /**
2
+ * Debug script to investigate cache token omission in Bedrock responses.
3
+ *
4
+ * This script:
5
+ * 1. Makes a streaming call to Bedrock and logs the raw metadata event
6
+ * 2. Shows exactly what fields the AWS SDK returns in usage (including cache tokens)
7
+ * 3. Shows what our handleConverseStreamMetadata produces vs what it should produce
8
+ * 4. Makes a multi-turn call to trigger caching and verify cache tokens appear
9
+ */
10
+ import { config } from 'dotenv';
11
+ config();
12
+ import { concat } from '@langchain/core/utils/stream';
13
+ import { HumanMessage } from '@langchain/core/messages';
14
+ import {
15
+ BedrockRuntimeClient,
16
+ ConverseStreamCommand,
17
+ } from '@aws-sdk/client-bedrock-runtime';
18
+ import type { AIMessageChunk } from '@langchain/core/messages';
19
+ import { CustomChatBedrockConverse } from '@/llm/bedrock';
20
+
21
+ const region = process.env.BEDROCK_AWS_REGION ?? 'us-east-1';
22
+ const credentials = {
23
+ accessKeyId: process.env.BEDROCK_AWS_ACCESS_KEY_ID!,
24
+ secretAccessKey: process.env.BEDROCK_AWS_SECRET_ACCESS_KEY!,
25
+ };
26
+
27
+ const MODEL_ID = 'us.anthropic.claude-sonnet-4-5-20250929-v1:0';
28
+
29
+ // A long system prompt to increase likelihood of cache usage
30
+ // Bedrock requires minimum 1024 tokens for prompt caching to activate
31
+ const SYSTEM_PROMPT = `You are an expert assistant. Here is a large context block to help trigger cache behavior:
32
+
33
+ ${Array(200).fill('This is padding content to make the prompt large enough to trigger Bedrock prompt caching. The minimum requirement for Anthropic models on Bedrock is 1024 tokens in the cached prefix. We need to ensure this prompt is well above that threshold. ').join('')}
34
+
35
+ When answering, be brief and direct.`;
36
+
37
+ async function rawSdkCall(): Promise<void> {
38
+ console.log('='.repeat(60));
39
+ console.log('TEST 1: Raw AWS SDK call - inspect metadata.usage directly');
40
+ console.log('='.repeat(60));
41
+
42
+ const client = new BedrockRuntimeClient({ region, credentials });
43
+
44
+ // First call - should create cache
45
+ // Use cachePoint block to explicitly enable prompt caching
46
+ console.log('\n--- Call 1 (cache write expected) ---');
47
+ const command1 = new ConverseStreamCommand({
48
+ modelId: MODEL_ID,
49
+ system: [{ text: SYSTEM_PROMPT }, { cachePoint: { type: 'default' } }],
50
+ messages: [{ role: 'user', content: [{ text: 'What is 2+2?' }] }],
51
+ inferenceConfig: { maxTokens: 100 },
52
+ });
53
+
54
+ const response1 = await client.send(command1);
55
+ if (response1.stream) {
56
+ for await (const event of response1.stream) {
57
+ if (event.metadata != null) {
58
+ console.log('\nRAW metadata event (Call 1):');
59
+ console.dir(event.metadata, { depth: null });
60
+ console.log('\nRAW metadata.usage:');
61
+ console.dir(event.metadata.usage, { depth: null });
62
+ console.log('\nSpecific cache fields:');
63
+ console.log(
64
+ ' cacheReadInputTokens:',
65
+ (event.metadata.usage as unknown as Record<string, unknown>)
66
+ ?.cacheReadInputTokens
67
+ );
68
+ console.log(
69
+ ' cacheWriteInputTokens:',
70
+ (event.metadata.usage as unknown as Record<string, unknown>)
71
+ ?.cacheWriteInputTokens
72
+ );
73
+ }
74
+ }
75
+ }
76
+
77
+ // Second call - should read from cache
78
+ console.log('\n--- Call 2 (cache read expected) ---');
79
+ const command2 = new ConverseStreamCommand({
80
+ modelId: MODEL_ID,
81
+ system: [{ text: SYSTEM_PROMPT }, { cachePoint: { type: 'default' } }],
82
+ messages: [
83
+ { role: 'user', content: [{ text: 'What is 2+2?' }] },
84
+ { role: 'assistant', content: [{ text: '4' }] },
85
+ { role: 'user', content: [{ text: 'And what is 3+3?' }] },
86
+ ],
87
+ inferenceConfig: { maxTokens: 100 },
88
+ });
89
+
90
+ const response2 = await client.send(command2);
91
+ if (response2.stream) {
92
+ for await (const event of response2.stream) {
93
+ if (event.metadata != null) {
94
+ console.log('\nRAW metadata event (Call 2):');
95
+ console.dir(event.metadata, { depth: null });
96
+ console.log('\nRAW metadata.usage:');
97
+ console.dir(event.metadata.usage, { depth: null });
98
+ console.log('\nSpecific cache fields:');
99
+ console.log(
100
+ ' cacheReadInputTokens:',
101
+ (event.metadata.usage as unknown as Record<string, unknown>)
102
+ ?.cacheReadInputTokens
103
+ );
104
+ console.log(
105
+ ' cacheWriteInputTokens:',
106
+ (event.metadata.usage as unknown as Record<string, unknown>)
107
+ ?.cacheWriteInputTokens
108
+ );
109
+ }
110
+ }
111
+ }
112
+ }
113
+
114
+ async function wrapperStreamCallNoCachePoint(): Promise<void> {
115
+ console.log('\n' + '='.repeat(60));
116
+ console.log(
117
+ 'TEST 2: CustomChatBedrockConverse stream (NO cachePoint) - check usage_metadata'
118
+ );
119
+ console.log('='.repeat(60));
120
+ console.log('(Without cachePoint, Bedrock does NOT return cache tokens)');
121
+
122
+ const model = new CustomChatBedrockConverse({
123
+ model: MODEL_ID,
124
+ region,
125
+ credentials,
126
+ maxTokens: 100,
127
+ streaming: true,
128
+ streamUsage: true,
129
+ });
130
+
131
+ console.log('\n--- Wrapper Call (no cachePoint) ---');
132
+ const messages1 = [new HumanMessage(SYSTEM_PROMPT + '\n\nWhat is 2+2?')];
133
+ let finalChunk1: AIMessageChunk | undefined;
134
+
135
+ for await (const chunk of await model.stream(messages1)) {
136
+ finalChunk1 = finalChunk1 ? concat(finalChunk1, chunk) : chunk;
137
+ }
138
+
139
+ console.log(
140
+ '\nFinal usage_metadata:',
141
+ JSON.stringify(finalChunk1!.usage_metadata)
142
+ );
143
+ console.log('(No cache tokens expected since no cachePoint block was sent)');
144
+ }
145
+
146
+ async function wrapperStreamCallWithCachePoint(): Promise<void> {
147
+ console.log('\n' + '='.repeat(60));
148
+ console.log(
149
+ 'TEST 3: Raw SDK with cachePoint -> verify handleConverseStreamMetadata extracts cache tokens'
150
+ );
151
+ console.log('='.repeat(60));
152
+
153
+ // We use the raw SDK with cachePoint to trigger caching, then verify
154
+ // that our handleConverseStreamMetadata function properly extracts cache fields
155
+ const { handleConverseStreamMetadata } = await import(
156
+ '@/llm/bedrock/utils/message_outputs'
157
+ );
158
+
159
+ const client = new BedrockRuntimeClient({ region, credentials });
160
+
161
+ // Call 1 - establish cache
162
+ console.log('\n--- Call 1 (cache write) ---');
163
+ const command1 = new ConverseStreamCommand({
164
+ modelId: MODEL_ID,
165
+ system: [{ text: SYSTEM_PROMPT }, { cachePoint: { type: 'default' } }],
166
+ messages: [{ role: 'user', content: [{ text: 'What is 2+2?' }] }],
167
+ inferenceConfig: { maxTokens: 100 },
168
+ });
169
+
170
+ const response1 = await client.send(command1);
171
+ if (response1.stream) {
172
+ for await (const event of response1.stream) {
173
+ if (event.metadata != null) {
174
+ console.log('Raw usage:', JSON.stringify(event.metadata.usage));
175
+
176
+ // Test our handler
177
+ const chunk = handleConverseStreamMetadata(event.metadata, {
178
+ streamUsage: true,
179
+ });
180
+ const msg = chunk.message as AIMessageChunk;
181
+ console.log(
182
+ 'handleConverseStreamMetadata output usage_metadata:',
183
+ JSON.stringify(msg.usage_metadata)
184
+ );
185
+
186
+ const hasDetails = msg.usage_metadata?.input_token_details != null;
187
+ console.log(
188
+ `Has input_token_details: ${hasDetails}`,
189
+ hasDetails
190
+ ? JSON.stringify(msg.usage_metadata!.input_token_details)
191
+ : '(MISSING - BUG!)'
192
+ );
193
+ }
194
+ }
195
+ }
196
+
197
+ // Call 2 - read from cache
198
+ console.log('\n--- Call 2 (cache read) ---');
199
+ const command2 = new ConverseStreamCommand({
200
+ modelId: MODEL_ID,
201
+ system: [{ text: SYSTEM_PROMPT }, { cachePoint: { type: 'default' } }],
202
+ messages: [
203
+ { role: 'user', content: [{ text: 'What is 2+2?' }] },
204
+ { role: 'assistant', content: [{ text: '4' }] },
205
+ { role: 'user', content: [{ text: 'What is 3+3?' }] },
206
+ ],
207
+ inferenceConfig: { maxTokens: 100 },
208
+ });
209
+
210
+ const response2 = await client.send(command2);
211
+ if (response2.stream) {
212
+ for await (const event of response2.stream) {
213
+ if (event.metadata != null) {
214
+ console.log('Raw usage:', JSON.stringify(event.metadata.usage));
215
+
216
+ const chunk = handleConverseStreamMetadata(event.metadata, {
217
+ streamUsage: true,
218
+ });
219
+ const msg = chunk.message as AIMessageChunk;
220
+ console.log(
221
+ 'handleConverseStreamMetadata output usage_metadata:',
222
+ JSON.stringify(msg.usage_metadata)
223
+ );
224
+
225
+ const hasDetails = msg.usage_metadata?.input_token_details != null;
226
+ console.log(
227
+ `Has input_token_details: ${hasDetails}`,
228
+ hasDetails
229
+ ? JSON.stringify(msg.usage_metadata!.input_token_details)
230
+ : '(MISSING - BUG!)'
231
+ );
232
+ }
233
+ }
234
+ }
235
+ }
236
+
237
+ async function main(): Promise<void> {
238
+ console.log('Bedrock Cache Token Debug Script');
239
+ console.log(`Model: ${MODEL_ID}`);
240
+ console.log(`Region: ${region}\n`);
241
+
242
+ await rawSdkCall();
243
+ await wrapperStreamCallNoCachePoint();
244
+ await wrapperStreamCallWithCachePoint();
245
+ }
246
+
247
+ main().catch((err) => {
248
+ console.error('Fatal error:', err);
249
+ process.exit(1);
250
+ });
@@ -0,0 +1,195 @@
1
+ import { config } from 'dotenv';
2
+ config();
3
+ import { HumanMessage } from '@langchain/core/messages';
4
+ import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
5
+ import { createCodeExecutionTool } from '@/tools/CodeExecutor';
6
+ import { ToolEndHandler, ModelEndHandler } from '@/events';
7
+ import { GraphEvents, ContentTypes, Providers } from '@/common';
8
+ import { getLLMConfig } from '@/utils/llmConfig';
9
+ import { Run } from '@/run';
10
+ const conversationHistory = [];
11
+ let _contentParts = [];
12
+ const collectedUsage = [];
13
+ async function testBedrockContentAggregation() {
14
+ const instructions = 'You are a helpful AI assistant with coding capabilities. When answering questions, be thorough in your reasoning.';
15
+ const { contentParts, aggregateContent } = createContentAggregator();
16
+ _contentParts = contentParts;
17
+ const customHandlers = {
18
+ [GraphEvents.TOOL_END]: new ToolEndHandler(),
19
+ [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
20
+ [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
21
+ [GraphEvents.ON_RUN_STEP_COMPLETED]: {
22
+ handle: (event, data) => {
23
+ const result = data.result;
24
+ console.log(`[ON_RUN_STEP_COMPLETED] stepId=${result.id} index=${result.index} type=${result.type} tool=${result.tool_call?.name ?? 'n/a'}`);
25
+ aggregateContent({
26
+ event,
27
+ data: data,
28
+ });
29
+ },
30
+ },
31
+ [GraphEvents.ON_RUN_STEP]: {
32
+ handle: (event, data) => {
33
+ const toolCalls = data.stepDetails.type === 'tool_calls' && data.stepDetails.tool_calls
34
+ ? data.stepDetails.tool_calls
35
+ .map((tc) => `${tc.name ?? '?'}(${tc.id ?? '?'})`)
36
+ .join(', ')
37
+ : 'none';
38
+ console.log(`[ON_RUN_STEP] stepId=${data.id} index=${data.index} type=${data.type} stepIndex=${data.stepIndex} toolCalls=[${toolCalls}]`);
39
+ aggregateContent({ event, data });
40
+ },
41
+ },
42
+ [GraphEvents.ON_RUN_STEP_DELTA]: {
43
+ handle: (event, data) => {
44
+ const tcNames = data.delta.tool_calls
45
+ ?.map((tc) => `${tc.name ?? '?'}(args=${(tc.args ?? '').substring(0, 30)}...)`)
46
+ .join(', ') ?? 'none';
47
+ console.log(`[ON_RUN_STEP_DELTA] stepId=${data.id} type=${data.delta.type} toolCalls=[${tcNames}]`);
48
+ aggregateContent({ event, data });
49
+ },
50
+ },
51
+ [GraphEvents.ON_MESSAGE_DELTA]: {
52
+ handle: (event, data) => {
53
+ const preview = Array.isArray(data.delta.content)
54
+ ? data.delta.content
55
+ .map((c) => `${c.type}:"${String(c.text ?? c.think ?? '').substring(0, 40)}"`)
56
+ .join(', ')
57
+ : String(data.delta.content).substring(0, 40);
58
+ console.log(`[ON_MESSAGE_DELTA] stepId=${data.id} content=[${preview}]`);
59
+ aggregateContent({ event, data });
60
+ },
61
+ },
62
+ [GraphEvents.ON_REASONING_DELTA]: {
63
+ handle: (event, data) => {
64
+ const preview = Array.isArray(data.delta.content)
65
+ ? data.delta.content
66
+ .map((c) => `${c.type}:"${String(c.think ?? '').substring(0, 40)}"`)
67
+ .join(', ')
68
+ : '?';
69
+ console.log(`[ON_REASONING_DELTA] stepId=${data.id} content=[${preview}]`);
70
+ aggregateContent({ event, data });
71
+ },
72
+ },
73
+ };
74
+ const baseLlmConfig = getLLMConfig(Providers.BEDROCK);
75
+ const llmConfig = {
76
+ ...baseLlmConfig,
77
+ model: 'global.anthropic.claude-opus-4-6-v1',
78
+ maxTokens: 16000,
79
+ additionalModelRequestFields: {
80
+ thinking: { type: 'enabled', budget_tokens: 10000 },
81
+ },
82
+ };
83
+ const run = await Run.create({
84
+ runId: 'bedrock-content-aggregation-test',
85
+ graphConfig: {
86
+ instructions,
87
+ type: 'standard',
88
+ tools: [createCodeExecutionTool()],
89
+ llmConfig,
90
+ },
91
+ returnContent: true,
92
+ skipCleanup: true,
93
+ customHandlers: customHandlers,
94
+ });
95
+ const streamConfig = {
96
+ configurable: {
97
+ thread_id: 'bedrock-content-aggregation-thread',
98
+ },
99
+ streamMode: 'values',
100
+ version: 'v2',
101
+ };
102
+ const userMessage = `im testing edge cases with our code interpreter. i know we can persist files, but what happens when we put them in directories?`;
103
+ conversationHistory.push(new HumanMessage(userMessage));
104
+ console.log('Running Bedrock content aggregation test...\n');
105
+ console.log(`Prompt: "${userMessage}"\n`);
106
+ const inputs = { messages: [...conversationHistory] };
107
+ await run.processStream(inputs, streamConfig);
108
+ console.log('\n\n========== CONTENT PARTS ANALYSIS ==========\n');
109
+ let hasEmptyToolCall = false;
110
+ let hasReasoningOrderIssue = false;
111
+ for (let i = 0; i < _contentParts.length; i++) {
112
+ const part = _contentParts[i];
113
+ if (!part) {
114
+ console.log(` [${i}] undefined`);
115
+ continue;
116
+ }
117
+ const partType = part.type;
118
+ if (partType === ContentTypes.TOOL_CALL) {
119
+ const tc = part.tool_call;
120
+ if (!tc || !tc.name) {
121
+ hasEmptyToolCall = true;
122
+ console.log(` [${i}] TOOL_CALL *** EMPTY (no tool_call data) ***`);
123
+ }
124
+ else {
125
+ const outputPreview = tc.output
126
+ ? `output=${tc.output.substring(0, 80)}...`
127
+ : 'no output';
128
+ console.log(` [${i}] TOOL_CALL name=${tc.name} ${outputPreview}`);
129
+ }
130
+ }
131
+ else if (partType === ContentTypes.THINK) {
132
+ const think = part.think ?? '';
133
+ console.log(` [${i}] THINK (${think.length} chars): "${think.substring(0, 80)}..."`);
134
+ }
135
+ else if (partType === ContentTypes.TEXT) {
136
+ const text = part.text ?? '';
137
+ console.log(` [${i}] TEXT (${text.length} chars): "${text.substring(0, 80)}..."`);
138
+ }
139
+ else {
140
+ console.log(` [${i}] ${partType}`);
141
+ }
142
+ }
143
+ /**
144
+ * Check reasoning ordering within a single invocation cycle.
145
+ * A tool_call resets the cycle — text before think across different
146
+ * invocations (e.g., text from invocation 2, think from invocation 3) is valid.
147
+ */
148
+ let lastTextInCycle = null;
149
+ for (let i = 0; i < _contentParts.length; i++) {
150
+ const part = _contentParts[i];
151
+ if (!part)
152
+ continue;
153
+ if (part.type === ContentTypes.TOOL_CALL) {
154
+ lastTextInCycle = null;
155
+ continue;
156
+ }
157
+ if (part.type === ContentTypes.TEXT) {
158
+ lastTextInCycle = i;
159
+ }
160
+ else if (part.type === ContentTypes.THINK && lastTextInCycle !== null) {
161
+ const prevText = _contentParts[lastTextInCycle];
162
+ const thinkContent = part.think ?? '';
163
+ if (prevText?.text &&
164
+ prevText.text.trim().length > 5 &&
165
+ thinkContent.length > 0) {
166
+ hasReasoningOrderIssue = true;
167
+ console.log(`\n *** ORDERING ISSUE (same invocation): TEXT at [${lastTextInCycle}] appears before THINK at [${i}]`);
168
+ console.log(` Text ends with: "...${prevText.text.substring(prevText.text.length - 60)}"`);
169
+ console.log(` Think starts with: "${thinkContent.substring(0, 60)}..."`);
170
+ }
171
+ }
172
+ }
173
+ console.log('\n========== SUMMARY ==========\n');
174
+ console.log(`Total content parts: ${_contentParts.filter(Boolean).length}`);
175
+ console.log(`Empty tool_call parts: ${hasEmptyToolCall ? 'YES (BUG)' : 'No'}`);
176
+ console.log(`Reasoning order issues: ${hasReasoningOrderIssue ? 'YES (BUG)' : 'No'}`);
177
+ console.log('\nFull contentParts dump:');
178
+ console.dir(_contentParts, { depth: null });
179
+ }
180
+ process.on('unhandledRejection', (reason, promise) => {
181
+ console.error('Unhandled Rejection at:', promise, 'reason:', reason);
182
+ console.log('Content parts:');
183
+ console.dir(_contentParts, { depth: null });
184
+ process.exit(1);
185
+ });
186
+ process.on('uncaughtException', (err) => {
187
+ console.error('Uncaught Exception:', err);
188
+ });
189
+ testBedrockContentAggregation().catch((err) => {
190
+ console.error(err);
191
+ console.log('Content parts:');
192
+ console.dir(_contentParts, { depth: null });
193
+ process.exit(1);
194
+ });
195
+ //# sourceMappingURL=bedrock-content-aggregation-test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"bedrock-content-aggregation-test.js","sourceRoot":"","sources":["bedrock-content-aggregation-test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AAGrE,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAC3E,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAChE,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,MAAM,mBAAmB,GAAkB,EAAE,CAAC;AAC9C,IAAI,aAAa,GAA8B,EAAE,CAAC;AAClD,MAAM,cAAc,GAAoB,EAAE,CAAC;AAE3C,KAAK,UAAU,6BAA6B;IAC1C,MAAM,YAAY,GAChB,mHAAmH,CAAC;IACtH,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;IACrE,aAAa,GAAG,YAAyC,CAAC;IAE1D,MAAM,cAAc,GAAG;QACrB,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;QAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,CAAC,cAAc,CAAC;QACjE,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,sBAAsB,EAAE;QAC7D,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;YACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACjB,EAAE;gBACR,MAAM,MAAM,GAAI,IAA8C,CAAC,MAAM,CAAC;gBACtE,OAAO,CAAC,GAAG,CACT,kCAAkC,MAAM,CAAC,EAAE,UAAU,MAAM,CAAC,KAAK,SAAS,MAAM,CAAC,IAAI,SAAS,MAAM,CAAC,SAAS,EAAE,IAAI,IAAI,KAAK,EAAE,CAChI,CAAC;gBACF,gBAAgB,CAAC;oBACf,KAAK;oBACL,IAAI,EAAE,IAA6C;iBACpD,CAAC,CAAC;YACL,CAAC;SACF;QACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;YACzB,MAAM,EAAE,CAAC,KAA8B,EAAE,IAAe,EAAE,EAAE;gBAC1D,MAAM,SAAS,GACb,IAAI,CAAC,WAAW,CAAC,IAAI,KAAK,YAAY,IAAI,IAAI,CAAC,WAAW,CAAC,UAAU;oBACnE,CAAC,CACG,IAAI,CAAC,WAAW,CAAC,UAIlB;yBACE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,GAAG,IAAI,EAAE,CAAC,EAAE,IAAI,GAAG,GAAG,CAAC;yBACjD,IAAI,CAAC,IAAI,CAAC;oBACf,CAAC,CAAC,MAAM,CAAC;gBACb,OAAO,CAAC,GAAG,CACT,wBAAwB,IAAI,CAAC,EAAE,UAAU,IAAI,CAAC,KAAK,SAAS,IAAI,CAAC,IAAI,cAAc,IAAI,CAAC,SAAS,eAAe,SAAS,GAAG,CAC7H,CAAC;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;YAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAyB,EACzB,EAAE;gBACF,MAAM,OAAO,GACX,IAAI,CAAC,KAAK,CAAC,UAAU;oBACnB,EAAE,GAAG,CACH,CAAC,EAAE,EAAE,EAAE,CACL,GAAG,EAAE,CAAC,IAAI,IAAI,GAAG,SAAS,CAAC,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CACnE;qBACA,IAAI,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC;gBAC1B,OAAO,CAAC,GAAG,CACT,8BAA8B,IAAI,CAAC,EAAE,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,eAAe,OAAO,GAAG,CACvF,CAAC;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;YAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAyB,EACzB,EAAE;gBACF,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;oBAC/C,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO;yBACf,GAAG,CACF,CAAC,CAAC,EAAE,EAAE,CACJ,GAAG,CAAC,CAAC,IAAI,KAAK,MAAM,CAAE,CAA6B,CAAC,IAAI,IAAK,CAA6B,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAC9H;yBACA,IAAI,CAAC,IAAI,CAAC;oBACf,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAChD,OAAO,CAAC,GAAG,CACT,6BAA6B,IAAI,CAAC,EAAE,aAAa,OAAO,GAAG,CAC5D,CAAC;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;QACD,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE;YAChC,MAAM,EAAE,CACN,KAAqC,EACrC,IAA2B,EAC3B,EAAE;gBACF,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;oBAC/C,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO;yBACf,GAAG,CACF,CAAC,CAAC,EAAE,EAAE,CACJ,GAAG,CAAC,CAAC,IAAI,KAAK,MAAM,CAAE,CAA6B,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CACvF;yBACA,IAAI,CAAC,IAAI,CAAC;oBACf,CAAC,CAAC,GAAG,CAAC;gBACR,OAAO,CAAC,GAAG,CACT,+BAA+B,IAAI,CAAC,EAAE,aAAa,OAAO,GAAG,CAC9D,CAAC;gBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;SACF;KACF,CAAC;IAEF,MAAM,aAAa,GAAG,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAEtD,MAAM,SAAS,GAAG;QAChB,GAAG,aAAa;QAChB,KAAK,EAAE,qCAAqC;QAC5C,SAAS,EAAE,KAAK;QAChB,4BAA4B,EAAE;YAC5B,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,aAAa,EAAE,KAAK,EAAE;SACpD;KACF,CAAC;IAEF,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;QACrC,KAAK,EAAE,kCAAkC;QACzC,WAAW,EAAE;YACX,YAAY;YACZ,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,CAAC,uBAAuB,EAAE,CAAC;YAClC,SAAS;SACV;QACD,aAAa,EAAE,IAAI;QACnB,WAAW,EAAE,IAAI;QACjB,cAAc,EAAE,cAA+C;KAChE,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG;QACnB,YAAY,EAAE;YACZ,SAAS,EAAE,oCAAoC;SAChD;QACD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,MAAM,WAAW,GAAG,iIAAiI,CAAC;IACtJ,mBAAmB,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC;IAExD,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,YAAY,WAAW,KAAK,CAAC,CAAC;IAE1C,MAAM,MAAM,GAAG,EAAE,QAAQ,EAAE,CAAC,GAAG,mBAAmB,CAAC,EAAE,CAAC;IACtD,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAE9C,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;IAElE,IAAI,gBAAgB,GAAG,KAAK,CAAC;IAC7B,IAAI,sBAAsB,GAAG,KAAK,CAAC;IAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,IAAI,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAClC,SAAS;QACX,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;QAC3B,IAAI,QAAQ,KAAK,YAAY,CAAC,SAAS,EAAE,CAAC;YACxC,MAAM,EAAE,GAAI,IAA0B,CAAC,SAAS,CAAC;YACjD,IAAI,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;gBACpB,gBAAgB,GAAG,IAAI,CAAC;gBACxB,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,+CAA+C,CAAC,CAAC;YACtE,CAAC;iBAAM,CAAC;gBACN,MAAM,aAAa,GAAG,EAAE,CAAC,MAAM;oBAC7B,CAAC,CAAC,UAAW,EAAE,CAAC,MAAiB,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK;oBACvD,CAAC,CAAC,WAAW,CAAC;gBAChB,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,IAAI,IAAI,aAAa,EAAE,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;aAAM,IAAI,QAAQ,KAAK,YAAY,CAAC,KAAK,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAI,IAA+B,CAAC,KAAK,IAAI,EAAE,CAAC;YAC3D,OAAO,CAAC,GAAG,CACT,MAAM,CAAC,YAAY,KAAK,CAAC,MAAM,aAAa,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CACzE,CAAC;QACJ,CAAC;aAAM,IAAI,QAAQ,KAAK,YAAY,CAAC,IAAI,EAAE,CAAC;YAC1C,MAAM,IAAI,GAAI,IAA6B,CAAC,IAAI,IAAI,EAAE,CAAC;YACvD,OAAO,CAAC,GAAG,CACT,MAAM,CAAC,WAAW,IAAI,CAAC,MAAM,aAAa,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CACtE,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,QAAQ,EAAE,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,IAAI,eAAe,GAAkB,IAAI,CAAC;IAC1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,IAAI,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,IAAI;YAAE,SAAS;QAEpB,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,SAAS,EAAE,CAAC;YACzC,eAAe,GAAG,IAAI,CAAC;YACvB,SAAS;QACX,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI,EAAE,CAAC;YACpC,eAAe,GAAG,CAAC,CAAC;QACtB,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,KAAK,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;YACxE,MAAM,QAAQ,GAAG,aAAa,CAAC,eAAe,CAAyB,CAAC;YACxE,MAAM,YAAY,GAAI,IAA+B,CAAC,KAAK,IAAI,EAAE,CAAC;YAClE,IACE,QAAQ,EAAE,IAAI;gBACd,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;gBAC/B,YAAY,CAAC,MAAM,GAAG,CAAC,EACvB,CAAC;gBACD,sBAAsB,GAAG,IAAI,CAAC;gBAC9B,OAAO,CAAC,GAAG,CACT,sDAAsD,eAAe,8BAA8B,CAAC,GAAG,CACxG,CAAC;gBACF,OAAO,CAAC,GAAG,CACT,6BAA6B,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,GAAG,CACnF,CAAC;gBACF,OAAO,CAAC,GAAG,CACT,6BAA6B,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CACjE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;IACjD,OAAO,CAAC,GAAG,CAAC,wBAAwB,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5E,OAAO,CAAC,GAAG,CACT,0BAA0B,gBAAgB,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,EAAE,CAClE,CAAC;IACF,OAAO,CAAC,GAAG,CACT,2BAA2B,sBAAsB,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,EAAE,CACzE,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IACzC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC9C,CAAC;AAED,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnD,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;IACrE,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,GAAG,EAAE,EAAE;IACtC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC;AAC5C,CAAC,CAAC,CAAC;AAEH,6BAA6B,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IAC5C,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACnB,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC9B,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}