@namzu/sdk 0.6.0 → 1.1.0

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 (429) hide show
  1. package/CHANGELOG.md +529 -0
  2. package/dist/advisory/executor.d.ts.map +1 -1
  3. package/dist/advisory/executor.js +9 -2
  4. package/dist/advisory/executor.js.map +1 -1
  5. package/dist/advisory/executor.test.d.ts +2 -1
  6. package/dist/advisory/executor.test.d.ts.map +1 -1
  7. package/dist/advisory/executor.test.js +7 -4
  8. package/dist/advisory/executor.test.js.map +1 -1
  9. package/dist/agents/ReactiveAgent.d.ts.map +1 -1
  10. package/dist/agents/ReactiveAgent.js +7 -0
  11. package/dist/agents/ReactiveAgent.js.map +1 -1
  12. package/dist/agents/SupervisorAgent.d.ts +22 -1
  13. package/dist/agents/SupervisorAgent.d.ts.map +1 -1
  14. package/dist/agents/SupervisorAgent.js +54 -14
  15. package/dist/agents/SupervisorAgent.js.map +1 -1
  16. package/dist/agents/__tests__/supervisor-ledger.test.d.ts +2 -0
  17. package/dist/agents/__tests__/supervisor-ledger.test.d.ts.map +1 -0
  18. package/dist/agents/__tests__/supervisor-ledger.test.js +131 -0
  19. package/dist/agents/__tests__/supervisor-ledger.test.js.map +1 -0
  20. package/dist/bridge/sse/mapper.test.js +2 -2
  21. package/dist/config/runtime.d.ts +24 -0
  22. package/dist/config/runtime.d.ts.map +1 -1
  23. package/dist/config/runtime.js +9 -0
  24. package/dist/config/runtime.js.map +1 -1
  25. package/dist/connector/builtins/webhook.test.js +17 -0
  26. package/dist/connector/builtins/webhook.test.js.map +1 -1
  27. package/dist/connector/index.d.ts +1 -0
  28. package/dist/connector/index.d.ts.map +1 -1
  29. package/dist/connector/index.js +1 -0
  30. package/dist/connector/index.js.map +1 -1
  31. package/dist/connector/mcp/client.d.ts.map +1 -1
  32. package/dist/connector/mcp/client.js +4 -0
  33. package/dist/connector/mcp/client.js.map +1 -1
  34. package/dist/connector/mcp/index.d.ts +1 -0
  35. package/dist/connector/mcp/index.d.ts.map +1 -1
  36. package/dist/connector/mcp/index.js +1 -0
  37. package/dist/connector/mcp/index.js.map +1 -1
  38. package/dist/connector/mcp/streamable-http.d.ts +22 -0
  39. package/dist/connector/mcp/streamable-http.d.ts.map +1 -0
  40. package/dist/connector/mcp/streamable-http.js +128 -0
  41. package/dist/connector/mcp/streamable-http.js.map +1 -0
  42. package/dist/connector/mcp/streamable-http.test.d.ts +2 -0
  43. package/dist/connector/mcp/streamable-http.test.d.ts.map +1 -0
  44. package/dist/connector/mcp/streamable-http.test.js +132 -0
  45. package/dist/connector/mcp/streamable-http.test.js.map +1 -0
  46. package/dist/constants/compaction/index.d.ts.map +1 -1
  47. package/dist/constants/compaction/index.js +8 -3
  48. package/dist/constants/compaction/index.js.map +1 -1
  49. package/dist/constants/sandbox/index.d.ts +21 -0
  50. package/dist/constants/sandbox/index.d.ts.map +1 -1
  51. package/dist/constants/sandbox/index.js +30 -0
  52. package/dist/constants/sandbox/index.js.map +1 -1
  53. package/dist/constants/tools/index.d.ts.map +1 -1
  54. package/dist/constants/tools/index.js +33 -2
  55. package/dist/constants/tools/index.js.map +1 -1
  56. package/dist/manager/run/persistence.d.ts.map +1 -1
  57. package/dist/manager/run/persistence.js +35 -5
  58. package/dist/manager/run/persistence.js.map +1 -1
  59. package/dist/persona/assembler.d.ts +1 -0
  60. package/dist/persona/assembler.d.ts.map +1 -1
  61. package/dist/persona/assembler.js +37 -6
  62. package/dist/persona/assembler.js.map +1 -1
  63. package/dist/provider/collect.d.ts.map +1 -1
  64. package/dist/provider/collect.js +4 -1
  65. package/dist/provider/collect.js.map +1 -1
  66. package/dist/provider/collect.test.js +2 -2
  67. package/dist/public-runtime.d.ts +6 -5
  68. package/dist/public-runtime.d.ts.map +1 -1
  69. package/dist/public-runtime.js +6 -5
  70. package/dist/public-runtime.js.map +1 -1
  71. package/dist/public-tools.d.ts +2 -0
  72. package/dist/public-tools.d.ts.map +1 -1
  73. package/dist/public-tools.js +2 -0
  74. package/dist/public-tools.js.map +1 -1
  75. package/dist/public-types.d.ts +3 -0
  76. package/dist/public-types.d.ts.map +1 -1
  77. package/dist/registry/index.d.ts +2 -0
  78. package/dist/registry/index.d.ts.map +1 -1
  79. package/dist/registry/index.js +1 -0
  80. package/dist/registry/index.js.map +1 -1
  81. package/dist/registry/tool/execute.d.ts +27 -0
  82. package/dist/registry/tool/execute.d.ts.map +1 -1
  83. package/dist/registry/tool/execute.js +192 -7
  84. package/dist/registry/tool/execute.js.map +1 -1
  85. package/dist/registry/tool/execute.test.d.ts +9 -4
  86. package/dist/registry/tool/execute.test.d.ts.map +1 -1
  87. package/dist/registry/tool/execute.test.js +178 -7
  88. package/dist/registry/tool/execute.test.js.map +1 -1
  89. package/dist/registry/toolset/catalog.d.ts +42 -0
  90. package/dist/registry/toolset/catalog.d.ts.map +1 -0
  91. package/dist/registry/toolset/catalog.js +217 -0
  92. package/dist/registry/toolset/catalog.js.map +1 -0
  93. package/dist/registry/toolset/catalog.test.d.ts +2 -0
  94. package/dist/registry/toolset/catalog.test.d.ts.map +1 -0
  95. package/dist/registry/toolset/catalog.test.js +147 -0
  96. package/dist/registry/toolset/catalog.test.js.map +1 -0
  97. package/dist/runtime/query/__tests__/deferred-tools.test.d.ts +2 -0
  98. package/dist/runtime/query/__tests__/deferred-tools.test.d.ts.map +1 -0
  99. package/dist/runtime/query/__tests__/deferred-tools.test.js +185 -0
  100. package/dist/runtime/query/__tests__/deferred-tools.test.js.map +1 -0
  101. package/dist/runtime/query/__tests__/executor-concurrency.test.d.ts +2 -0
  102. package/dist/runtime/query/__tests__/executor-concurrency.test.d.ts.map +1 -0
  103. package/dist/runtime/query/__tests__/executor-concurrency.test.js +98 -0
  104. package/dist/runtime/query/__tests__/executor-concurrency.test.js.map +1 -0
  105. package/dist/runtime/query/__tests__/executor-plugin-hooks.test.js +38 -3
  106. package/dist/runtime/query/__tests__/executor-plugin-hooks.test.js.map +1 -1
  107. package/dist/runtime/query/__tests__/long-document-flow.test.d.ts +2 -0
  108. package/dist/runtime/query/__tests__/long-document-flow.test.d.ts.map +1 -0
  109. package/dist/runtime/query/__tests__/long-document-flow.test.js +154 -0
  110. package/dist/runtime/query/__tests__/long-document-flow.test.js.map +1 -0
  111. package/dist/runtime/query/__tests__/prompt.test.js +51 -2
  112. package/dist/runtime/query/__tests__/prompt.test.js.map +1 -1
  113. package/dist/runtime/query/__tests__/stream-recovery.test.d.ts +2 -0
  114. package/dist/runtime/query/__tests__/stream-recovery.test.d.ts.map +1 -0
  115. package/dist/runtime/query/__tests__/stream-recovery.test.js +126 -0
  116. package/dist/runtime/query/__tests__/stream-recovery.test.js.map +1 -0
  117. package/dist/runtime/query/continuation.d.ts +16 -0
  118. package/dist/runtime/query/continuation.d.ts.map +1 -0
  119. package/dist/runtime/query/continuation.js +16 -0
  120. package/dist/runtime/query/continuation.js.map +1 -0
  121. package/dist/runtime/query/executor.d.ts +3 -0
  122. package/dist/runtime/query/executor.d.ts.map +1 -1
  123. package/dist/runtime/query/executor.js +71 -3
  124. package/dist/runtime/query/executor.js.map +1 -1
  125. package/dist/runtime/query/index.d.ts +9 -0
  126. package/dist/runtime/query/index.d.ts.map +1 -1
  127. package/dist/runtime/query/index.js +42 -5
  128. package/dist/runtime/query/index.js.map +1 -1
  129. package/dist/runtime/query/iteration/index.d.ts +24 -0
  130. package/dist/runtime/query/iteration/index.d.ts.map +1 -1
  131. package/dist/runtime/query/iteration/index.js +326 -66
  132. package/dist/runtime/query/iteration/index.js.map +1 -1
  133. package/dist/runtime/query/iteration/phases/checkpoint.d.ts.map +1 -1
  134. package/dist/runtime/query/iteration/phases/checkpoint.js +2 -2
  135. package/dist/runtime/query/iteration/phases/checkpoint.js.map +1 -1
  136. package/dist/runtime/query/iteration/phases/compaction.d.ts +8 -0
  137. package/dist/runtime/query/iteration/phases/compaction.d.ts.map +1 -1
  138. package/dist/runtime/query/iteration/phases/compaction.js +84 -3
  139. package/dist/runtime/query/iteration/phases/compaction.js.map +1 -1
  140. package/dist/runtime/query/iteration/phases/context.d.ts +22 -0
  141. package/dist/runtime/query/iteration/phases/context.d.ts.map +1 -1
  142. package/dist/runtime/query/iteration/phases/context.js +53 -0
  143. package/dist/runtime/query/iteration/phases/context.js.map +1 -1
  144. package/dist/runtime/query/iteration/phases/hitl-abort-race.test.d.ts +2 -0
  145. package/dist/runtime/query/iteration/phases/hitl-abort-race.test.d.ts.map +1 -0
  146. package/dist/runtime/query/iteration/phases/hitl-abort-race.test.js +73 -0
  147. package/dist/runtime/query/iteration/phases/hitl-abort-race.test.js.map +1 -0
  148. package/dist/runtime/query/iteration/phases/hitl-answer-question.test.d.ts +2 -0
  149. package/dist/runtime/query/iteration/phases/hitl-answer-question.test.d.ts.map +1 -0
  150. package/dist/runtime/query/iteration/phases/hitl-answer-question.test.js +120 -0
  151. package/dist/runtime/query/iteration/phases/hitl-answer-question.test.js.map +1 -0
  152. package/dist/runtime/query/iteration/phases/tool-review.d.ts +1 -1
  153. package/dist/runtime/query/iteration/phases/tool-review.d.ts.map +1 -1
  154. package/dist/runtime/query/iteration/phases/tool-review.js +7 -2
  155. package/dist/runtime/query/iteration/phases/tool-review.js.map +1 -1
  156. package/dist/runtime/query/iteration/phases/working-memory-compaction.test.d.ts +19 -0
  157. package/dist/runtime/query/iteration/phases/working-memory-compaction.test.d.ts.map +1 -0
  158. package/dist/runtime/query/iteration/phases/working-memory-compaction.test.js +177 -0
  159. package/dist/runtime/query/iteration/phases/working-memory-compaction.test.js.map +1 -0
  160. package/dist/runtime/query/iteration/phases/working-memory.d.ts +31 -0
  161. package/dist/runtime/query/iteration/phases/working-memory.d.ts.map +1 -0
  162. package/dist/runtime/query/iteration/phases/working-memory.js +81 -0
  163. package/dist/runtime/query/iteration/phases/working-memory.js.map +1 -0
  164. package/dist/runtime/query/prompt.d.ts.map +1 -1
  165. package/dist/runtime/query/prompt.js +21 -1
  166. package/dist/runtime/query/prompt.js.map +1 -1
  167. package/dist/runtime/query/tooling.d.ts +1 -0
  168. package/dist/runtime/query/tooling.d.ts.map +1 -1
  169. package/dist/runtime/query/tooling.js +1 -0
  170. package/dist/runtime/query/tooling.js.map +1 -1
  171. package/dist/sandbox/provider/local.d.ts.map +1 -1
  172. package/dist/sandbox/provider/local.js +32 -1
  173. package/dist/sandbox/provider/local.js.map +1 -1
  174. package/dist/session/workspace/__tests__/shared-run.test.d.ts +2 -0
  175. package/dist/session/workspace/__tests__/shared-run.test.d.ts.map +1 -0
  176. package/dist/session/workspace/__tests__/shared-run.test.js +147 -0
  177. package/dist/session/workspace/__tests__/shared-run.test.js.map +1 -0
  178. package/dist/session/workspace/index.d.ts +2 -0
  179. package/dist/session/workspace/index.d.ts.map +1 -1
  180. package/dist/session/workspace/index.js +1 -0
  181. package/dist/session/workspace/index.js.map +1 -1
  182. package/dist/session/workspace/shared-run.d.ts +81 -0
  183. package/dist/session/workspace/shared-run.d.ts.map +1 -0
  184. package/dist/session/workspace/shared-run.js +252 -0
  185. package/dist/session/workspace/shared-run.js.map +1 -0
  186. package/dist/skills/loader.d.ts.map +1 -1
  187. package/dist/skills/loader.js +36 -6
  188. package/dist/skills/loader.js.map +1 -1
  189. package/dist/skills/loader.test.d.ts +2 -0
  190. package/dist/skills/loader.test.d.ts.map +1 -0
  191. package/dist/skills/loader.test.js +65 -0
  192. package/dist/skills/loader.test.js.map +1 -0
  193. package/dist/streaming/coalesce.test.js +1 -1
  194. package/dist/streaming/coalesce.test.js.map +1 -1
  195. package/dist/tools/builtins/__tests__/edit.test.d.ts +2 -0
  196. package/dist/tools/builtins/__tests__/edit.test.d.ts.map +1 -0
  197. package/dist/tools/builtins/__tests__/edit.test.js +38 -0
  198. package/dist/tools/builtins/__tests__/edit.test.js.map +1 -0
  199. package/dist/tools/builtins/__tests__/payload-budget.test.d.ts +2 -0
  200. package/dist/tools/builtins/__tests__/payload-budget.test.d.ts.map +1 -0
  201. package/dist/tools/builtins/__tests__/payload-budget.test.js +70 -0
  202. package/dist/tools/builtins/__tests__/payload-budget.test.js.map +1 -0
  203. package/dist/tools/builtins/__tests__/read-file.test.d.ts +2 -0
  204. package/dist/tools/builtins/__tests__/read-file.test.d.ts.map +1 -0
  205. package/dist/tools/builtins/__tests__/read-file.test.js +33 -0
  206. package/dist/tools/builtins/__tests__/read-file.test.js.map +1 -0
  207. package/dist/tools/builtins/__tests__/verify-outputs.test.d.ts +2 -0
  208. package/dist/tools/builtins/__tests__/verify-outputs.test.d.ts.map +1 -0
  209. package/dist/tools/builtins/__tests__/verify-outputs.test.js +52 -0
  210. package/dist/tools/builtins/__tests__/verify-outputs.test.js.map +1 -0
  211. package/dist/tools/builtins/__tests__/write-file.test.d.ts +2 -0
  212. package/dist/tools/builtins/__tests__/write-file.test.d.ts.map +1 -0
  213. package/dist/tools/builtins/__tests__/write-file.test.js +74 -0
  214. package/dist/tools/builtins/__tests__/write-file.test.js.map +1 -0
  215. package/dist/tools/builtins/bash.d.ts.map +1 -1
  216. package/dist/tools/builtins/bash.js +40 -7
  217. package/dist/tools/builtins/bash.js.map +1 -1
  218. package/dist/tools/builtins/edit.d.ts +5 -2
  219. package/dist/tools/builtins/edit.d.ts.map +1 -1
  220. package/dist/tools/builtins/edit.js +114 -18
  221. package/dist/tools/builtins/edit.js.map +1 -1
  222. package/dist/tools/builtins/index.d.ts +1 -0
  223. package/dist/tools/builtins/index.d.ts.map +1 -1
  224. package/dist/tools/builtins/index.js +13 -13
  225. package/dist/tools/builtins/index.js.map +1 -1
  226. package/dist/tools/builtins/read-file.d.ts +1 -0
  227. package/dist/tools/builtins/read-file.d.ts.map +1 -1
  228. package/dist/tools/builtins/read-file.js +73 -10
  229. package/dist/tools/builtins/read-file.js.map +1 -1
  230. package/dist/tools/builtins/search-tools.d.ts.map +1 -1
  231. package/dist/tools/builtins/search-tools.js +35 -7
  232. package/dist/tools/builtins/search-tools.js.map +1 -1
  233. package/dist/tools/builtins/verify-outputs.d.ts +5 -0
  234. package/dist/tools/builtins/verify-outputs.d.ts.map +1 -0
  235. package/dist/tools/builtins/verify-outputs.js +103 -0
  236. package/dist/tools/builtins/verify-outputs.js.map +1 -0
  237. package/dist/tools/builtins/write-file.d.ts +3 -2
  238. package/dist/tools/builtins/write-file.d.ts.map +1 -1
  239. package/dist/tools/builtins/write-file.js +72 -12
  240. package/dist/tools/builtins/write-file.js.map +1 -1
  241. package/dist/tools/coordinator/__tests__/agent.test.d.ts +15 -0
  242. package/dist/tools/coordinator/__tests__/agent.test.d.ts.map +1 -0
  243. package/dist/tools/coordinator/__tests__/agent.test.js +142 -0
  244. package/dist/tools/coordinator/__tests__/agent.test.js.map +1 -0
  245. package/dist/tools/coordinator/__tests__/approve-plan.test.d.ts +2 -0
  246. package/dist/tools/coordinator/__tests__/approve-plan.test.d.ts.map +1 -0
  247. package/dist/tools/coordinator/__tests__/approve-plan.test.js +114 -0
  248. package/dist/tools/coordinator/__tests__/approve-plan.test.js.map +1 -0
  249. package/dist/tools/coordinator/__tests__/ask-user-question.test.d.ts +2 -0
  250. package/dist/tools/coordinator/__tests__/ask-user-question.test.d.ts.map +1 -0
  251. package/dist/tools/coordinator/__tests__/ask-user-question.test.js +309 -0
  252. package/dist/tools/coordinator/__tests__/ask-user-question.test.js.map +1 -0
  253. package/dist/tools/coordinator/__tests__/task-list.test.d.ts +13 -0
  254. package/dist/tools/coordinator/__tests__/task-list.test.d.ts.map +1 -0
  255. package/dist/tools/coordinator/__tests__/task-list.test.js +162 -0
  256. package/dist/tools/coordinator/__tests__/task-list.test.js.map +1 -0
  257. package/dist/tools/coordinator/agent.d.ts +34 -0
  258. package/dist/tools/coordinator/agent.d.ts.map +1 -0
  259. package/dist/tools/coordinator/agent.js +107 -0
  260. package/dist/tools/coordinator/agent.js.map +1 -0
  261. package/dist/tools/coordinator/index.d.ts +15 -0
  262. package/dist/tools/coordinator/index.d.ts.map +1 -1
  263. package/dist/tools/coordinator/index.js +305 -35
  264. package/dist/tools/coordinator/index.js.map +1 -1
  265. package/dist/types/agent/base.d.ts +8 -0
  266. package/dist/types/agent/base.d.ts.map +1 -1
  267. package/dist/types/agent/index.d.ts +1 -0
  268. package/dist/types/agent/index.d.ts.map +1 -1
  269. package/dist/types/agent/index.js +1 -0
  270. package/dist/types/agent/index.js.map +1 -1
  271. package/dist/types/agent/reactive.d.ts +38 -0
  272. package/dist/types/agent/reactive.d.ts.map +1 -1
  273. package/dist/types/agent/supervisor.d.ts +32 -0
  274. package/dist/types/agent/supervisor.d.ts.map +1 -1
  275. package/dist/types/agent/working-memory.d.ts +29 -0
  276. package/dist/types/agent/working-memory.d.ts.map +1 -0
  277. package/dist/types/agent/working-memory.js +2 -0
  278. package/dist/types/agent/working-memory.js.map +1 -0
  279. package/dist/types/common/index.d.ts +10 -0
  280. package/dist/types/common/index.d.ts.map +1 -1
  281. package/dist/types/common/index.js +18 -0
  282. package/dist/types/common/index.js.map +1 -1
  283. package/dist/types/common/usage-merge.test.d.ts +2 -0
  284. package/dist/types/common/usage-merge.test.d.ts.map +1 -0
  285. package/dist/types/common/usage-merge.test.js +48 -0
  286. package/dist/types/common/usage-merge.test.js.map +1 -0
  287. package/dist/types/connector/mcp.d.ts +9 -2
  288. package/dist/types/connector/mcp.d.ts.map +1 -1
  289. package/dist/types/hitl/index.d.ts +42 -0
  290. package/dist/types/hitl/index.d.ts.map +1 -1
  291. package/dist/types/hitl/index.js +11 -0
  292. package/dist/types/hitl/index.js.map +1 -1
  293. package/dist/types/message/index.d.ts +22 -1
  294. package/dist/types/message/index.d.ts.map +1 -1
  295. package/dist/types/message/index.js +7 -2
  296. package/dist/types/message/index.js.map +1 -1
  297. package/dist/types/provider/chat.d.ts +11 -9
  298. package/dist/types/provider/chat.d.ts.map +1 -1
  299. package/dist/types/run/events.d.ts +6 -0
  300. package/dist/types/run/events.d.ts.map +1 -1
  301. package/dist/types/run/events.js.map +1 -1
  302. package/dist/types/sandbox/index.d.ts +193 -0
  303. package/dist/types/sandbox/index.d.ts.map +1 -1
  304. package/dist/types/sandbox/index.js.map +1 -1
  305. package/dist/types/skills/index.d.ts +2 -0
  306. package/dist/types/skills/index.d.ts.map +1 -1
  307. package/dist/types/tool/index.d.ts +22 -0
  308. package/dist/types/tool/index.d.ts.map +1 -1
  309. package/dist/types/toolset/index.d.ts +71 -0
  310. package/dist/types/toolset/index.d.ts.map +1 -0
  311. package/dist/types/toolset/index.js +2 -0
  312. package/dist/types/toolset/index.js.map +1 -0
  313. package/dist/types/workspace/index.d.ts +1 -0
  314. package/dist/types/workspace/index.d.ts.map +1 -1
  315. package/dist/types/workspace/shared-run.d.ts +61 -0
  316. package/dist/types/workspace/shared-run.d.ts.map +1 -0
  317. package/dist/types/workspace/shared-run.js +2 -0
  318. package/dist/types/workspace/shared-run.js.map +1 -0
  319. package/dist/verification/index.d.ts +1 -0
  320. package/dist/verification/index.d.ts.map +1 -1
  321. package/dist/verification/index.js +1 -0
  322. package/dist/verification/index.js.map +1 -1
  323. package/dist/verification/presets.d.ts +53 -0
  324. package/dist/verification/presets.d.ts.map +1 -0
  325. package/dist/verification/presets.js +70 -0
  326. package/dist/verification/presets.js.map +1 -0
  327. package/dist/verification/presets.test.d.ts +16 -0
  328. package/dist/verification/presets.test.d.ts.map +1 -0
  329. package/dist/verification/presets.test.js +79 -0
  330. package/dist/verification/presets.test.js.map +1 -0
  331. package/dist/version.d.ts.map +1 -1
  332. package/dist/version.js +25 -4
  333. package/dist/version.js.map +1 -1
  334. package/package.json +3 -2
  335. package/src/advisory/executor.test.ts +7 -4
  336. package/src/advisory/executor.ts +11 -2
  337. package/src/agents/ReactiveAgent.ts +7 -0
  338. package/src/agents/SupervisorAgent.ts +63 -16
  339. package/src/agents/__tests__/supervisor-ledger.test.ts +185 -0
  340. package/src/bridge/sse/mapper.test.ts +2 -2
  341. package/src/config/runtime.ts +9 -0
  342. package/src/connector/builtins/webhook.test.ts +21 -0
  343. package/src/connector/index.ts +1 -0
  344. package/src/connector/mcp/client.ts +4 -0
  345. package/src/connector/mcp/index.ts +1 -0
  346. package/src/connector/mcp/streamable-http.test.ts +169 -0
  347. package/src/connector/mcp/streamable-http.ts +152 -0
  348. package/src/constants/compaction/index.ts +8 -3
  349. package/src/constants/sandbox/index.ts +37 -0
  350. package/src/constants/tools/index.ts +33 -2
  351. package/src/manager/run/persistence.ts +34 -6
  352. package/src/persona/assembler.ts +43 -8
  353. package/src/provider/collect.test.ts +2 -2
  354. package/src/provider/collect.ts +4 -1
  355. package/src/public-runtime.ts +15 -1
  356. package/src/public-tools.ts +2 -0
  357. package/src/public-types.ts +7 -0
  358. package/src/registry/index.ts +7 -0
  359. package/src/registry/tool/execute.test.ts +211 -7
  360. package/src/registry/tool/execute.ts +199 -11
  361. package/src/registry/toolset/catalog.test.ts +165 -0
  362. package/src/registry/toolset/catalog.ts +283 -0
  363. package/src/runtime/query/__tests__/deferred-tools.test.ts +230 -0
  364. package/src/runtime/query/__tests__/executor-concurrency.test.ts +122 -0
  365. package/src/runtime/query/__tests__/executor-plugin-hooks.test.ts +48 -3
  366. package/src/runtime/query/__tests__/long-document-flow.test.ts +180 -0
  367. package/src/runtime/query/__tests__/prompt.test.ts +55 -2
  368. package/src/runtime/query/__tests__/stream-recovery.test.ts +156 -0
  369. package/src/runtime/query/continuation.ts +16 -0
  370. package/src/runtime/query/executor.ts +82 -13
  371. package/src/runtime/query/index.ts +58 -5
  372. package/src/runtime/query/iteration/index.ts +361 -74
  373. package/src/runtime/query/iteration/phases/checkpoint.ts +7 -2
  374. package/src/runtime/query/iteration/phases/compaction.ts +89 -3
  375. package/src/runtime/query/iteration/phases/context.ts +68 -0
  376. package/src/runtime/query/iteration/phases/hitl-abort-race.test.ts +86 -0
  377. package/src/runtime/query/iteration/phases/hitl-answer-question.test.ts +139 -0
  378. package/src/runtime/query/iteration/phases/tool-review.ts +7 -3
  379. package/src/runtime/query/iteration/phases/working-memory-compaction.test.ts +215 -0
  380. package/src/runtime/query/iteration/phases/working-memory.ts +86 -0
  381. package/src/runtime/query/prompt.ts +17 -1
  382. package/src/runtime/query/tooling.ts +2 -0
  383. package/src/sandbox/provider/local.ts +33 -0
  384. package/src/session/workspace/__tests__/shared-run.test.ts +181 -0
  385. package/src/session/workspace/index.ts +6 -0
  386. package/src/session/workspace/shared-run.ts +317 -0
  387. package/src/skills/loader.test.ts +89 -0
  388. package/src/skills/loader.ts +37 -6
  389. package/src/streaming/coalesce.test.ts +4 -4
  390. package/src/tools/builtins/__tests__/edit.test.ts +57 -0
  391. package/src/tools/builtins/__tests__/payload-budget.test.ts +92 -0
  392. package/src/tools/builtins/__tests__/read-file.test.ts +43 -0
  393. package/src/tools/builtins/__tests__/verify-outputs.test.ts +71 -0
  394. package/src/tools/builtins/__tests__/write-file.test.ts +97 -0
  395. package/src/tools/builtins/bash.ts +48 -7
  396. package/src/tools/builtins/edit.ts +162 -27
  397. package/src/tools/builtins/index.ts +13 -13
  398. package/src/tools/builtins/read-file.ts +81 -10
  399. package/src/tools/builtins/search-tools.ts +42 -7
  400. package/src/tools/builtins/verify-outputs.ts +126 -0
  401. package/src/tools/builtins/write-file.ts +83 -14
  402. package/src/tools/coordinator/__tests__/agent.test.ts +172 -0
  403. package/src/tools/coordinator/__tests__/approve-plan.test.ts +139 -0
  404. package/src/tools/coordinator/__tests__/ask-user-question.test.ts +370 -0
  405. package/src/tools/coordinator/__tests__/task-list.test.ts +182 -0
  406. package/src/tools/coordinator/agent.ts +157 -0
  407. package/src/tools/coordinator/index.ts +355 -39
  408. package/src/types/agent/base.ts +8 -0
  409. package/src/types/agent/index.ts +1 -0
  410. package/src/types/agent/reactive.ts +42 -0
  411. package/src/types/agent/supervisor.ts +36 -0
  412. package/src/types/agent/working-memory.ts +29 -0
  413. package/src/types/common/index.ts +19 -0
  414. package/src/types/common/usage-merge.test.ts +53 -0
  415. package/src/types/connector/mcp.ts +14 -2
  416. package/src/types/hitl/index.ts +52 -1
  417. package/src/types/message/index.ts +32 -2
  418. package/src/types/provider/chat.ts +12 -9
  419. package/src/types/run/events.ts +6 -0
  420. package/src/types/sandbox/index.ts +219 -0
  421. package/src/types/skills/index.ts +4 -0
  422. package/src/types/tool/index.ts +24 -0
  423. package/src/types/toolset/index.ts +86 -0
  424. package/src/types/workspace/index.ts +9 -0
  425. package/src/types/workspace/shared-run.ts +65 -0
  426. package/src/verification/index.ts +1 -0
  427. package/src/verification/presets.test.ts +112 -0
  428. package/src/verification/presets.ts +72 -0
  429. package/src/version.ts +26 -4
@@ -0,0 +1,185 @@
1
+ import { mkdtemp, rm } from 'node:fs/promises';
2
+ import { tmpdir } from 'node:os';
3
+ import { join } from 'node:path';
4
+ import { afterEach, describe, expect, it } from 'vitest';
5
+ import { z } from 'zod';
6
+ import { ToolRegistry } from '../../../registry/tool/execute.js';
7
+ import { SearchToolsTool } from '../../../tools/builtins/search-tools.js';
8
+ import { createUserMessage } from '../../../types/message/index.js';
9
+ import { drainQuery } from '../index.js';
10
+ const ZERO_USAGE = {
11
+ promptTokens: 0,
12
+ completionTokens: 0,
13
+ totalTokens: 0,
14
+ cachedTokens: 0,
15
+ cacheWriteTokens: 0,
16
+ };
17
+ class CapturingProvider {
18
+ id = 'capturing';
19
+ name = 'Capturing Provider';
20
+ lastParams;
21
+ async *chatStream(params) {
22
+ this.lastParams = params;
23
+ yield {
24
+ id: 'msg_1',
25
+ delta: { content: 'done' },
26
+ };
27
+ yield {
28
+ id: 'msg_1',
29
+ delta: {},
30
+ finishReason: 'stop',
31
+ usage: ZERO_USAGE,
32
+ };
33
+ }
34
+ }
35
+ function registerDeferredDocumentTool(tools, name = 'generate_document') {
36
+ tools.register({
37
+ name,
38
+ description: 'Generate a project document by document id.',
39
+ inputSchema: z.object({
40
+ documentId: z.string(),
41
+ }),
42
+ execute: async () => ({ success: true, output: 'generated' }),
43
+ }, 'deferred');
44
+ }
45
+ describe('query deferred tool discovery', () => {
46
+ let workdirs = [];
47
+ afterEach(async () => {
48
+ await Promise.all(workdirs.map((dir) => rm(dir, { recursive: true, force: true })));
49
+ workdirs = [];
50
+ });
51
+ it('auto-exposes search_tools when deferred tools are registered', async () => {
52
+ const provider = new CapturingProvider();
53
+ const tools = new ToolRegistry();
54
+ registerDeferredDocumentTool(tools);
55
+ const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-deferred-tools-'));
56
+ workdirs.push(workingDirectory);
57
+ const run = await drainQuery({
58
+ provider,
59
+ tools,
60
+ runConfig: {
61
+ model: 'mock-model',
62
+ timeoutMs: 5_000,
63
+ tokenBudget: 100_000,
64
+ maxIterations: 1,
65
+ maxResponseTokens: 256,
66
+ },
67
+ agentId: 'agent_test',
68
+ agentName: 'Test Agent',
69
+ messages: [createUserMessage('what tools can you use?')],
70
+ workingDirectory,
71
+ sessionId: 'ses_deferred_tools',
72
+ threadId: 'thd_deferred_tools',
73
+ projectId: 'prj_deferred_tools',
74
+ tenantId: 'tnt_deferred_tools',
75
+ });
76
+ expect(run.status).toBe('completed');
77
+ expect(tools.has(SearchToolsTool.name)).toBe(true);
78
+ expect(tools.getAvailability(SearchToolsTool.name)).toBe('active');
79
+ expect(tools.getAvailability('generate_document')).toBe('deferred');
80
+ const toolNames = provider.lastParams?.tools?.map((tool) => tool.function.name).sort() ?? [];
81
+ expect(toolNames).toEqual([SearchToolsTool.name]);
82
+ const systemPrompt = (provider.lastParams?.messages ?? [])
83
+ .filter((message) => message.role === 'system')
84
+ .map((message) => message.content)
85
+ .join('\n');
86
+ expect(systemPrompt).toContain('Use search_tools to load these before use:');
87
+ expect(systemPrompt).toContain('- generate_document');
88
+ expect(systemPrompt).not.toContain('Deferred tools are discoverable but not executable until the runtime activates them');
89
+ });
90
+ it('keeps search_tools executable when allowedTools names a deferred tool', async () => {
91
+ const provider = new CapturingProvider();
92
+ const tools = new ToolRegistry();
93
+ registerDeferredDocumentTool(tools);
94
+ const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-deferred-tools-'));
95
+ workdirs.push(workingDirectory);
96
+ const run = await drainQuery({
97
+ provider,
98
+ tools,
99
+ allowedTools: ['generate_document'],
100
+ runConfig: {
101
+ model: 'mock-model',
102
+ timeoutMs: 5_000,
103
+ tokenBudget: 100_000,
104
+ maxIterations: 1,
105
+ maxResponseTokens: 256,
106
+ },
107
+ agentId: 'agent_test',
108
+ agentName: 'Test Agent',
109
+ messages: [createUserMessage('generate D-01')],
110
+ workingDirectory,
111
+ sessionId: 'ses_deferred_allowed_tools',
112
+ threadId: 'thd_deferred_allowed_tools',
113
+ projectId: 'prj_deferred_allowed_tools',
114
+ tenantId: 'tnt_deferred_allowed_tools',
115
+ });
116
+ expect(run.status).toBe('completed');
117
+ expect(tools.getAvailability('generate_document')).toBe('deferred');
118
+ const toolNames = provider.lastParams?.tools?.map((tool) => tool.function.name).sort() ?? [];
119
+ expect(toolNames).toEqual([SearchToolsTool.name]);
120
+ const systemPrompt = (provider.lastParams?.messages ?? [])
121
+ .filter((message) => message.role === 'system')
122
+ .map((message) => message.content)
123
+ .join('\n');
124
+ expect(systemPrompt).toContain('Use search_tools to load these before use:');
125
+ expect(systemPrompt).toContain('- generate_document');
126
+ });
127
+ it('does not let search_tools reveal or activate deferred tools outside allowedTools', async () => {
128
+ const tools = new ToolRegistry();
129
+ registerDeferredDocumentTool(tools);
130
+ registerDeferredDocumentTool(tools, 'dangerous_purge_document');
131
+ // 'dangerous' matches only the out-of-allowlist tool ('delete'-style
132
+ // CRUD verbs are stop tokens and never match anything by themselves).
133
+ const result = await SearchToolsTool.execute({ query: 'dangerous' }, {
134
+ runId: 'run_deferred_allowed_tools',
135
+ workingDirectory: '/tmp',
136
+ abortSignal: new AbortController().signal,
137
+ env: {},
138
+ log: () => undefined,
139
+ toolRegistry: tools,
140
+ allowedTools: ['generate_document', SearchToolsTool.name],
141
+ });
142
+ expect(result.success).toBe(true);
143
+ expect(result.output).toContain('No deferred tools matching "dangerous"');
144
+ expect(result.output).not.toContain('dangerous_purge_document');
145
+ expect(tools.getAvailability('generate_document')).toBe('deferred');
146
+ expect(tools.getAvailability('dangerous_purge_document')).toBe('deferred');
147
+ });
148
+ it('activates only the top-5 ranked matches and reports near-misses without activating', async () => {
149
+ const tools = new ToolRegistry();
150
+ // Eight deferred tools that all match "invoice" equally by name; the
151
+ // alphabetical tie-break makes the top-5 cut deterministic.
152
+ const names = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'].map((s) => `invoice_${s}`);
153
+ for (const name of names) {
154
+ tools.register({
155
+ name,
156
+ description: `Billing helper ${name.slice(-1)}.`,
157
+ inputSchema: z.object({ id: z.string() }),
158
+ execute: async () => ({ success: true, output: 'ok' }),
159
+ }, 'deferred');
160
+ }
161
+ const result = await SearchToolsTool.execute({ query: 'invoice' }, {
162
+ runId: 'run_deferred_top_k',
163
+ workingDirectory: '/tmp',
164
+ abortSignal: new AbortController().signal,
165
+ env: {},
166
+ log: () => undefined,
167
+ toolRegistry: tools,
168
+ });
169
+ expect(result.success).toBe(true);
170
+ expect(result.output).toContain('Activated 5 tool(s)');
171
+ expect(result.output).toContain('NOT loaded');
172
+ expect(result.data).toMatchObject({
173
+ activated: ['invoice_a', 'invoice_b', 'invoice_c', 'invoice_d', 'invoice_e'],
174
+ count: 5,
175
+ nearMisses: ['invoice_f', 'invoice_g', 'invoice_h'],
176
+ });
177
+ for (const name of ['invoice_a', 'invoice_b', 'invoice_c', 'invoice_d', 'invoice_e']) {
178
+ expect(tools.getAvailability(name)).toBe('active');
179
+ }
180
+ for (const name of ['invoice_f', 'invoice_g', 'invoice_h']) {
181
+ expect(tools.getAvailability(name)).toBe('deferred');
182
+ }
183
+ });
184
+ });
185
+ //# sourceMappingURL=deferred-tools.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"deferred-tools.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/deferred-tools.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACxD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAA;AAEzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAOnE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAED,MAAM,iBAAiB;IACb,EAAE,GAAG,WAAW,CAAA;IAChB,IAAI,GAAG,oBAAoB,CAAA;IACpC,UAAU,CAAuB;IAEjC,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QAC7C,IAAI,CAAC,UAAU,GAAG,MAAM,CAAA;QACxB,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;SAC1B,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,SAAS,4BAA4B,CAAC,KAAmB,EAAE,IAAI,GAAG,mBAAmB;IACpF,KAAK,CAAC,QAAQ,CACb;QACC,IAAI;QACJ,WAAW,EAAE,6CAA6C;QAC1D,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;SACtB,CAAC;QACF,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;KAC7D,EACD,UAAU,CACV,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,IAAI,QAAQ,GAAa,EAAE,CAAA;IAE3B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,GAAG,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QAEnC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,CAAC;YACxD,gBAAgB;YAChB,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;YAC1C,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;SAC1C,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAClD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAClE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEnE,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;QAC5F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAA;QAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CACjC,qFAAqF,CACrF,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QAEnC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,YAAY,EAAE,CAAC,mBAAmB,CAAC;YACnC,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;YAC9C,gBAAgB;YAChB,SAAS,EAAE,4BAAyC;YACpD,QAAQ,EAAE,4BAAwC;YAClD,SAAS,EAAE,4BAAyC;YACpD,QAAQ,EAAE,4BAAwC;SAClD,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEnE,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;QAC5F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAA;QAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QACnC,4BAA4B,CAAC,KAAK,EAAE,0BAA0B,CAAC,CAAA;QAE/D,qEAAqE;QACrE,sEAAsE;QACtE,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,WAAW,EAAE,EACtB;YACC,KAAK,EAAE,4BAAqC;YAC5C,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,GAAG,EAAE,CAAC,SAAS;YACpB,YAAY,EAAE,KAAK;YACnB,YAAY,EAAE,CAAC,mBAAmB,EAAE,eAAe,CAAC,IAAI,CAAC;SACzD,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wCAAwC,CAAC,CAAA;QACzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAA;QAC/D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACnE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;QACnG,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,qEAAqE;QACrE,4DAA4D;QAC5D,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAA;QACjF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,KAAK,CAAC,QAAQ,CACb;gBACC,IAAI;gBACJ,WAAW,EAAE,kBAAkB,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG;gBAChD,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACzC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aACtD,EACD,UAAU,CACV,CAAA;QACF,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,SAAS,EAAE,EACpB;YACC,KAAK,EAAE,oBAA6B;YACpC,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,GAAG,EAAE,CAAC,SAAS;YACpB,YAAY,EAAE,KAAK;SACnB,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QACtD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAA;QAC7C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;YACjC,SAAS,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;YAC5E,KAAK,EAAE,CAAC;YACR,UAAU,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;SACnD,CAAC,CAAA;QACF,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACnD,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACrD,CAAC;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=executor-concurrency.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"executor-concurrency.test.d.ts","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/executor-concurrency.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,98 @@
1
+ import { beforeEach, describe, expect, it, vi } from 'vitest';
2
+ import { ActivityStore } from '../../../store/activity/memory.js';
3
+ import { ToolExecutor } from '../executor.js';
4
+ const mockRunId = 'run_test';
5
+ function makeLogger() {
6
+ const stub = { info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() };
7
+ return { ...stub, child: vi.fn(() => ({ ...stub, child: vi.fn() })) };
8
+ }
9
+ const delay = (ms) => new Promise((r) => setTimeout(r, ms));
10
+ /** Two tool calls in one assistant message (one batch). */
11
+ function twoCallResponse(name, a, b) {
12
+ return {
13
+ message: {
14
+ role: 'assistant',
15
+ content: null,
16
+ toolCalls: [
17
+ { id: 'c1', type: 'function', function: { name, arguments: JSON.stringify(a) } },
18
+ { id: 'c2', type: 'function', function: { name, arguments: JSON.stringify(b) } },
19
+ ],
20
+ },
21
+ finishReason: 'tool_calls',
22
+ usage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },
23
+ };
24
+ }
25
+ describe('ToolExecutor — concurrencySafe batching', () => {
26
+ let activityStore;
27
+ let emitEvent;
28
+ beforeEach(() => {
29
+ activityStore = new ActivityStore(mockRunId, {
30
+ enabled: true,
31
+ trackToolCalls: true,
32
+ trackLlmTurns: true,
33
+ });
34
+ emitEvent = async () => { };
35
+ });
36
+ it('serializes concurrency-unsafe tools so read-modify-write does not race', async () => {
37
+ // Shared mutable state, mutated via read → await → write (like `edit`).
38
+ let file = 'A';
39
+ const execute = vi.fn(async (_name, input) => {
40
+ const current = file; // read
41
+ await delay(10); // window a parallel run would exploit
42
+ file = current + input.add; // write
43
+ return { success: true, output: 'ok' };
44
+ });
45
+ const tools = {
46
+ register: vi.fn(),
47
+ unregister: vi.fn(),
48
+ execute,
49
+ // edit/write/bash declare concurrencySafe:false → isConcurrencySafe()=>false
50
+ get: vi.fn(() => ({ isConcurrencySafe: () => false })),
51
+ has: vi.fn(() => true),
52
+ listNames: vi.fn(() => []),
53
+ getAvailability: vi.fn(),
54
+ };
55
+ const exec = new ToolExecutor({
56
+ tools,
57
+ runId: mockRunId,
58
+ workingDirectory: '/tmp',
59
+ permissionMode: 'auto',
60
+ env: {},
61
+ abortSignal: new AbortController().signal,
62
+ }, activityStore, emitEvent, makeLogger());
63
+ await exec.executeBatch(twoCallResponse('edit', { add: 'B' }, { add: 'C' }));
64
+ // Serialized: A→AB→ABC. A racing run would lose one append (e.g. 'AC').
65
+ expect(file).toBe('ABC');
66
+ });
67
+ it('runs concurrency-safe tools in parallel', async () => {
68
+ let active = 0;
69
+ let maxActive = 0;
70
+ const execute = vi.fn(async () => {
71
+ active++;
72
+ maxActive = Math.max(maxActive, active);
73
+ await delay(10);
74
+ active--;
75
+ return { success: true, output: 'ok' };
76
+ });
77
+ const tools = {
78
+ register: vi.fn(),
79
+ unregister: vi.fn(),
80
+ execute,
81
+ get: vi.fn(() => ({ isConcurrencySafe: () => true })),
82
+ has: vi.fn(() => true),
83
+ listNames: vi.fn(() => []),
84
+ getAvailability: vi.fn(),
85
+ };
86
+ const exec = new ToolExecutor({
87
+ tools,
88
+ runId: mockRunId,
89
+ workingDirectory: '/tmp',
90
+ permissionMode: 'auto',
91
+ env: {},
92
+ abortSignal: new AbortController().signal,
93
+ }, activityStore, emitEvent, makeLogger());
94
+ await exec.executeBatch(twoCallResponse('grep', { p: '1' }, { p: '2' }));
95
+ expect(maxActive).toBe(2); // both ran at once
96
+ });
97
+ });
98
+ //# sourceMappingURL=executor-concurrency.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"executor-concurrency.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/executor-concurrency.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAA;AAMjE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAE7C,MAAM,SAAS,GAAG,UAAmB,CAAA;AAErC,SAAS,UAAU;IAClB,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAA;IAC7E,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAuB,CAAA;AAC3F,CAAC;AAED,MAAM,KAAK,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAA;AAEnE,2DAA2D;AAC3D,SAAS,eAAe,CAAC,IAAY,EAAE,CAAS,EAAE,CAAS;IAC1D,OAAO;QACN,OAAO,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI;YACb,SAAS,EAAE;gBACV,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChF,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE;aAChF;SACD;QACD,YAAY,EAAE,YAAY;QAC1B,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;KACrC,CAAA;AAC5B,CAAC;AAED,QAAQ,CAAC,yCAAyC,EAAE,GAAG,EAAE;IACxD,IAAI,aAA4B,CAAA;IAChC,IAAI,SAAyC,CAAA;IAE7C,UAAU,CAAC,GAAG,EAAE;QACf,aAAa,GAAG,IAAI,aAAa,CAAC,SAAS,EAAE;YAC5C,OAAO,EAAE,IAAI;YACb,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACnB,CAAC,CAAA;QACF,SAAS,GAAG,KAAK,IAAI,EAAE,GAAE,CAAC,CAAA;IAC3B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACvF,wEAAwE;QACxE,IAAI,IAAI,GAAG,GAAG,CAAA;QACd,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAa,EAAE,KAAc,EAAE,EAAE;YAC7D,MAAM,OAAO,GAAG,IAAI,CAAA,CAAC,OAAO;YAC5B,MAAM,KAAK,CAAC,EAAE,CAAC,CAAA,CAAC,sCAAsC;YACtD,IAAI,GAAG,OAAO,GAAI,KAAyB,CAAC,GAAG,CAAA,CAAC,QAAQ;YACxD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACvC,CAAC,CAAC,CAAA;QACF,MAAM,KAAK,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,OAAO;YACP,6EAA6E;YAC7E,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;YACtD,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;YACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACW,CAAA;QAEpC,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;QAC5E,wEAAwE;QACxE,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACzB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACxD,IAAI,MAAM,GAAG,CAAC,CAAA;QACd,IAAI,SAAS,GAAG,CAAC,CAAA;QACjB,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YAChC,MAAM,EAAE,CAAA;YACR,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;YACvC,MAAM,KAAK,CAAC,EAAE,CAAC,CAAA;YACf,MAAM,EAAE,CAAA;YACR,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACvC,CAAC,CAAC,CAAA;QACF,MAAM,KAAK,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,OAAO;YACP,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YACrD,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;YACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACW,CAAA;QAEpC,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAA;QACxE,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA,CAAC,mBAAmB;IAC9C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
@@ -9,7 +9,10 @@ function makeLogger() {
9
9
  error: vi.fn(),
10
10
  debug: vi.fn(),
11
11
  };
12
- return { ...stub, child: vi.fn(() => ({ ...stub, child: vi.fn() })) };
12
+ return {
13
+ ...stub,
14
+ child: vi.fn(() => ({ ...stub, child: vi.fn() })),
15
+ };
13
16
  }
14
17
  function makeToolRegistry(execute) {
15
18
  return {
@@ -70,6 +73,32 @@ describe('ToolExecutor plugin hooks', () => {
70
73
  const batch = await exec.executeBatch(buildResponse('echo', { msg: 'hi' }));
71
74
  expect(batch.results[0]?.output).toBe('ok');
72
75
  });
76
+ it('preserves tool stdout/stderr when a tool exits unsuccessfully', async () => {
77
+ const tools = makeToolRegistry(vi.fn(async () => ({
78
+ success: false,
79
+ output: 'STDOUT:\npartial result\n\nSTDERR:\nboom',
80
+ error: 'Command exited with code 1',
81
+ })));
82
+ const exec = new ToolExecutor({
83
+ tools,
84
+ runId: mockRunId,
85
+ workingDirectory: '/tmp',
86
+ permissionMode: 'auto',
87
+ env: {},
88
+ abortSignal: new AbortController().signal,
89
+ }, activityStore, emitEvent, makeLogger());
90
+ const batch = await exec.executeBatch(buildResponse('bash', { command: 'false' }));
91
+ expect(batch.results[0]?.output).toContain('STDOUT:\npartial result');
92
+ expect(batch.results[0]?.output).toContain('STDERR:\nboom');
93
+ expect(batch.results[0]?.output).toContain('Error: Command exited with code 1');
94
+ const completed = emitted.find((e) => e.type === 'tool_completed');
95
+ expect(completed).toMatchObject({
96
+ type: 'tool_completed',
97
+ toolName: 'bash',
98
+ result: expect.stringContaining('STDOUT:\npartial result'),
99
+ isError: true,
100
+ });
101
+ });
73
102
  it('replaces input on pre_tool_use modify', async () => {
74
103
  const executeMock = vi.fn(async () => ({ success: true, output: 'ok' }));
75
104
  const tools = makeToolRegistry(executeMock);
@@ -89,7 +118,10 @@ describe('ToolExecutor plugin hooks', () => {
89
118
  expect(executeMock).toHaveBeenCalledWith('echo', { replaced: true }, expect.any(Object));
90
119
  });
91
120
  it('skips registry execution and synthesizes output on pre_tool_use skip', async () => {
92
- const executeMock = vi.fn(async () => ({ success: true, output: 'should-not-run' }));
121
+ const executeMock = vi.fn(async () => ({
122
+ success: true,
123
+ output: 'should-not-run',
124
+ }));
93
125
  const tools = makeToolRegistry(executeMock);
94
126
  const pluginManager = makePluginManager(async (event) => event === 'pre_tool_use'
95
127
  ? [{ action: 'skip', reason: 'policy' }]
@@ -169,7 +201,10 @@ describe('ToolExecutor plugin hooks', () => {
169
201
  ]);
170
202
  });
171
203
  it('carries modified input into synthetic skip outcome (modify -> skip chain)', async () => {
172
- const executeMock = vi.fn(async () => ({ success: true, output: 'should-not-run' }));
204
+ const executeMock = vi.fn(async () => ({
205
+ success: true,
206
+ output: 'should-not-run',
207
+ }));
173
208
  const tools = makeToolRegistry(executeMock);
174
209
  const { PluginLifecycleManager } = await import('../../../plugin/lifecycle.js');
175
210
  const realManager = new PluginLifecycleManager({
@@ -1 +1 @@
1
- {"version":3,"file":"executor-plugin-hooks.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/executor-plugin-hooks.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE7D,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAA;AAOjE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAE7C,MAAM,SAAS,GAAG,UAAmB,CAAA;AAErC,SAAS,UAAU;IAClB,MAAM,IAAI,GAAG;QACZ,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACd,CAAA;IACD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAuB,CAAA;AAC3F,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAwC;IACjE,OAAO;QACN,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,OAAO;QACP,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAC3B,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;QACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;KACW,CAAA;AACrC,CAAC;AAED,SAAS,iBAAiB,CACzB,YAAoD;IAEpD,OAAO,EAAE,YAAY,EAAuC,CAAA;AAC7D,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,IAAY;IACpD,OAAO;QACN,OAAO,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI;YACb,SAAS,EAAE;gBACV;oBACC,EAAE,EAAE,QAAQ;oBACZ,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;iBAC7D;aACD;SACD;QACD,YAAY,EAAE,YAAY;QAC1B,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;KACrC,CAAA;AAC5B,CAAC;AAED,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IAC1C,IAAI,aAA4B,CAAA;IAChC,IAAI,OAAmB,CAAA;IACvB,IAAI,SAAyC,CAAA;IAE7C,UAAU,CAAC,GAAG,EAAE;QACf,aAAa,GAAG,IAAI,aAAa,CAAC,SAAS,EAAE;YAC5C,OAAO,EAAE,IAAI;YACb,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACnB,CAAC,CAAA;QACF,OAAO,GAAG,EAAE,CAAA;QACZ,SAAS,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE;YACvB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAChB,CAAC,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAC3E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QACxE,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc;YACvB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAwB;YAC3E,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAClE,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAA;IACzF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAA;QACpF,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc;YACvB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAwB;YAChE,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAA;IAC7E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAA;QACrF,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,eAAe;YACxB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAwB;YACpE,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;IACzD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,UAAU,GAAc,EAAE,CAAA;QAChC,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAChD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACtB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACvC,CAAC,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,QAAQ,GAAG;YAChB,KAAK,EAAE,GAAQ,EAAE,EAAE;gBAClB,UAAU,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC7C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAsB,CAAA;YACpE,CAAC;YACD,KAAK,EAAE,GAAQ,EAAE,EAAE;gBAClB,UAAU,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC7C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAsB,CAAA;YACpE,CAAC;SACD,CAAA;QACD,oFAAoF;QACpF,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC;YAC9C,cAAc,EAAE,EAAS;YACzB,YAAY,EAAE,EAAS;YACvB,GAAG,EAAE,UAAU,EAAE;SACjB,CAAC,CAAA;QACF,WAAW,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE;YAC/C,EAAE,QAAQ,EAAE,IAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAQ,EAAE;YACtD,EAAE,QAAQ,EAAE,IAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAQ,EAAE;SACtD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa,EAAE,WAAW;SAC1B,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAClE,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAC1B,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1B,EAAE,IAAI,EAAE,CAAC,EAAE;SACX,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAA;QACpF,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC;YAC9C,cAAc,EAAE,EAAS;YACzB,YAAY,EAAE,EAAS;YACvB,GAAG,EAAE,UAAU,EAAE;SACjB,CAAC,CAAA;QACF,MAAM,UAAU,GAAc,EAAE,CAAA;QAChC,WAAW,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE;YAC/C;gBACC,QAAQ,EAAE,IAAW;gBACrB,OAAO,EAAE,CAAC,KAAK,EAAE,GAAQ,EAAE,EAAE;oBAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;oBACzC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAA;gBACtD,CAAC,CAAQ;aACT;YACD;gBACC,QAAQ,EAAE,IAAW;gBACrB,OAAO,EAAE,CAAC,KAAK,EAAE,GAAQ,EAAE,EAAE;oBAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;oBACzC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,CAAA;gBACjD,CAAC,CAAQ;aACT;SACD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa,EAAE,WAAW;SAC1B,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAChF,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAA;QACjF,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAC1B,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YAC7B,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;SAC5B,CAAC,CAAA;QACF,oFAAoF;QACpF,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAA;QACvE,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,MAAM;YAChB,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACxB,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc,CAAC,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAwB,CAAC,CAAC,CAAC,EAAE,CAC7E,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzE,4BAA4B,CAC5B,CAAA;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
1
+ {"version":3,"file":"executor-plugin-hooks.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/executor-plugin-hooks.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE7D,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAA;AAOjE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAE7C,MAAM,SAAS,GAAG,UAAmB,CAAA;AAErC,SAAS,UAAU;IAClB,MAAM,IAAI,GAAG;QACZ,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACd,CAAA;IACD,OAAO;QACN,GAAG,IAAI;QACP,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;KAC5B,CAAA;AACvB,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAwC;IACjE,OAAO;QACN,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,OAAO;QACP,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAC3B,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;QACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;KACW,CAAA;AACrC,CAAC;AAED,SAAS,iBAAiB,CACzB,YAAoD;IAEpD,OAAO,EAAE,YAAY,EAAuC,CAAA;AAC7D,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,IAAY;IACpD,OAAO;QACN,OAAO,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI;YACb,SAAS,EAAE;gBACV;oBACC,EAAE,EAAE,QAAQ;oBACZ,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;iBAC7D;aACD;SACD;QACD,YAAY,EAAE,YAAY;QAC1B,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;KACrC,CAAA;AAC5B,CAAC;AAED,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IAC1C,IAAI,aAA4B,CAAA;IAChC,IAAI,OAAmB,CAAA;IACvB,IAAI,SAAyC,CAAA;IAE7C,UAAU,CAAC,GAAG,EAAE;QACf,aAAa,GAAG,IAAI,aAAa,CAAC,SAAS,EAAE;YAC5C,OAAO,EAAE,IAAI;YACb,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;SACnB,CAAC,CAAA;QACF,OAAO,GAAG,EAAE,CAAA;QACZ,SAAS,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE;YACvB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAChB,CAAC,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAC3E,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,KAAK,GAAG,gBAAgB,CAC7B,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YAClB,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,0CAA0C;YAClD,KAAK,EAAE,4BAA4B;SACnC,CAAC,CAAC,CACH,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAA;QAClF,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAA;QACrE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAA;QAC3D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAA;QAE/E,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAA;QAClE,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,CAAC;YAC/B,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,MAAM;YAChB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,yBAAyB,CAAC;YAC1D,OAAO,EAAE,IAAI;SACb,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QACxE,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc;YACvB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAwB;YAC3E,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAClE,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAA;IACzF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACtC,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,gBAAgB;SACxB,CAAC,CAAC,CAAA;QACH,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc;YACvB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAwB;YAChE,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAA;IAC7E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAA;QACrF,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,eAAe;YACxB,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAwB;YACpE,CAAC,CAAC,EAAE,CACL,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAA;IACzD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,UAAU,GAAc,EAAE,CAAA;QAChC,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YAChD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACtB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACvC,CAAC,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,QAAQ,GAAG;YAChB,KAAK,EAAE,GAAQ,EAAE,EAAE;gBAClB,UAAU,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC7C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAsB,CAAA;YACpE,CAAC;YACD,KAAK,EAAE,GAAQ,EAAE,EAAE;gBAClB,UAAU,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC7C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAsB,CAAA;YACpE,CAAC;SACD,CAAA;QACD,oFAAoF;QACpF,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC;YAC9C,cAAc,EAAE,EAAS;YACzB,YAAY,EAAE,EAAS;YACvB,GAAG,EAAE,UAAU,EAAE;SACjB,CAAC,CAAA;QACF,WAAW,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE;YAC/C,EAAE,QAAQ,EAAE,IAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAQ,EAAE;YACtD,EAAE,QAAQ,EAAE,IAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAQ,EAAE;SACtD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa,EAAE,WAAW;SAC1B,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAClE,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAC1B,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1B,EAAE,IAAI,EAAE,CAAC,EAAE;SACX,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACtC,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,gBAAgB;SACxB,CAAC,CAAC,CAAA;QACH,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAA;QAC3C,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAA;QAC/E,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC;YAC9C,cAAc,EAAE,EAAS;YACzB,YAAY,EAAE,EAAS;YACvB,GAAG,EAAE,UAAU,EAAE;SACjB,CAAC,CAAA;QACF,MAAM,UAAU,GAAc,EAAE,CAAA;QAChC,WAAW,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE;YAC/C;gBACC,QAAQ,EAAE,IAAW;gBACrB,OAAO,EAAE,CAAC,KAAK,EAAE,GAAQ,EAAE,EAAE;oBAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;oBACzC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAA;gBACtD,CAAC,CAAQ;aACT;YACD;gBACC,QAAQ,EAAE,IAAW;gBACrB,OAAO,EAAE,CAAC,KAAK,EAAE,GAAQ,EAAE,EAAE;oBAC5B,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;oBACzC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,CAAA;gBACjD,CAAC,CAAQ;aACT;SACD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa,EAAE,WAAW;SAC1B,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAChF,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAA;QACjF,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YAC1B,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;YAC7B,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;SAC5B,CAAC,CAAA;QACF,oFAAoF;QACpF,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAA;QACvE,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,CAAC;YACpC,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,MAAM;YAChB,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACxB,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,MAAM,aAAa,GAAG,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvD,KAAK,KAAK,cAAc,CAAC,CAAC,CAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAwB,CAAC,CAAC,CAAC,EAAE,CAC7E,CAAA;QACD,MAAM,IAAI,GAAG,IAAI,YAAY,CAC5B;YACC,KAAK;YACL,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE,MAAM;YACxB,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,aAAa;SACb,EACD,aAAa,EACb,SAAS,EACT,UAAU,EAAE,CACZ,CAAA;QACD,MAAM,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzE,4BAA4B,CAC5B,CAAA;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=long-document-flow.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"long-document-flow.test.d.ts","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/long-document-flow.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,154 @@
1
+ import { readFile, rm } from 'node:fs/promises';
2
+ import { mkdtemp } from 'node:fs/promises';
3
+ import { tmpdir } from 'node:os';
4
+ import { join } from 'node:path';
5
+ import { afterEach, describe, expect, it } from 'vitest';
6
+ import { ToolRegistry } from '../../../registry/tool/execute.js';
7
+ import { EditTool, WriteFileTool } from '../../../tools/builtins/index.js';
8
+ import { createUserMessage } from '../../../types/message/index.js';
9
+ import { drainQuery } from '../index.js';
10
+ const ZERO_USAGE = {
11
+ promptTokens: 0,
12
+ completionTokens: 0,
13
+ totalTokens: 0,
14
+ cachedTokens: 0,
15
+ cacheWriteTokens: 0,
16
+ };
17
+ class ScriptedLongDocumentProvider {
18
+ steps;
19
+ id = 'scripted-long-document';
20
+ name = 'Scripted Long Document Provider';
21
+ calls = 0;
22
+ constructor(steps) {
23
+ this.steps = steps;
24
+ }
25
+ async *chatStream() {
26
+ const step = this.steps[this.calls];
27
+ this.calls += 1;
28
+ if (!step) {
29
+ yield {
30
+ id: 'msg_done',
31
+ delta: { content: 'Long document created and verified.' },
32
+ finishReason: 'stop',
33
+ usage: ZERO_USAGE,
34
+ };
35
+ return;
36
+ }
37
+ yield {
38
+ id: `msg_${this.calls}`,
39
+ delta: {
40
+ toolCalls: [
41
+ {
42
+ index: 0,
43
+ id: `toolu_${this.calls}`,
44
+ type: 'function',
45
+ function: { name: step.name },
46
+ },
47
+ ],
48
+ },
49
+ };
50
+ yield {
51
+ id: `msg_${this.calls}`,
52
+ delta: {
53
+ toolCalls: [
54
+ {
55
+ index: 0,
56
+ id: `toolu_${this.calls}`,
57
+ function: { arguments: JSON.stringify(step.input) },
58
+ },
59
+ ],
60
+ },
61
+ };
62
+ yield {
63
+ id: `msg_${this.calls}`,
64
+ delta: {},
65
+ finishReason: 'tool_calls',
66
+ usage: ZERO_USAGE,
67
+ };
68
+ }
69
+ }
70
+ describe('query long-document tool flow', () => {
71
+ const workdirs = [];
72
+ afterEach(async () => {
73
+ await Promise.all(workdirs.map((dir) => rm(dir, { recursive: true, force: true })));
74
+ workdirs.length = 0;
75
+ });
76
+ it('executes the canonical write skeleton plus bounded edit chunks flow end to end', async () => {
77
+ const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-query-long-doc-'));
78
+ workdirs.push(workingDirectory);
79
+ const chunks = Array.from({ length: 4 }, (_, sectionIndex) => {
80
+ const lines = Array.from({ length: 40 }, (_, lineIndex) => {
81
+ return `Section ${sectionIndex + 1}.${lineIndex + 1}: this content is intentionally long enough to require repeated bounded edit calls.`;
82
+ });
83
+ const chunk = [`## Section ${sectionIndex + 1}`, ...lines, ''].join('\n');
84
+ expect(chunk.length).toBeLessThan(12_000);
85
+ return chunk;
86
+ });
87
+ const provider = new ScriptedLongDocumentProvider([
88
+ {
89
+ name: 'write',
90
+ input: {
91
+ path: 'outputs/long-document-flow.md',
92
+ content: '# Long document flow\n\n{{BODY}}\n',
93
+ },
94
+ },
95
+ {
96
+ name: 'edit',
97
+ input: {
98
+ path: 'outputs/long-document-flow.md',
99
+ oldStr: '{{BODY}}',
100
+ newStr: chunks[0],
101
+ replace_all: false,
102
+ },
103
+ },
104
+ ...chunks.slice(1).map((chunk) => ({
105
+ name: 'edit',
106
+ input: {
107
+ path: 'outputs/long-document-flow.md',
108
+ insertLine: 'end',
109
+ newStr: chunk,
110
+ replace_all: false,
111
+ },
112
+ })),
113
+ ]);
114
+ const tools = new ToolRegistry();
115
+ tools.register(WriteFileTool);
116
+ tools.register(EditTool);
117
+ const events = [];
118
+ const run = await drainQuery({
119
+ provider,
120
+ tools,
121
+ runConfig: {
122
+ model: 'mock-model',
123
+ timeoutMs: 5_000,
124
+ tokenBudget: 100_000,
125
+ maxIterations: 8,
126
+ maxResponseTokens: 256,
127
+ },
128
+ agentId: 'agent_long_doc',
129
+ agentName: 'Long Document Agent',
130
+ messages: [createUserMessage('create a long document')],
131
+ workingDirectory,
132
+ sessionId: 'ses_long_document_flow',
133
+ threadId: 'thd_long_document_flow',
134
+ projectId: 'prj_long_document_flow',
135
+ tenantId: 'tnt_long_document_flow',
136
+ }, (event) => {
137
+ events.push(event);
138
+ });
139
+ const final = await readFile(join(workingDirectory, 'outputs/long-document-flow.md'), 'utf-8');
140
+ const executingTools = events
141
+ .filter((event) => event.type === 'tool_executing')
142
+ .map((event) => event.toolName);
143
+ expect(run.status).toBe('completed');
144
+ expect(run.result).toBe('Long document created and verified.');
145
+ expect(provider.calls).toBe(6);
146
+ expect(executingTools).toEqual(['write', 'edit', 'edit', 'edit', 'edit']);
147
+ expect(final).not.toContain('{{BODY}}');
148
+ expect(final.split('\n').length).toBeGreaterThan(160);
149
+ expect(final).toContain('## Section 1');
150
+ expect(final).toContain('## Section 4');
151
+ expect(final).toContain('Section 4.40');
152
+ });
153
+ });
154
+ //# sourceMappingURL=long-document-flow.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"long-document-flow.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/long-document-flow.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAA;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAExD,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAChE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAA;AAE1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAInE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAOD,MAAM,4BAA4B;IAKJ;IAJpB,EAAE,GAAG,wBAAwB,CAAA;IAC7B,IAAI,GAAG,iCAAiC,CAAA;IACjD,KAAK,GAAG,CAAC,CAAA;IAET,YAA6B,KAA0B;QAA1B,UAAK,GAAL,KAAK,CAAqB;IAAG,CAAC;IAE3D,KAAK,CAAC,CAAC,UAAU;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnC,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QAEf,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,MAAM;gBACL,EAAE,EAAE,UAAU;gBACd,KAAK,EAAE,EAAE,OAAO,EAAE,qCAAqC,EAAE;gBACzD,YAAY,EAAE,MAAM;gBACpB,KAAK,EAAE,UAAU;aACjB,CAAA;YACD,OAAM;QACP,CAAC;QAED,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE;gBACN,SAAS,EAAE;oBACV;wBACC,KAAK,EAAE,CAAC;wBACR,EAAE,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE;wBACzB,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;qBAC7B;iBACD;aACD;SACD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE;gBACN,SAAS,EAAE;oBACV;wBACC,KAAK,EAAE,CAAC;wBACR,EAAE,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE;wBACzB,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;qBACnD;iBACD;aACD;SACD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,YAAY;YAC1B,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,MAAM,QAAQ,GAAa,EAAE,CAAA;IAE7B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,EAAE;YAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE;gBACzD,OAAO,WAAW,YAAY,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,qFAAqF,CAAA;YACzI,CAAC,CAAC,CAAA;YACF,MAAM,KAAK,GAAG,CAAC,cAAc,YAAY,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACzE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;YACzC,OAAO,KAAK,CAAA;QACb,CAAC,CAAC,CAAA;QAEF,MAAM,QAAQ,GAAG,IAAI,4BAA4B,CAAC;YACjD;gBACC,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,OAAO,EAAE,oCAAoC;iBAC7C;aACD;YACD;gBACC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,MAAM,EAAE,UAAU;oBAClB,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;oBACjB,WAAW,EAAE,KAAK;iBAClB;aACD;YACD,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAClC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,UAAU,EAAE,KAAK;oBACjB,MAAM,EAAE,KAAK;oBACb,WAAW,EAAE,KAAK;iBAClB;aACD,CAAC,CAAC;SACH,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAA;QAC7B,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;QACxB,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,gBAAgB;YACzB,SAAS,EAAE,qBAAqB;YAChC,QAAQ,EAAE,CAAC,iBAAiB,CAAC,wBAAwB,CAAC,CAAC;YACvD,gBAAgB;YAChB,SAAS,EAAE,wBAAqC;YAChD,QAAQ,EAAE,wBAAoC;YAC9C,SAAS,EAAE,wBAAqC;YAChD,QAAQ,EAAE,wBAAoC;SAC9C,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,gBAAgB,EAAE,+BAA+B,CAAC,EAAE,OAAO,CAAC,CAAA;QAC9F,MAAM,cAAc,GAAG,MAAM;aAC3B,MAAM,CACN,CAAC,KAAK,EAA0D,EAAE,CACjE,KAAK,CAAC,IAAI,KAAK,gBAAgB,CAChC;aACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;QAEhC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAA;QAC9D,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC9B,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAA;QACzE,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAA;QACrD,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}