@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,807 @@
1
+ // src/tools/__tests__/ProgrammaticToolCalling.test.ts
2
+ /**
3
+ * Unit tests for Programmatic Tool Calling.
4
+ * Tests manual invocation with mock tools and Code API responses.
5
+ */
6
+ import { describe, it, expect, beforeEach } from '@jest/globals';
7
+ import { createProgrammaticToolCallingTool, formatCompletedResponse, extractUsedToolNames, filterToolsByUsage, executeTools, normalizeToPythonIdentifier, unwrapToolResponse, } from '../ProgrammaticToolCalling';
8
+ import { createProgrammaticToolRegistry, createGetTeamMembersTool, createGetExpensesTool, createGetWeatherTool, createCalculatorTool, } from '@/test/mockTools';
9
+ describe('ProgrammaticToolCalling', () => {
10
+ describe('executeTools', () => {
11
+ let toolMap;
12
+ beforeEach(() => {
13
+ const tools = [
14
+ createGetTeamMembersTool(),
15
+ createGetExpensesTool(),
16
+ createGetWeatherTool(),
17
+ createCalculatorTool(),
18
+ ];
19
+ toolMap = new Map(tools.map((t) => [t.name, t]));
20
+ });
21
+ it('executes a single tool successfully', async () => {
22
+ const toolCalls = [
23
+ {
24
+ id: 'call_001',
25
+ name: 'get_weather',
26
+ input: { city: 'San Francisco' },
27
+ },
28
+ ];
29
+ const results = await executeTools(toolCalls, toolMap);
30
+ expect(results).toHaveLength(1);
31
+ expect(results[0].call_id).toBe('call_001');
32
+ expect(results[0].is_error).toBe(false);
33
+ expect(results[0].result).toEqual({
34
+ temperature: 65,
35
+ condition: 'Foggy',
36
+ });
37
+ });
38
+ it('executes multiple tools in parallel', async () => {
39
+ const toolCalls = [
40
+ {
41
+ id: 'call_001',
42
+ name: 'get_weather',
43
+ input: { city: 'San Francisco' },
44
+ },
45
+ {
46
+ id: 'call_002',
47
+ name: 'get_weather',
48
+ input: { city: 'New York' },
49
+ },
50
+ {
51
+ id: 'call_003',
52
+ name: 'get_weather',
53
+ input: { city: 'London' },
54
+ },
55
+ ];
56
+ const startTime = Date.now();
57
+ const results = await executeTools(toolCalls, toolMap);
58
+ const duration = Date.now() - startTime;
59
+ // Should execute in parallel (< 150ms total, not 120ms sequential)
60
+ expect(duration).toBeLessThan(150);
61
+ expect(results).toHaveLength(3);
62
+ expect(results[0].is_error).toBe(false);
63
+ expect(results[1].is_error).toBe(false);
64
+ expect(results[2].is_error).toBe(false);
65
+ expect(results[0].result.temperature).toBe(65);
66
+ expect(results[1].result.temperature).toBe(75);
67
+ expect(results[2].result.temperature).toBe(55);
68
+ });
69
+ it('handles tool not found error', async () => {
70
+ const toolCalls = [
71
+ {
72
+ id: 'call_001',
73
+ name: 'nonexistent_tool',
74
+ input: {},
75
+ },
76
+ ];
77
+ const results = await executeTools(toolCalls, toolMap);
78
+ expect(results).toHaveLength(1);
79
+ expect(results[0].call_id).toBe('call_001');
80
+ expect(results[0].is_error).toBe(true);
81
+ expect(results[0].error_message).toContain('nonexistent_tool');
82
+ expect(results[0].error_message).toContain('Available tools:');
83
+ });
84
+ it('handles tool execution error', async () => {
85
+ const toolCalls = [
86
+ {
87
+ id: 'call_001',
88
+ name: 'get_weather',
89
+ input: { city: 'InvalidCity' },
90
+ },
91
+ ];
92
+ const results = await executeTools(toolCalls, toolMap);
93
+ expect(results).toHaveLength(1);
94
+ expect(results[0].call_id).toBe('call_001');
95
+ expect(results[0].is_error).toBe(true);
96
+ expect(results[0].error_message).toContain('Weather data not available');
97
+ });
98
+ it('handles mix of successful and failed tool calls', async () => {
99
+ const toolCalls = [
100
+ {
101
+ id: 'call_001',
102
+ name: 'get_weather',
103
+ input: { city: 'San Francisco' },
104
+ },
105
+ {
106
+ id: 'call_002',
107
+ name: 'get_weather',
108
+ input: { city: 'InvalidCity' },
109
+ },
110
+ {
111
+ id: 'call_003',
112
+ name: 'get_weather',
113
+ input: { city: 'New York' },
114
+ },
115
+ ];
116
+ const results = await executeTools(toolCalls, toolMap);
117
+ expect(results).toHaveLength(3);
118
+ expect(results[0].is_error).toBe(false);
119
+ expect(results[1].is_error).toBe(true);
120
+ expect(results[2].is_error).toBe(false);
121
+ });
122
+ it('executes tools with different parameters', async () => {
123
+ const toolCalls = [
124
+ {
125
+ id: 'call_001',
126
+ name: 'get_team_members',
127
+ input: {},
128
+ },
129
+ {
130
+ id: 'call_002',
131
+ name: 'get_expenses',
132
+ input: { user_id: 'u1' },
133
+ },
134
+ {
135
+ id: 'call_003',
136
+ name: 'calculator',
137
+ input: { expression: '2 + 2 * 3' },
138
+ },
139
+ ];
140
+ const results = await executeTools(toolCalls, toolMap);
141
+ expect(results).toHaveLength(3);
142
+ expect(results[0].is_error).toBe(false);
143
+ expect(results[1].is_error).toBe(false);
144
+ expect(results[2].is_error).toBe(false);
145
+ expect(Array.isArray(results[0].result)).toBe(true);
146
+ expect(results[0].result).toHaveLength(3);
147
+ expect(Array.isArray(results[1].result)).toBe(true);
148
+ expect(results[2].result.result).toBe(8);
149
+ });
150
+ });
151
+ describe('normalizeToPythonIdentifier', () => {
152
+ it('converts hyphens to underscores', () => {
153
+ expect(normalizeToPythonIdentifier('my-tool-name')).toBe('my_tool_name');
154
+ });
155
+ it('converts spaces to underscores', () => {
156
+ expect(normalizeToPythonIdentifier('my tool name')).toBe('my_tool_name');
157
+ });
158
+ it('leaves underscores unchanged', () => {
159
+ expect(normalizeToPythonIdentifier('my_tool_name')).toBe('my_tool_name');
160
+ });
161
+ it('handles mixed hyphens and underscores', () => {
162
+ expect(normalizeToPythonIdentifier('my-tool_name-v2')).toBe('my_tool_name_v2');
163
+ });
164
+ it('handles MCP-style names with hyphens', () => {
165
+ expect(normalizeToPythonIdentifier('create_spreadsheet_mcp_Google-Workspace')).toBe('create_spreadsheet_mcp_Google_Workspace');
166
+ });
167
+ it('removes invalid characters', () => {
168
+ expect(normalizeToPythonIdentifier('tool@name!v2')).toBe('toolnamev2');
169
+ expect(normalizeToPythonIdentifier('get.data.v2')).toBe('getdatav2');
170
+ });
171
+ it('prefixes with underscore if starts with number', () => {
172
+ expect(normalizeToPythonIdentifier('123tool')).toBe('_123tool');
173
+ expect(normalizeToPythonIdentifier('1-tool')).toBe('_1_tool');
174
+ });
175
+ it('appends _tool suffix for Python keywords', () => {
176
+ expect(normalizeToPythonIdentifier('return')).toBe('return_tool');
177
+ expect(normalizeToPythonIdentifier('async')).toBe('async_tool');
178
+ expect(normalizeToPythonIdentifier('import')).toBe('import_tool');
179
+ });
180
+ });
181
+ describe('unwrapToolResponse', () => {
182
+ describe('non-MCP tools', () => {
183
+ it('returns result as-is for non-MCP tools', () => {
184
+ const result = { temperature: 65, condition: 'Foggy' };
185
+ expect(unwrapToolResponse(result, false)).toEqual(result);
186
+ });
187
+ it('returns string as-is for non-MCP tools', () => {
188
+ expect(unwrapToolResponse('plain string', false)).toBe('plain string');
189
+ });
190
+ it('returns array as-is for non-MCP tools', () => {
191
+ const result = [1, 2, 3];
192
+ expect(unwrapToolResponse(result, false)).toEqual(result);
193
+ });
194
+ });
195
+ describe('MCP tools - tuple format [content, artifacts]', () => {
196
+ it('extracts string content from tuple', () => {
197
+ const result = ['Hello world', { artifacts: [] }];
198
+ expect(unwrapToolResponse(result, true)).toBe('Hello world');
199
+ });
200
+ it('parses JSON string content from tuple', () => {
201
+ const result = ['{"temperature": 65}', { artifacts: [] }];
202
+ expect(unwrapToolResponse(result, true)).toEqual({ temperature: 65 });
203
+ });
204
+ it('parses JSON array string content from tuple', () => {
205
+ const result = ['[1, 2, 3]', { artifacts: [] }];
206
+ expect(unwrapToolResponse(result, true)).toEqual([1, 2, 3]);
207
+ });
208
+ it('extracts text from single content block in tuple', () => {
209
+ const result = [{ type: 'text', text: 'Spreadsheet info here' }, {}];
210
+ expect(unwrapToolResponse(result, true)).toBe('Spreadsheet info here');
211
+ });
212
+ it('extracts and parses JSON from single content block in tuple', () => {
213
+ const result = [
214
+ { type: 'text', text: '{"id": "123", "name": "Test"}' },
215
+ {},
216
+ ];
217
+ expect(unwrapToolResponse(result, true)).toEqual({
218
+ id: '123',
219
+ name: 'Test',
220
+ });
221
+ });
222
+ it('extracts text from array of content blocks in tuple', () => {
223
+ const result = [
224
+ [
225
+ { type: 'text', text: 'Line 1' },
226
+ { type: 'text', text: 'Line 2' },
227
+ ],
228
+ {},
229
+ ];
230
+ expect(unwrapToolResponse(result, true)).toBe('Line 1\nLine 2');
231
+ });
232
+ it('returns object content as-is when not a text block', () => {
233
+ const result = [{ temperature: 65, condition: 'Foggy' }, {}];
234
+ expect(unwrapToolResponse(result, true)).toEqual({
235
+ temperature: 65,
236
+ condition: 'Foggy',
237
+ });
238
+ });
239
+ });
240
+ describe('MCP tools - single content block (not in tuple)', () => {
241
+ it('extracts text from single content block object', () => {
242
+ const result = { type: 'text', text: 'No data found in range' };
243
+ expect(unwrapToolResponse(result, true)).toBe('No data found in range');
244
+ });
245
+ it('extracts and parses JSON from single content block object', () => {
246
+ const result = {
247
+ type: 'text',
248
+ text: '{"sheets": [{"name": "raw_data"}]}',
249
+ };
250
+ expect(unwrapToolResponse(result, true)).toEqual({
251
+ sheets: [{ name: 'raw_data' }],
252
+ });
253
+ });
254
+ it('handles real-world MCP spreadsheet response', () => {
255
+ const result = {
256
+ type: 'text',
257
+ text: 'Spreadsheet: "NYC Taxi - Top Pickup Neighborhoods" (ID: abc123)\nSheets (2):\n - "raw_data" (ID: 123) | Size: 1000x26',
258
+ };
259
+ expect(unwrapToolResponse(result, true)).toBe('Spreadsheet: "NYC Taxi - Top Pickup Neighborhoods" (ID: abc123)\nSheets (2):\n - "raw_data" (ID: 123) | Size: 1000x26');
260
+ });
261
+ it('handles real-world MCP no data response', () => {
262
+ const result = {
263
+ type: 'text',
264
+ text: 'No data found in range \'raw_data!A1:D25\' for user@example.com.',
265
+ };
266
+ expect(unwrapToolResponse(result, true)).toBe('No data found in range \'raw_data!A1:D25\' for user@example.com.');
267
+ });
268
+ });
269
+ describe('MCP tools - array of content blocks (not in tuple)', () => {
270
+ it('extracts text from array of content blocks', () => {
271
+ const result = [
272
+ { type: 'text', text: 'First block' },
273
+ { type: 'text', text: 'Second block' },
274
+ ];
275
+ expect(unwrapToolResponse(result, true)).toBe('First block\nSecond block');
276
+ });
277
+ it('filters out non-text blocks', () => {
278
+ const result = [
279
+ { type: 'text', text: 'Text content' },
280
+ { type: 'image', data: 'base64...' },
281
+ { type: 'text', text: 'More text' },
282
+ ];
283
+ expect(unwrapToolResponse(result, true)).toBe('Text content\nMore text');
284
+ });
285
+ });
286
+ describe('edge cases', () => {
287
+ it('returns non-text block object as-is', () => {
288
+ const result = { type: 'image', data: 'base64...' };
289
+ expect(unwrapToolResponse(result, true)).toEqual(result);
290
+ });
291
+ it('handles empty array', () => {
292
+ expect(unwrapToolResponse([], true)).toEqual([]);
293
+ });
294
+ it('handles malformed JSON in text block gracefully', () => {
295
+ const result = { type: 'text', text: '{ invalid json }' };
296
+ expect(unwrapToolResponse(result, true)).toBe('{ invalid json }');
297
+ });
298
+ it('handles null', () => {
299
+ expect(unwrapToolResponse(null, true)).toBe(null);
300
+ });
301
+ it('handles undefined', () => {
302
+ expect(unwrapToolResponse(undefined, true)).toBe(undefined);
303
+ });
304
+ });
305
+ });
306
+ describe('extractUsedToolNames', () => {
307
+ const createToolMap = (names) => {
308
+ const map = new Map();
309
+ for (const name of names) {
310
+ map.set(normalizeToPythonIdentifier(name), name);
311
+ }
312
+ return map;
313
+ };
314
+ const availableTools = createToolMap([
315
+ 'get_weather',
316
+ 'get_team_members',
317
+ 'get_expenses',
318
+ 'calculator',
319
+ 'search_docs',
320
+ ]);
321
+ it('extracts single tool name from simple code', () => {
322
+ const code = `result = await get_weather(city="SF")
323
+ print(result)`;
324
+ const used = extractUsedToolNames(code, availableTools);
325
+ expect(used.size).toBe(1);
326
+ expect(used.has('get_weather')).toBe(true);
327
+ });
328
+ it('extracts multiple tool names from code', () => {
329
+ const code = `team = await get_team_members()
330
+ for member in team:
331
+ expenses = await get_expenses(user_id=member['id'])
332
+ print(f"{member['name']}: {sum(e['amount'] for e in expenses)}")`;
333
+ const used = extractUsedToolNames(code, availableTools);
334
+ expect(used.size).toBe(2);
335
+ expect(used.has('get_team_members')).toBe(true);
336
+ expect(used.has('get_expenses')).toBe(true);
337
+ });
338
+ it('extracts tools from asyncio.gather calls', () => {
339
+ const code = `results = await asyncio.gather(
340
+ get_weather(city="SF"),
341
+ get_weather(city="NYC"),
342
+ get_expenses(user_id="u1")
343
+ )`;
344
+ const used = extractUsedToolNames(code, availableTools);
345
+ expect(used.size).toBe(2);
346
+ expect(used.has('get_weather')).toBe(true);
347
+ expect(used.has('get_expenses')).toBe(true);
348
+ });
349
+ it('does not match partial tool names', () => {
350
+ const code = `# Using get_weather_data instead
351
+ result = await get_weather_data(city="SF")`;
352
+ const used = extractUsedToolNames(code, availableTools);
353
+ expect(used.has('get_weather')).toBe(false);
354
+ });
355
+ it('matches tool names in different contexts', () => {
356
+ const code = `# direct call
357
+ x = await calculator(expression="1+1")
358
+ # in list comprehension
359
+ results = [await get_weather(city=c) for c in cities]
360
+ # conditional
361
+ if condition:
362
+ await get_team_members()`;
363
+ const used = extractUsedToolNames(code, availableTools);
364
+ expect(used.size).toBe(3);
365
+ expect(used.has('calculator')).toBe(true);
366
+ expect(used.has('get_weather')).toBe(true);
367
+ expect(used.has('get_team_members')).toBe(true);
368
+ });
369
+ it('returns empty set when no tools are used', () => {
370
+ const code = `print("Hello, World!")
371
+ x = 1 + 2`;
372
+ const used = extractUsedToolNames(code, availableTools);
373
+ expect(used.size).toBe(0);
374
+ });
375
+ it('handles tool names with special characters via normalization', () => {
376
+ const specialTools = createToolMap(['get_data.v2', 'calc+plus']);
377
+ const code = `await get_datav2()
378
+ await calcplus()`;
379
+ const used = extractUsedToolNames(code, specialTools);
380
+ expect(used.has('get_data.v2')).toBe(true);
381
+ expect(used.has('calc+plus')).toBe(true);
382
+ });
383
+ it('matches hyphenated tool names using underscore in code', () => {
384
+ const mcpTools = createToolMap([
385
+ 'create_spreadsheet_mcp_Google-Workspace',
386
+ 'search_gmail_mcp_Google-Workspace',
387
+ ]);
388
+ const code = `result = await create_spreadsheet_mcp_Google_Workspace(title="Test")
389
+ print(result)`;
390
+ const used = extractUsedToolNames(code, mcpTools);
391
+ expect(used.size).toBe(1);
392
+ expect(used.has('create_spreadsheet_mcp_Google-Workspace')).toBe(true);
393
+ });
394
+ });
395
+ describe('filterToolsByUsage', () => {
396
+ const allToolDefs = [
397
+ {
398
+ name: 'get_weather',
399
+ description: 'Get weather for a city',
400
+ parameters: {
401
+ type: 'object',
402
+ properties: { city: { type: 'string' } },
403
+ },
404
+ },
405
+ {
406
+ name: 'get_team_members',
407
+ description: 'Get team members',
408
+ parameters: { type: 'object', properties: {} },
409
+ },
410
+ {
411
+ name: 'get_expenses',
412
+ description: 'Get expenses for a user',
413
+ parameters: {
414
+ type: 'object',
415
+ properties: { user_id: { type: 'string' } },
416
+ },
417
+ },
418
+ {
419
+ name: 'calculator',
420
+ description: 'Evaluate an expression',
421
+ parameters: {
422
+ type: 'object',
423
+ properties: { expression: { type: 'string' } },
424
+ },
425
+ },
426
+ ];
427
+ it('filters to only used tools', () => {
428
+ const code = `result = await get_weather(city="SF")
429
+ print(result)`;
430
+ const filtered = filterToolsByUsage(allToolDefs, code);
431
+ expect(filtered).toHaveLength(1);
432
+ expect(filtered[0].name).toBe('get_weather');
433
+ });
434
+ it('filters to multiple used tools', () => {
435
+ const code = `team = await get_team_members()
436
+ for member in team:
437
+ expenses = await get_expenses(user_id=member['id'])`;
438
+ const filtered = filterToolsByUsage(allToolDefs, code);
439
+ expect(filtered).toHaveLength(2);
440
+ expect(filtered.map((t) => t.name).sort()).toEqual([
441
+ 'get_expenses',
442
+ 'get_team_members',
443
+ ]);
444
+ });
445
+ it('returns all tools when no tools are detected', () => {
446
+ const code = 'print("Hello, World!")';
447
+ const filtered = filterToolsByUsage(allToolDefs, code);
448
+ expect(filtered).toHaveLength(4);
449
+ });
450
+ it('preserves tool definition structure', () => {
451
+ const code = 'await calculator(expression="2+2")';
452
+ const filtered = filterToolsByUsage(allToolDefs, code);
453
+ expect(filtered).toHaveLength(1);
454
+ expect(filtered[0]).toEqual(allToolDefs[3]);
455
+ expect(filtered[0].parameters).toBeDefined();
456
+ expect(filtered[0].description).toBe('Evaluate an expression');
457
+ });
458
+ it('handles empty tool definitions', () => {
459
+ const code = 'await get_weather(city="SF")';
460
+ const filtered = filterToolsByUsage([], code);
461
+ expect(filtered).toHaveLength(0);
462
+ });
463
+ });
464
+ describe('formatCompletedResponse', () => {
465
+ it('formats response with stdout', () => {
466
+ const response = {
467
+ status: 'completed',
468
+ stdout: 'Hello, World!\n',
469
+ stderr: '',
470
+ files: [],
471
+ session_id: 'sess_abc123',
472
+ };
473
+ const [output, artifact] = formatCompletedResponse(response);
474
+ expect(output).toContain('stdout:\nHello, World!');
475
+ expect(artifact.session_id).toBe('sess_abc123');
476
+ expect(artifact.files).toEqual([]);
477
+ });
478
+ it('shows empty output message when no stdout', () => {
479
+ const response = {
480
+ status: 'completed',
481
+ stdout: '',
482
+ stderr: '',
483
+ files: [],
484
+ session_id: 'sess_abc123',
485
+ };
486
+ const [output] = formatCompletedResponse(response);
487
+ expect(output).toContain('stdout: Empty. Ensure you\'re writing output explicitly');
488
+ });
489
+ it('includes stderr when present', () => {
490
+ const response = {
491
+ status: 'completed',
492
+ stdout: 'Output\n',
493
+ stderr: 'Warning: deprecated function\n',
494
+ files: [],
495
+ session_id: 'sess_abc123',
496
+ };
497
+ const [output] = formatCompletedResponse(response);
498
+ expect(output).toContain('stdout:\nOutput');
499
+ expect(output).toContain('stderr:\nWarning: deprecated function');
500
+ });
501
+ it('formats file information correctly', () => {
502
+ const response = {
503
+ status: 'completed',
504
+ stdout: 'Generated report\n',
505
+ stderr: '',
506
+ files: [
507
+ { id: '1', name: 'report.pdf' },
508
+ { id: '2', name: 'data.csv' },
509
+ ],
510
+ session_id: 'sess_abc123',
511
+ };
512
+ const [output, artifact] = formatCompletedResponse(response);
513
+ expect(output).toContain('Generated files:');
514
+ expect(output).toContain('report.pdf');
515
+ expect(output).toContain('data.csv');
516
+ expect(artifact.files).toHaveLength(2);
517
+ expect(artifact.files).toEqual(response.files);
518
+ });
519
+ it('handles image files with special message', () => {
520
+ const response = {
521
+ status: 'completed',
522
+ stdout: '',
523
+ stderr: '',
524
+ files: [
525
+ { id: '1', name: 'chart.png' },
526
+ { id: '2', name: 'photo.jpg' },
527
+ ],
528
+ session_id: 'sess_abc123',
529
+ };
530
+ const [output] = formatCompletedResponse(response);
531
+ expect(output).toContain('chart.png');
532
+ expect(output).toContain('Image is already displayed to the user');
533
+ });
534
+ });
535
+ describe('createProgrammaticToolCallingTool - Manual Invocation', () => {
536
+ let ptcTool;
537
+ let toolMap;
538
+ let toolDefinitions;
539
+ beforeEach(() => {
540
+ const tools = [
541
+ createGetTeamMembersTool(),
542
+ createGetExpensesTool(),
543
+ createGetWeatherTool(),
544
+ ];
545
+ toolMap = new Map(tools.map((t) => [t.name, t]));
546
+ toolDefinitions = Array.from(createProgrammaticToolRegistry().values()).filter((t) => ['get_team_members', 'get_expenses', 'get_weather'].includes(t.name));
547
+ ptcTool = createProgrammaticToolCallingTool({
548
+ apiKey: 'test-key',
549
+ baseUrl: 'http://mock-api',
550
+ });
551
+ });
552
+ it('throws error when no toolMap provided', async () => {
553
+ await expect(ptcTool.invoke({
554
+ code: 'result = await get_weather(city="SF")\nprint(result)',
555
+ tools: toolDefinitions,
556
+ toolMap,
557
+ })).rejects.toThrow('No toolMap provided');
558
+ });
559
+ it('throws error when toolMap is empty', async () => {
560
+ const args = {
561
+ code: 'result = await get_weather(city="SF")\nprint(result)',
562
+ tools: toolDefinitions,
563
+ toolMap: new Map(),
564
+ };
565
+ const toolCall = {
566
+ name: 'programmatic_tool_calling',
567
+ args,
568
+ };
569
+ await expect(ptcTool.invoke(args, {
570
+ toolCall,
571
+ })).rejects.toThrow('No toolMap provided');
572
+ });
573
+ it('throws error when no tool definitions provided', async () => {
574
+ const args = {
575
+ code: 'result = await get_weather(city="SF")\nprint(result)',
576
+ // No tools
577
+ };
578
+ const toolCall = {
579
+ name: 'programmatic_code_execution',
580
+ args,
581
+ toolMap,
582
+ // No `toolDefs`
583
+ };
584
+ await expect(ptcTool.invoke(args, { toolCall })).rejects.toThrow('No tool definitions provided');
585
+ });
586
+ it('uses toolDefs from config when tools not provided', async () => {
587
+ // Skip this test - requires mocking fetch which has complex typing
588
+ // This functionality is tested in the live script tests instead
589
+ });
590
+ });
591
+ describe('Tool Classification', () => {
592
+ it('filters tools by allowed_callers', () => {
593
+ const registry = createProgrammaticToolRegistry();
594
+ const codeExecutionTools = Array.from(registry.values()).filter((t) => (t.allowed_callers ?? ['direct']).includes('code_execution'));
595
+ // get_team_members, get_expenses, calculator: code_execution only
596
+ const codeOnlyTools = codeExecutionTools.filter((t) => !(t.allowed_callers?.includes('direct') === true));
597
+ expect(codeOnlyTools.length).toBeGreaterThanOrEqual(3);
598
+ // get_weather: both direct and code_execution
599
+ const bothTools = Array.from(registry.values()).filter((t) => t.allowed_callers?.includes('direct') === true &&
600
+ t.allowed_callers.includes('code_execution'));
601
+ expect(bothTools.length).toBeGreaterThanOrEqual(1);
602
+ expect(bothTools.some((t) => t.name === 'get_weather')).toBe(true);
603
+ });
604
+ });
605
+ describe('Error Handling', () => {
606
+ let toolMap;
607
+ beforeEach(() => {
608
+ const tools = [createGetWeatherTool()];
609
+ toolMap = new Map(tools.map((t) => [t.name, t]));
610
+ });
611
+ it('returns error for invalid city without throwing', async () => {
612
+ const toolCalls = [
613
+ {
614
+ id: 'call_001',
615
+ name: 'get_weather',
616
+ input: { city: 'InvalidCity' },
617
+ },
618
+ ];
619
+ const results = await executeTools(toolCalls, toolMap);
620
+ expect(results).toHaveLength(1);
621
+ expect(results[0].is_error).toBe(true);
622
+ expect(results[0].result).toBeNull();
623
+ expect(results[0].error_message).toContain('Weather data not available');
624
+ });
625
+ it('continues execution when one tool fails', async () => {
626
+ const toolCalls = [
627
+ {
628
+ id: 'call_001',
629
+ name: 'get_weather',
630
+ input: { city: 'San Francisco' },
631
+ },
632
+ {
633
+ id: 'call_002',
634
+ name: 'get_weather',
635
+ input: { city: 'InvalidCity' },
636
+ },
637
+ {
638
+ id: 'call_003',
639
+ name: 'get_weather',
640
+ input: { city: 'London' },
641
+ },
642
+ ];
643
+ const results = await executeTools(toolCalls, toolMap);
644
+ expect(results).toHaveLength(3);
645
+ expect(results[0].is_error).toBe(false);
646
+ expect(results[1].is_error).toBe(true);
647
+ expect(results[2].is_error).toBe(false);
648
+ });
649
+ });
650
+ describe('Parallel Execution Performance', () => {
651
+ let toolMap;
652
+ beforeEach(() => {
653
+ const tools = [createGetExpensesTool()];
654
+ toolMap = new Map(tools.map((t) => [t.name, t]));
655
+ });
656
+ it('executes tools in parallel, not sequentially', async () => {
657
+ const toolCalls = [
658
+ { id: 'call_001', name: 'get_expenses', input: { user_id: 'u1' } },
659
+ { id: 'call_002', name: 'get_expenses', input: { user_id: 'u2' } },
660
+ { id: 'call_003', name: 'get_expenses', input: { user_id: 'u3' } },
661
+ ];
662
+ const startTime = Date.now();
663
+ const results = await executeTools(toolCalls, toolMap);
664
+ const duration = Date.now() - startTime;
665
+ // Each tool has 30ms delay
666
+ // Sequential would be ~90ms, parallel should be ~30-50ms
667
+ expect(duration).toBeLessThan(80);
668
+ expect(results).toHaveLength(3);
669
+ expect(results.every((r) => r.is_error === false)).toBe(true);
670
+ });
671
+ });
672
+ describe('Response Formatting', () => {
673
+ it('formats stdout-only response', () => {
674
+ const response = {
675
+ status: 'completed',
676
+ stdout: 'Team size: 3\n- Alice\n- Bob\n- Charlie\n',
677
+ stderr: '',
678
+ files: [],
679
+ session_id: 'sess_xyz',
680
+ };
681
+ const [output, artifact] = formatCompletedResponse(response);
682
+ expect(output).toBe('stdout:\nTeam size: 3\n- Alice\n- Bob\n- Charlie');
683
+ expect(artifact).toEqual({
684
+ session_id: 'sess_xyz',
685
+ files: [],
686
+ });
687
+ });
688
+ it('formats response with files', () => {
689
+ const response = {
690
+ status: 'completed',
691
+ stdout: 'Report generated\n',
692
+ stderr: '',
693
+ files: [
694
+ { id: '1', name: 'report.csv' },
695
+ { id: '2', name: 'chart.png' },
696
+ ],
697
+ session_id: 'sess_xyz',
698
+ };
699
+ const [output, artifact] = formatCompletedResponse(response);
700
+ expect(output).toContain('Generated files:');
701
+ expect(output).toContain('report.csv');
702
+ expect(output).toContain('chart.png');
703
+ expect(output).toContain('File is already downloaded');
704
+ expect(output).toContain('Image is already displayed');
705
+ expect(artifact.files).toHaveLength(2);
706
+ });
707
+ it('handles multiple files with correct separators', () => {
708
+ const response = {
709
+ status: 'completed',
710
+ stdout: 'Done\n',
711
+ stderr: '',
712
+ files: [
713
+ { id: '1', name: 'file1.txt' },
714
+ { id: '2', name: 'file2.txt' },
715
+ ],
716
+ session_id: 'sess_xyz',
717
+ };
718
+ const [output] = formatCompletedResponse(response);
719
+ // 2 files format: "- /mnt/data/file1.txt | ..., - /mnt/data/file2.txt | ..."
720
+ expect(output).toContain('file1.txt');
721
+ expect(output).toContain('file2.txt');
722
+ expect(output).toContain('- /mnt/data/file1.txt');
723
+ expect(output).toContain('- /mnt/data/file2.txt');
724
+ });
725
+ it('handles many files with newline separators', () => {
726
+ const response = {
727
+ status: 'completed',
728
+ stdout: 'Done\n',
729
+ stderr: '',
730
+ files: [
731
+ { id: '1', name: 'file1.txt' },
732
+ { id: '2', name: 'file2.txt' },
733
+ { id: '3', name: 'file3.txt' },
734
+ { id: '4', name: 'file4.txt' },
735
+ ],
736
+ session_id: 'sess_xyz',
737
+ };
738
+ const [output] = formatCompletedResponse(response);
739
+ // More than 3 files should use newline separators
740
+ expect(output).toContain('file1.txt');
741
+ expect(output).toContain('file4.txt');
742
+ expect(output.match(/,\n/g)?.length).toBeGreaterThanOrEqual(2);
743
+ });
744
+ });
745
+ describe('Tool Data Extraction', () => {
746
+ let toolMap;
747
+ beforeEach(() => {
748
+ const tools = [
749
+ createGetTeamMembersTool(),
750
+ createGetExpensesTool(),
751
+ createCalculatorTool(),
752
+ ];
753
+ toolMap = new Map(tools.map((t) => [t.name, t]));
754
+ });
755
+ it('extracts correct data from team members tool', async () => {
756
+ const toolCalls = [
757
+ { id: 'call_001', name: 'get_team_members', input: {} },
758
+ ];
759
+ const results = await executeTools(toolCalls, toolMap);
760
+ expect(results[0].result).toEqual([
761
+ { id: 'u1', name: 'Alice', department: 'Engineering' },
762
+ { id: 'u2', name: 'Bob', department: 'Marketing' },
763
+ { id: 'u3', name: 'Charlie', department: 'Engineering' },
764
+ ]);
765
+ });
766
+ it('extracts correct data from expenses tool', async () => {
767
+ const toolCalls = [
768
+ { id: 'call_001', name: 'get_expenses', input: { user_id: 'u1' } },
769
+ ];
770
+ const results = await executeTools(toolCalls, toolMap);
771
+ expect(results[0].result).toEqual([
772
+ { amount: 150.0, category: 'travel' },
773
+ { amount: 75.5, category: 'meals' },
774
+ ]);
775
+ });
776
+ it('handles empty expense data', async () => {
777
+ const toolCalls = [
778
+ {
779
+ id: 'call_001',
780
+ name: 'get_expenses',
781
+ input: { user_id: 'nonexistent' },
782
+ },
783
+ ];
784
+ const results = await executeTools(toolCalls, toolMap);
785
+ expect(results[0].is_error).toBe(false);
786
+ expect(results[0].result).toEqual([]);
787
+ });
788
+ it('calculates correct result', async () => {
789
+ const toolCalls = [
790
+ {
791
+ id: 'call_001',
792
+ name: 'calculator',
793
+ input: { expression: '2 + 2 * 3' },
794
+ },
795
+ {
796
+ id: 'call_002',
797
+ name: 'calculator',
798
+ input: { expression: '(10 + 5) / 3' },
799
+ },
800
+ ];
801
+ const results = await executeTools(toolCalls, toolMap);
802
+ expect(results[0].result.result).toBe(8);
803
+ expect(results[1].result.result).toBe(5);
804
+ });
805
+ });
806
+ });
807
+ //# sourceMappingURL=ProgrammaticToolCalling.test.js.map