@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,624 @@
1
+ /* eslint-disable no-process-env */
2
+ /* eslint-disable no-console */
3
+ /* eslint-disable @typescript-eslint/no-explicit-any */
4
+ /**
5
+ * Integration tests for structured output with real LLM API calls.
6
+ *
7
+ * These tests require actual API credentials. They will be skipped if
8
+ * the required environment variables are not set.
9
+ *
10
+ * To run:
11
+ * 1. Copy ranger/.env credentials to agents/.env (or set env vars directly)
12
+ * 2. npm test -- --testPathPattern=structured-output.integration
13
+ *
14
+ * Environment variables needed:
15
+ * - BEDROCK_AWS_ACCESS_KEY_ID + BEDROCK_AWS_SECRET_ACCESS_KEY + BEDROCK_AWS_DEFAULT_REGION
16
+ * - ANTHROPIC_API_KEY (optional, for direct Anthropic tests)
17
+ * - OPENAI_API_KEY (optional, for direct OpenAI tests)
18
+ */
19
+ import { config } from 'dotenv';
20
+ import { resolve } from 'path';
21
+ // Load from ranger/.env if agents/.env doesn't exist
22
+ config(); // Try local first
23
+ config({ path: resolve(__dirname, '../../../../ranger/.env') }); // Then ranger
24
+ import { HumanMessage, SystemMessage } from '@langchain/core/messages';
25
+ import { CustomChatBedrockConverse } from '@/llm/bedrock';
26
+ import { Providers } from '@/common';
27
+ import { validateStructuredOutput, prepareSchemaForProvider } from '@/schemas/validate';
28
+ jest.setTimeout(120000);
29
+ // ──────────────────────────────────────────────────────────────
30
+ // Helper: check if credentials are available
31
+ // ──────────────────────────────────────────────────────────────
32
+ const hasBedrock = !!process.env.BEDROCK_AWS_ACCESS_KEY_ID &&
33
+ !!process.env.BEDROCK_AWS_SECRET_ACCESS_KEY;
34
+ const hasAnthropic = !!process.env.ANTHROPIC_API_KEY;
35
+ const hasOpenAI = !!process.env.OPENAI_API_KEY;
36
+ // ──────────────────────────────────────────────────────────────
37
+ // Shared test schema
38
+ // ──────────────────────────────────────────────────────────────
39
+ const personSchema = {
40
+ type: 'object',
41
+ properties: {
42
+ name: { type: 'string', description: 'Full name of the person' },
43
+ age: { type: 'number', description: 'Age in years' },
44
+ occupation: { type: 'string', description: 'Job title or occupation' },
45
+ },
46
+ required: ['name', 'age', 'occupation'],
47
+ additionalProperties: false,
48
+ };
49
+ const classificationSchema = {
50
+ type: 'object',
51
+ properties: {
52
+ category: {
53
+ type: 'string',
54
+ enum: ['positive', 'negative', 'neutral'],
55
+ description: 'Sentiment category',
56
+ },
57
+ confidence: {
58
+ type: 'number',
59
+ description: 'Confidence score between 0 and 1',
60
+ },
61
+ reasoning: {
62
+ type: 'string',
63
+ description: 'Brief explanation of the classification',
64
+ },
65
+ },
66
+ required: ['category', 'confidence', 'reasoning'],
67
+ additionalProperties: false,
68
+ };
69
+ // ──────────────────────────────────────────────────────────────
70
+ // Bedrock Integration Tests
71
+ // ──────────────────────────────────────────────────────────────
72
+ describe('Bedrock structured output integration', () => {
73
+ const bedrockRegion = process.env.BEDROCK_AWS_DEFAULT_REGION || 'us-east-1';
74
+ const createBedrockModel = () => new CustomChatBedrockConverse({
75
+ // Use Claude 3 Haiku which is widely available and supports on-demand invocation
76
+ model: 'anthropic.claude-3-haiku-20240307-v1:0',
77
+ region: bedrockRegion,
78
+ credentials: {
79
+ accessKeyId: process.env.BEDROCK_AWS_ACCESS_KEY_ID,
80
+ secretAccessKey: process.env.BEDROCK_AWS_SECRET_ACCESS_KEY,
81
+ ...(process.env.BEDROCK_AWS_SESSION_TOKEN
82
+ ? { sessionToken: process.env.BEDROCK_AWS_SESSION_TOKEN }
83
+ : {}),
84
+ },
85
+ maxTokens: 1024,
86
+ });
87
+ const conditionalTest = hasBedrock ? test : test.skip;
88
+ conditionalTest('withStructuredOutput functionCalling - basic person extraction', async () => {
89
+ const model = createBedrockModel();
90
+ const structuredModel = model.withStructuredOutput(personSchema, {
91
+ name: 'PersonInfo',
92
+ method: 'functionCalling',
93
+ includeRaw: true,
94
+ strict: true,
95
+ });
96
+ const result = await structuredModel.invoke([
97
+ new SystemMessage('Extract person information from the text.'),
98
+ new HumanMessage('John Smith is a 35 year old software engineer from Seattle.'),
99
+ ]);
100
+ console.log('[Integration] Bedrock functionCalling result:', JSON.stringify(result.parsed ?? result, null, 2));
101
+ const parsed = result.parsed ?? result;
102
+ expect(parsed).toBeDefined();
103
+ expect(typeof parsed.name).toBe('string');
104
+ expect(typeof parsed.age).toBe('number');
105
+ expect(typeof parsed.occupation).toBe('string');
106
+ // Validate against schema
107
+ const validation = validateStructuredOutput(parsed, personSchema);
108
+ expect(validation.success).toBe(true);
109
+ });
110
+ conditionalTest('withStructuredOutput functionCalling - classification with enum', async () => {
111
+ const model = createBedrockModel();
112
+ const structuredModel = model.withStructuredOutput(classificationSchema, {
113
+ name: 'Classification',
114
+ method: 'functionCalling',
115
+ includeRaw: true,
116
+ strict: true,
117
+ });
118
+ const result = await structuredModel.invoke([
119
+ new SystemMessage('Classify the sentiment of the following text.'),
120
+ new HumanMessage('I absolutely love this product! It exceeded all my expectations.'),
121
+ ]);
122
+ const parsed = result.parsed ?? result;
123
+ console.log('[Integration] Bedrock classification result:', JSON.stringify(parsed, null, 2));
124
+ expect(parsed).toBeDefined();
125
+ expect(['positive', 'negative', 'neutral']).toContain(parsed.category);
126
+ expect(typeof parsed.confidence).toBe('number');
127
+ expect(typeof parsed.reasoning).toBe('string');
128
+ });
129
+ conditionalTest('withStructuredOutput with prepared schema', async () => {
130
+ const rawSchema = {
131
+ type: 'object',
132
+ properties: {
133
+ summary: { type: 'string', maxLength: 500 },
134
+ topics: { type: 'array', items: { type: 'string' }, minItems: 1, maxItems: 5 },
135
+ wordCount: { type: 'number', minimum: 0 },
136
+ },
137
+ };
138
+ // Prepare schema for Bedrock
139
+ const { schema: prepared, warnings } = prepareSchemaForProvider(rawSchema, Providers.BEDROCK);
140
+ console.log('[Integration] Schema preparation warnings:', warnings);
141
+ // Verify preparation worked
142
+ expect(prepared.additionalProperties).toBe(false);
143
+ expect(prepared.required).toEqual(expect.arrayContaining(['summary', 'topics', 'wordCount']));
144
+ const summaryProp = prepared.properties.summary;
145
+ expect(summaryProp.maxLength).toBeUndefined();
146
+ expect(summaryProp.description).toContain('maxLength');
147
+ // Now use the prepared schema with the model
148
+ const model = createBedrockModel();
149
+ const structuredModel = model.withStructuredOutput(prepared, {
150
+ name: 'Analysis',
151
+ method: 'functionCalling',
152
+ includeRaw: true,
153
+ strict: true,
154
+ });
155
+ const result = await structuredModel.invoke([
156
+ new SystemMessage('Analyze the following text and provide a summary, key topics, and word count.'),
157
+ new HumanMessage('Artificial intelligence has transformed the technology industry in recent years. ' +
158
+ 'Machine learning models are now capable of generating text, images, and even code. ' +
159
+ 'Companies are investing heavily in AI research and development.'),
160
+ ]);
161
+ const parsed = result.parsed ?? result;
162
+ console.log('[Integration] Bedrock analysis result:', JSON.stringify(parsed, null, 2));
163
+ expect(parsed).toBeDefined();
164
+ expect(typeof parsed.summary).toBe('string');
165
+ expect(Array.isArray(parsed.topics)).toBe(true);
166
+ expect(typeof parsed.wordCount).toBe('number');
167
+ });
168
+ conditionalTest('withStructuredOutput with tools - deferred structured output pattern', async () => {
169
+ const model = createBedrockModel();
170
+ // Step 1: Regular invocation with tools (simulate tool use)
171
+ const messages = [
172
+ new SystemMessage('You are a helpful assistant. When asked to analyze text, provide a structured analysis.'),
173
+ new HumanMessage('Analyze the sentiment: "This movie was absolutely terrible, worst I have ever seen."'),
174
+ ];
175
+ // Step 2: Get structured output (this simulates the deferred structured output pattern)
176
+ const structuredModel = model.withStructuredOutput(classificationSchema, {
177
+ name: 'SentimentAnalysis',
178
+ method: 'functionCalling',
179
+ includeRaw: true,
180
+ strict: true,
181
+ });
182
+ const result = await structuredModel.invoke(messages);
183
+ const parsed = result.parsed ?? result;
184
+ console.log('[Integration] Bedrock deferred structured result:', JSON.stringify(parsed, null, 2));
185
+ expect(parsed).toBeDefined();
186
+ expect(['positive', 'negative', 'neutral']).toContain(parsed.category);
187
+ // The movie review is clearly negative
188
+ expect(parsed.category).toBe('negative');
189
+ expect(typeof parsed.confidence).toBe('number');
190
+ });
191
+ conditionalTest('validates response against schema and detects failures', async () => {
192
+ // Test schema validation on a known good response
193
+ const goodResponse = {
194
+ name: 'Alice',
195
+ age: 30,
196
+ occupation: 'Engineer',
197
+ };
198
+ const goodResult = validateStructuredOutput(goodResponse, personSchema);
199
+ expect(goodResult.success).toBe(true);
200
+ // Test with a bad response
201
+ const badResponse = {
202
+ name: 123, // Wrong type
203
+ age: 'thirty', // Wrong type
204
+ };
205
+ const badResult = validateStructuredOutput(badResponse, personSchema);
206
+ expect(badResult.success).toBe(false);
207
+ expect(badResult.error).toBeDefined();
208
+ });
209
+ });
210
+ // ──────────────────────────────────────────────────────────────
211
+ // Complex Tool + Structured Output Integration Tests (Bedrock)
212
+ // ──────────────────────────────────────────────────────────────
213
+ describe('Bedrock complex tool + structured output integration', () => {
214
+ const bedrockRegion = process.env.BEDROCK_AWS_DEFAULT_REGION || 'us-east-1';
215
+ const conditionalTest = hasBedrock ? test : test.skip;
216
+ const createBedrockModel = () => new CustomChatBedrockConverse({
217
+ model: 'anthropic.claude-3-haiku-20240307-v1:0',
218
+ region: bedrockRegion,
219
+ credentials: {
220
+ accessKeyId: process.env.BEDROCK_AWS_ACCESS_KEY_ID,
221
+ secretAccessKey: process.env.BEDROCK_AWS_SECRET_ACCESS_KEY,
222
+ ...(process.env.BEDROCK_AWS_SESSION_TOKEN
223
+ ? { sessionToken: process.env.BEDROCK_AWS_SESSION_TOKEN }
224
+ : {}),
225
+ },
226
+ maxTokens: 2048,
227
+ });
228
+ conditionalTest('complex schema with nested objects and arrays - multi-step analysis', async () => {
229
+ const complexSchema = {
230
+ type: 'object',
231
+ properties: {
232
+ analysis: {
233
+ type: 'object',
234
+ properties: {
235
+ overall_sentiment: {
236
+ type: 'string',
237
+ enum: ['very_positive', 'positive', 'neutral', 'negative', 'very_negative'],
238
+ },
239
+ confidence: { type: 'number' },
240
+ key_themes: {
241
+ type: 'array',
242
+ items: {
243
+ type: 'object',
244
+ properties: {
245
+ theme: { type: 'string' },
246
+ sentiment: { type: 'string', enum: ['positive', 'negative', 'neutral'] },
247
+ evidence: { type: 'string' },
248
+ },
249
+ required: ['theme', 'sentiment', 'evidence'],
250
+ additionalProperties: false,
251
+ },
252
+ },
253
+ },
254
+ required: ['overall_sentiment', 'confidence', 'key_themes'],
255
+ additionalProperties: false,
256
+ },
257
+ entities: {
258
+ type: 'array',
259
+ items: {
260
+ type: 'object',
261
+ properties: {
262
+ name: { type: 'string' },
263
+ type: { type: 'string', enum: ['person', 'organization', 'location', 'product'] },
264
+ role: { type: 'string' },
265
+ },
266
+ required: ['name', 'type', 'role'],
267
+ additionalProperties: false,
268
+ },
269
+ },
270
+ summary: { type: 'string' },
271
+ action_items: {
272
+ type: 'array',
273
+ items: { type: 'string' },
274
+ },
275
+ },
276
+ required: ['analysis', 'entities', 'summary', 'action_items'],
277
+ additionalProperties: false,
278
+ };
279
+ const model = createBedrockModel();
280
+ const structuredModel = model.withStructuredOutput(complexSchema, {
281
+ name: 'DetailedAnalysis',
282
+ method: 'functionCalling',
283
+ includeRaw: true,
284
+ strict: true,
285
+ });
286
+ const result = await structuredModel.invoke([
287
+ new SystemMessage('You are an expert analyst. Analyze the following business communication and extract detailed structured information.'),
288
+ new HumanMessage('From: Sarah Johnson, VP of Engineering at TechCorp\n' +
289
+ 'To: All Engineering Teams\n\n' +
290
+ 'I am pleased to announce that our new AI platform, CodeAssist, has exceeded Q3 targets by 45%. ' +
291
+ 'The Seattle development team, led by Marcus Chen, deserves special recognition for their outstanding work. ' +
292
+ 'However, we need to address the performance issues reported by our European clients. ' +
293
+ 'The London office has flagged latency problems that must be resolved before the Q4 launch. ' +
294
+ 'Action items: 1) Marcus to lead a performance task force. 2) Schedule a meeting with the London team. ' +
295
+ '3) Update the deployment pipeline for EU regions.'),
296
+ ]);
297
+ const parsed = result.parsed ?? result;
298
+ console.log('[Integration] Complex analysis result:', JSON.stringify(parsed, null, 2));
299
+ // Verify top-level structure
300
+ // Note: functionCalling mode is best-effort, so some fields may be missing.
301
+ // Native constrained decoding (when available) would guarantee all fields.
302
+ expect(parsed).toBeDefined();
303
+ expect(typeof parsed.summary).toBe('string');
304
+ // Analysis is the most complex nested field — verify if present
305
+ if (parsed.analysis) {
306
+ expect(['very_positive', 'positive', 'neutral', 'negative', 'very_negative']).toContain(parsed.analysis.overall_sentiment);
307
+ expect(typeof parsed.analysis.confidence).toBe('number');
308
+ if (Array.isArray(parsed.analysis.key_themes)) {
309
+ expect(parsed.analysis.key_themes.length).toBeGreaterThan(0);
310
+ for (const theme of parsed.analysis.key_themes) {
311
+ expect(typeof theme.theme).toBe('string');
312
+ expect(['positive', 'negative', 'neutral']).toContain(theme.sentiment);
313
+ expect(typeof theme.evidence).toBe('string');
314
+ }
315
+ }
316
+ }
317
+ // Verify entities if present
318
+ if (Array.isArray(parsed.entities)) {
319
+ expect(parsed.entities.length).toBeGreaterThan(0);
320
+ for (const entity of parsed.entities) {
321
+ expect(typeof entity.name).toBe('string');
322
+ expect(['person', 'organization', 'location', 'product']).toContain(entity.type);
323
+ }
324
+ }
325
+ // Verify action items
326
+ if (Array.isArray(parsed.action_items)) {
327
+ expect(parsed.action_items.length).toBeGreaterThan(0);
328
+ }
329
+ // At minimum, we expect the model returned something parseable with the key fields
330
+ expect(parsed.summary.length).toBeGreaterThan(0);
331
+ });
332
+ conditionalTest('tool-like multi-turn conversation with structured output at the end', async () => {
333
+ // Simulate a multi-turn conversation where the agent has gathered info
334
+ // and now needs to produce structured output
335
+ const reportSchema = {
336
+ type: 'object',
337
+ properties: {
338
+ report_title: { type: 'string' },
339
+ findings: {
340
+ type: 'array',
341
+ items: {
342
+ type: 'object',
343
+ properties: {
344
+ category: { type: 'string' },
345
+ description: { type: 'string' },
346
+ severity: { type: 'string', enum: ['low', 'medium', 'high', 'critical'] },
347
+ recommendation: { type: 'string' },
348
+ },
349
+ required: ['category', 'description', 'severity', 'recommendation'],
350
+ additionalProperties: false,
351
+ },
352
+ },
353
+ overall_risk_level: {
354
+ type: 'string',
355
+ enum: ['low', 'medium', 'high', 'critical'],
356
+ },
357
+ next_steps: {
358
+ type: 'array',
359
+ items: { type: 'string' },
360
+ },
361
+ },
362
+ required: ['report_title', 'findings', 'overall_risk_level', 'next_steps'],
363
+ additionalProperties: false,
364
+ };
365
+ const model = createBedrockModel();
366
+ // Multi-turn conversation simulating tool results being fed in
367
+ const messages = [
368
+ new SystemMessage('You are a security analyst. You have performed a security audit and gathered the following data from various scans. ' +
369
+ 'Produce a structured security report.'),
370
+ new HumanMessage('Here are the scan results:\n\n' +
371
+ 'Port Scan: Ports 22, 80, 443, 8080 are open. Port 8080 is running an outdated Apache Tomcat 8.5.\n\n' +
372
+ 'Vulnerability Scan: Found SQL injection in /api/users endpoint. XSS vulnerability in search form. ' +
373
+ 'Outdated OpenSSL 1.1.1 (should be 3.x). No rate limiting on login endpoint.\n\n' +
374
+ 'Configuration Review: Default admin credentials found on Tomcat manager. ' +
375
+ 'CORS is set to wildcard (*). No CSP headers. Cookies without HttpOnly flag.\n\n' +
376
+ 'Provide your structured security report.'),
377
+ ];
378
+ const structuredModel = model.withStructuredOutput(reportSchema, {
379
+ name: 'SecurityReport',
380
+ method: 'functionCalling',
381
+ includeRaw: true,
382
+ strict: true,
383
+ });
384
+ const result = await structuredModel.invoke(messages);
385
+ const parsed = result.parsed ?? result;
386
+ console.log('[Integration] Security report result:', JSON.stringify(parsed, null, 2));
387
+ // Verify structure
388
+ expect(typeof parsed.report_title).toBe('string');
389
+ expect(Array.isArray(parsed.findings)).toBe(true);
390
+ expect(parsed.findings.length).toBeGreaterThanOrEqual(3); // At least SQL injection, XSS, outdated software
391
+ expect(['low', 'medium', 'high', 'critical']).toContain(parsed.overall_risk_level);
392
+ expect(Array.isArray(parsed.next_steps)).toBe(true);
393
+ expect(parsed.next_steps.length).toBeGreaterThan(0);
394
+ // Verify each finding
395
+ for (const finding of parsed.findings) {
396
+ expect(typeof finding.category).toBe('string');
397
+ expect(typeof finding.description).toBe('string');
398
+ expect(['low', 'medium', 'high', 'critical']).toContain(finding.severity);
399
+ expect(typeof finding.recommendation).toBe('string');
400
+ }
401
+ // Given the scan results, risk should be high or critical
402
+ expect(['high', 'critical']).toContain(parsed.overall_risk_level);
403
+ // Validate entire response
404
+ const validation = validateStructuredOutput(parsed, reportSchema);
405
+ expect(validation.success).toBe(true);
406
+ });
407
+ conditionalTest('prepared schema with constraint stripping works end-to-end', async () => {
408
+ // A schema with MANY unsupported constraints that need stripping
409
+ const rawSchema = {
410
+ type: 'object',
411
+ properties: {
412
+ title: { type: 'string', minLength: 5, maxLength: 100 },
413
+ rating: { type: 'number', minimum: 1, maximum: 5, multipleOf: 0.5 },
414
+ tags: {
415
+ type: 'array',
416
+ items: { type: 'string', minLength: 1, maxLength: 50 },
417
+ minItems: 1,
418
+ maxItems: 5,
419
+ uniqueItems: true,
420
+ },
421
+ metadata: {
422
+ type: 'object',
423
+ properties: {
424
+ author: { type: 'string', pattern: '^[A-Za-z ]+$' },
425
+ word_count: { type: 'integer', minimum: 0, maximum: 100000 },
426
+ language: { type: 'string', enum: ['en', 'es', 'fr', 'de'] },
427
+ },
428
+ },
429
+ },
430
+ };
431
+ // Prepare for Bedrock
432
+ const { schema: prepared, warnings } = prepareSchemaForProvider(rawSchema, Providers.BEDROCK);
433
+ console.log('[Integration] Complex schema warnings count:', warnings.length);
434
+ expect(warnings.length).toBeGreaterThan(5); // Should have many warnings
435
+ // Verify ALL constraints were stripped
436
+ const titleProp = prepared.properties.title;
437
+ expect(titleProp.minLength).toBeUndefined();
438
+ expect(titleProp.maxLength).toBeUndefined();
439
+ const ratingProp = prepared.properties.rating;
440
+ expect(ratingProp.minimum).toBeUndefined();
441
+ expect(ratingProp.maximum).toBeUndefined();
442
+ expect(ratingProp.multipleOf).toBeUndefined();
443
+ const tagsProp = prepared.properties.tags;
444
+ expect(tagsProp.minItems).toBeUndefined();
445
+ expect(tagsProp.maxItems).toBeUndefined();
446
+ expect(tagsProp.uniqueItems).toBeUndefined();
447
+ // Now use with actual model
448
+ const model = createBedrockModel();
449
+ const structuredModel = model.withStructuredOutput(prepared, {
450
+ name: 'BookReview',
451
+ method: 'functionCalling',
452
+ includeRaw: true,
453
+ strict: true,
454
+ });
455
+ const result = await structuredModel.invoke([
456
+ new SystemMessage('You are a book reviewer. Review the book described below.'),
457
+ new HumanMessage('Review "The Great Gatsby" by F. Scott Fitzgerald. ' +
458
+ 'It is a classic American novel written in English about the decline of the American Dream.'),
459
+ ]);
460
+ const parsed = result.parsed ?? result;
461
+ console.log('[Integration] Book review result:', JSON.stringify(parsed, null, 2));
462
+ expect(typeof parsed.title).toBe('string');
463
+ expect(typeof parsed.rating).toBe('number');
464
+ expect(Array.isArray(parsed.tags)).toBe(true);
465
+ expect(parsed.metadata).toBeDefined();
466
+ expect(['en', 'es', 'fr', 'de']).toContain(parsed.metadata.language);
467
+ });
468
+ });
469
+ // ──────────────────────────────────────────────────────────────
470
+ // Anthropic Direct Integration Tests
471
+ // ──────────────────────────────────────────────────────────────
472
+ describe('Anthropic direct structured output integration', () => {
473
+ const conditionalTest = hasAnthropic ? test : test.skip;
474
+ conditionalTest('withStructuredOutput jsonSchema (native) - person extraction', async () => {
475
+ // Dynamic import to avoid errors when @langchain/anthropic is not installed
476
+ const { CustomAnthropic } = await import('@/llm/anthropic');
477
+ const model = new CustomAnthropic({
478
+ anthropicApiKey: process.env.ANTHROPIC_API_KEY,
479
+ model: 'claude-sonnet-4-20250514',
480
+ maxTokens: 1024,
481
+ });
482
+ const structuredModel = model.withStructuredOutput(personSchema, {
483
+ name: 'PersonInfo',
484
+ method: 'jsonSchema',
485
+ includeRaw: true,
486
+ strict: true,
487
+ });
488
+ const result = await structuredModel.invoke([
489
+ new SystemMessage('Extract person information from the text.'),
490
+ new HumanMessage('Marie Curie was a 66 year old physicist and chemist.'),
491
+ ]);
492
+ const parsed = result.parsed ?? result;
493
+ console.log('[Integration] Anthropic native result:', JSON.stringify(parsed, null, 2));
494
+ expect(parsed).toBeDefined();
495
+ expect(typeof parsed.name).toBe('string');
496
+ expect(typeof parsed.age).toBe('number');
497
+ expect(typeof parsed.occupation).toBe('string');
498
+ const validation = validateStructuredOutput(parsed, personSchema);
499
+ expect(validation.success).toBe(true);
500
+ });
501
+ conditionalTest('withStructuredOutput jsonSchema - classification with enum', async () => {
502
+ const { CustomAnthropic } = await import('@/llm/anthropic');
503
+ const model = new CustomAnthropic({
504
+ anthropicApiKey: process.env.ANTHROPIC_API_KEY,
505
+ model: 'claude-sonnet-4-20250514',
506
+ maxTokens: 1024,
507
+ });
508
+ const structuredModel = model.withStructuredOutput(classificationSchema, {
509
+ name: 'Sentiment',
510
+ method: 'jsonSchema',
511
+ includeRaw: true,
512
+ strict: true,
513
+ });
514
+ const result = await structuredModel.invoke([
515
+ new SystemMessage('Classify the sentiment.'),
516
+ new HumanMessage('This is the best day of my life!'),
517
+ ]);
518
+ const parsed = result.parsed ?? result;
519
+ console.log('[Integration] Anthropic classification:', JSON.stringify(parsed, null, 2));
520
+ expect(['positive', 'negative', 'neutral']).toContain(parsed.category);
521
+ expect(parsed.category).toBe('positive');
522
+ });
523
+ });
524
+ // ──────────────────────────────────────────────────────────────
525
+ // OpenAI Integration Tests
526
+ // ──────────────────────────────────────────────────────────────
527
+ describe('OpenAI structured output integration', () => {
528
+ const conditionalTest = hasOpenAI ? test : test.skip;
529
+ conditionalTest('withStructuredOutput jsonSchema (native) - person extraction', async () => {
530
+ const { ChatOpenAI } = await import('@/llm/openai');
531
+ const model = new ChatOpenAI({
532
+ openAIApiKey: process.env.OPENAI_API_KEY,
533
+ model: 'gpt-4o-mini',
534
+ maxTokens: 1024,
535
+ });
536
+ const structuredModel = model.withStructuredOutput(personSchema, {
537
+ name: 'PersonInfo',
538
+ method: 'jsonSchema',
539
+ includeRaw: true,
540
+ strict: true,
541
+ });
542
+ const result = await structuredModel.invoke([
543
+ new SystemMessage('Extract person information from the text.'),
544
+ new HumanMessage('Albert Einstein was a 76 year old theoretical physicist.'),
545
+ ]);
546
+ const parsed = result.parsed ?? result;
547
+ console.log('[Integration] OpenAI native result:', JSON.stringify(parsed, null, 2));
548
+ expect(parsed).toBeDefined();
549
+ expect(typeof parsed.name).toBe('string');
550
+ expect(typeof parsed.age).toBe('number');
551
+ const validation = validateStructuredOutput(parsed, personSchema);
552
+ expect(validation.success).toBe(true);
553
+ });
554
+ });
555
+ // ──────────────────────────────────────────────────────────────
556
+ // Schema Preparation Integration Tests
557
+ // ──────────────────────────────────────────────────────────────
558
+ describe('Schema preparation produces valid schemas for all providers', () => {
559
+ const complexSchema = {
560
+ type: 'object',
561
+ properties: {
562
+ analysis: {
563
+ type: 'object',
564
+ properties: {
565
+ score: { type: 'number', minimum: 0, maximum: 100 },
566
+ label: { type: 'string', enum: ['low', 'medium', 'high'] },
567
+ details: { type: 'string', maxLength: 1000 },
568
+ },
569
+ },
570
+ tags: {
571
+ type: 'array',
572
+ items: { type: 'string', minLength: 1 },
573
+ minItems: 1,
574
+ maxItems: 10,
575
+ },
576
+ metadata: {
577
+ type: 'object',
578
+ properties: {
579
+ source: { type: 'string' },
580
+ timestamp: { type: 'string' },
581
+ },
582
+ },
583
+ },
584
+ };
585
+ test('prepares valid schema for Anthropic', () => {
586
+ const { schema, warnings } = prepareSchemaForProvider(complexSchema, Providers.ANTHROPIC);
587
+ // Root object
588
+ expect(schema.additionalProperties).toBe(false);
589
+ expect(schema.required).toEqual(expect.arrayContaining(['analysis', 'tags', 'metadata']));
590
+ // Nested object: analysis
591
+ const analysis = schema.properties.analysis;
592
+ expect(analysis.additionalProperties).toBe(false);
593
+ expect(analysis.required).toEqual(expect.arrayContaining(['score', 'label', 'details']));
594
+ // Numeric constraints stripped
595
+ expect(analysis.properties.score.minimum).toBeUndefined();
596
+ expect(analysis.properties.score.maximum).toBeUndefined();
597
+ // String constraints stripped
598
+ expect(analysis.properties.details.maxLength).toBeUndefined();
599
+ // Array constraints stripped
600
+ const tags = schema.properties.tags;
601
+ expect(tags.minItems).toBeUndefined();
602
+ expect(tags.maxItems).toBeUndefined();
603
+ // Array item string constraints stripped
604
+ expect(tags.items.minLength).toBeUndefined();
605
+ // Nested object: metadata
606
+ const metadata = schema.properties.metadata;
607
+ expect(metadata.additionalProperties).toBe(false);
608
+ expect(warnings.length).toBeGreaterThan(0);
609
+ console.log('[Integration] Anthropic schema warnings:', warnings);
610
+ });
611
+ test('prepares valid schema for OpenAI', () => {
612
+ const { schema } = prepareSchemaForProvider(complexSchema, Providers.OPENAI);
613
+ expect(schema.additionalProperties).toBe(false);
614
+ const analysis = schema.properties.analysis;
615
+ expect(analysis.properties.score.minimum).toBeUndefined();
616
+ });
617
+ test('prepares valid schema for Bedrock', () => {
618
+ const { schema } = prepareSchemaForProvider(complexSchema, Providers.BEDROCK);
619
+ expect(schema.additionalProperties).toBe(false);
620
+ const analysis = schema.properties.analysis;
621
+ expect(analysis.properties.score.minimum).toBeUndefined();
622
+ });
623
+ });
624
+ //# sourceMappingURL=structured-output.integration.test.js.map