@namzu/sdk 1.0.0 → 1.2.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 (330) hide show
  1. package/CHANGELOG.md +253 -0
  2. package/dist/agents/ReactiveAgent.d.ts.map +1 -1
  3. package/dist/agents/ReactiveAgent.js +5 -0
  4. package/dist/agents/ReactiveAgent.js.map +1 -1
  5. package/dist/agents/SupervisorAgent.d.ts +22 -1
  6. package/dist/agents/SupervisorAgent.d.ts.map +1 -1
  7. package/dist/agents/SupervisorAgent.js +47 -14
  8. package/dist/agents/SupervisorAgent.js.map +1 -1
  9. package/dist/agents/__tests__/supervisor-ledger.test.d.ts +2 -0
  10. package/dist/agents/__tests__/supervisor-ledger.test.d.ts.map +1 -0
  11. package/dist/agents/__tests__/supervisor-ledger.test.js +131 -0
  12. package/dist/agents/__tests__/supervisor-ledger.test.js.map +1 -0
  13. package/dist/bridge/a2a/mapper.d.ts.map +1 -1
  14. package/dist/bridge/a2a/mapper.js +2 -0
  15. package/dist/bridge/a2a/mapper.js.map +1 -1
  16. package/dist/bridge/sse/mapper.d.ts.map +1 -1
  17. package/dist/bridge/sse/mapper.js +4 -0
  18. package/dist/bridge/sse/mapper.js.map +1 -1
  19. package/dist/config/runtime.d.ts +24 -0
  20. package/dist/config/runtime.d.ts.map +1 -1
  21. package/dist/config/runtime.js +9 -0
  22. package/dist/config/runtime.js.map +1 -1
  23. package/dist/connector/builtins/webhook.test.js +17 -0
  24. package/dist/connector/builtins/webhook.test.js.map +1 -1
  25. package/dist/connector/index.d.ts +1 -0
  26. package/dist/connector/index.d.ts.map +1 -1
  27. package/dist/connector/index.js +1 -0
  28. package/dist/connector/index.js.map +1 -1
  29. package/dist/connector/mcp/client.d.ts.map +1 -1
  30. package/dist/connector/mcp/client.js +4 -0
  31. package/dist/connector/mcp/client.js.map +1 -1
  32. package/dist/connector/mcp/index.d.ts +1 -0
  33. package/dist/connector/mcp/index.d.ts.map +1 -1
  34. package/dist/connector/mcp/index.js +1 -0
  35. package/dist/connector/mcp/index.js.map +1 -1
  36. package/dist/connector/mcp/streamable-http.d.ts +22 -0
  37. package/dist/connector/mcp/streamable-http.d.ts.map +1 -0
  38. package/dist/connector/mcp/streamable-http.js +128 -0
  39. package/dist/connector/mcp/streamable-http.js.map +1 -0
  40. package/dist/connector/mcp/streamable-http.test.d.ts +2 -0
  41. package/dist/connector/mcp/streamable-http.test.d.ts.map +1 -0
  42. package/dist/connector/mcp/streamable-http.test.js +132 -0
  43. package/dist/connector/mcp/streamable-http.test.js.map +1 -0
  44. package/dist/{runtime/query → constants}/continuation.d.ts +5 -0
  45. package/dist/constants/continuation.d.ts.map +1 -0
  46. package/dist/{runtime/query → constants}/continuation.js +5 -0
  47. package/dist/constants/continuation.js.map +1 -0
  48. package/dist/constants/index.d.ts +1 -0
  49. package/dist/constants/index.d.ts.map +1 -1
  50. package/dist/constants/index.js +1 -0
  51. package/dist/constants/index.js.map +1 -1
  52. package/dist/manager/run/persistence.d.ts +11 -0
  53. package/dist/manager/run/persistence.d.ts.map +1 -1
  54. package/dist/manager/run/persistence.js +30 -1
  55. package/dist/manager/run/persistence.js.map +1 -1
  56. package/dist/persona/assembler.d.ts.map +1 -1
  57. package/dist/persona/assembler.js +18 -9
  58. package/dist/persona/assembler.js.map +1 -1
  59. package/dist/provider/capabilities.d.ts +27 -0
  60. package/dist/provider/capabilities.d.ts.map +1 -0
  61. package/dist/provider/capabilities.js +29 -0
  62. package/dist/provider/capabilities.js.map +1 -0
  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/index.d.ts +2 -0
  67. package/dist/provider/index.d.ts.map +1 -1
  68. package/dist/provider/index.js +1 -0
  69. package/dist/provider/index.js.map +1 -1
  70. package/dist/public-runtime.d.ts +4 -3
  71. package/dist/public-runtime.d.ts.map +1 -1
  72. package/dist/public-runtime.js +3 -3
  73. package/dist/public-runtime.js.map +1 -1
  74. package/dist/registry/tool/execute.d.ts +27 -0
  75. package/dist/registry/tool/execute.d.ts.map +1 -1
  76. package/dist/registry/tool/execute.js +124 -21
  77. package/dist/registry/tool/execute.js.map +1 -1
  78. package/dist/registry/tool/execute.test.d.ts +9 -6
  79. package/dist/registry/tool/execute.test.d.ts.map +1 -1
  80. package/dist/registry/tool/execute.test.js +72 -10
  81. package/dist/registry/tool/execute.test.js.map +1 -1
  82. package/dist/registry/toolset/catalog.test.js +62 -0
  83. package/dist/registry/toolset/catalog.test.js.map +1 -1
  84. package/dist/run/reporter.d.ts.map +1 -1
  85. package/dist/run/reporter.js +8 -0
  86. package/dist/run/reporter.js.map +1 -1
  87. package/dist/runtime/query/__tests__/capability-negotiation.test.d.ts +2 -0
  88. package/dist/runtime/query/__tests__/capability-negotiation.test.d.ts.map +1 -0
  89. package/dist/runtime/query/__tests__/capability-negotiation.test.js +184 -0
  90. package/dist/runtime/query/__tests__/capability-negotiation.test.js.map +1 -0
  91. package/dist/runtime/query/__tests__/checkpoint-store.test.d.ts +2 -0
  92. package/dist/runtime/query/__tests__/checkpoint-store.test.d.ts.map +1 -0
  93. package/dist/runtime/query/__tests__/checkpoint-store.test.js +259 -0
  94. package/dist/runtime/query/__tests__/checkpoint-store.test.js.map +1 -0
  95. package/dist/runtime/query/__tests__/checkpoint.test.js +9 -3
  96. package/dist/runtime/query/__tests__/checkpoint.test.js.map +1 -1
  97. package/dist/runtime/query/__tests__/deferred-tools.test.js +43 -5
  98. package/dist/runtime/query/__tests__/deferred-tools.test.js.map +1 -1
  99. package/dist/runtime/query/__tests__/long-document-flow.test.d.ts +2 -0
  100. package/dist/runtime/query/__tests__/long-document-flow.test.d.ts.map +1 -0
  101. package/dist/runtime/query/__tests__/long-document-flow.test.js +154 -0
  102. package/dist/runtime/query/__tests__/long-document-flow.test.js.map +1 -0
  103. package/dist/runtime/query/__tests__/orphan-task-turn-end.test.d.ts +2 -0
  104. package/dist/runtime/query/__tests__/orphan-task-turn-end.test.d.ts.map +1 -0
  105. package/dist/runtime/query/__tests__/orphan-task-turn-end.test.js +110 -0
  106. package/dist/runtime/query/__tests__/orphan-task-turn-end.test.js.map +1 -0
  107. package/dist/runtime/query/__tests__/prompt.test.js +15 -11
  108. package/dist/runtime/query/__tests__/prompt.test.js.map +1 -1
  109. package/dist/runtime/query/checkpoint.d.ts +16 -2
  110. package/dist/runtime/query/checkpoint.d.ts.map +1 -1
  111. package/dist/runtime/query/checkpoint.js +21 -7
  112. package/dist/runtime/query/checkpoint.js.map +1 -1
  113. package/dist/runtime/query/context.d.ts +7 -0
  114. package/dist/runtime/query/context.d.ts.map +1 -1
  115. package/dist/runtime/query/context.js +1 -0
  116. package/dist/runtime/query/context.js.map +1 -1
  117. package/dist/runtime/query/index.d.ts +29 -0
  118. package/dist/runtime/query/index.d.ts.map +1 -1
  119. package/dist/runtime/query/index.js +98 -8
  120. package/dist/runtime/query/index.js.map +1 -1
  121. package/dist/runtime/query/iteration/index.d.ts +3 -55
  122. package/dist/runtime/query/iteration/index.d.ts.map +1 -1
  123. package/dist/runtime/query/iteration/index.js +260 -656
  124. package/dist/runtime/query/iteration/index.js.map +1 -1
  125. package/dist/runtime/query/iteration/phases/checkpoint.d.ts.map +1 -1
  126. package/dist/runtime/query/iteration/phases/checkpoint.js +24 -2
  127. package/dist/runtime/query/iteration/phases/checkpoint.js.map +1 -1
  128. package/dist/runtime/query/iteration/phases/compaction.d.ts +8 -0
  129. package/dist/runtime/query/iteration/phases/compaction.d.ts.map +1 -1
  130. package/dist/runtime/query/iteration/phases/compaction.js +84 -3
  131. package/dist/runtime/query/iteration/phases/compaction.js.map +1 -1
  132. package/dist/runtime/query/iteration/phases/context.d.ts +13 -2
  133. package/dist/runtime/query/iteration/phases/context.d.ts.map +1 -1
  134. package/dist/runtime/query/iteration/phases/context.js +53 -0
  135. package/dist/runtime/query/iteration/phases/context.js.map +1 -1
  136. package/dist/runtime/query/iteration/phases/hitl-abort-race.test.d.ts +2 -0
  137. package/dist/runtime/query/iteration/phases/hitl-abort-race.test.d.ts.map +1 -0
  138. package/dist/runtime/query/iteration/phases/hitl-abort-race.test.js +73 -0
  139. package/dist/runtime/query/iteration/phases/hitl-abort-race.test.js.map +1 -0
  140. package/dist/runtime/query/iteration/phases/hitl-answer-question.test.d.ts +2 -0
  141. package/dist/runtime/query/iteration/phases/hitl-answer-question.test.d.ts.map +1 -0
  142. package/dist/runtime/query/iteration/phases/hitl-answer-question.test.js +120 -0
  143. package/dist/runtime/query/iteration/phases/hitl-answer-question.test.js.map +1 -0
  144. package/dist/runtime/query/iteration/phases/tool-review.d.ts +1 -1
  145. package/dist/runtime/query/iteration/phases/tool-review.d.ts.map +1 -1
  146. package/dist/runtime/query/iteration/phases/tool-review.js +7 -2
  147. package/dist/runtime/query/iteration/phases/tool-review.js.map +1 -1
  148. package/dist/runtime/query/iteration/phases/working-memory-compaction.test.d.ts +19 -0
  149. package/dist/runtime/query/iteration/phases/working-memory-compaction.test.d.ts.map +1 -0
  150. package/dist/runtime/query/iteration/phases/working-memory-compaction.test.js +177 -0
  151. package/dist/runtime/query/iteration/phases/working-memory-compaction.test.js.map +1 -0
  152. package/dist/runtime/query/iteration/phases/working-memory.d.ts +31 -0
  153. package/dist/runtime/query/iteration/phases/working-memory.d.ts.map +1 -0
  154. package/dist/runtime/query/iteration/phases/working-memory.js +81 -0
  155. package/dist/runtime/query/iteration/phases/working-memory.js.map +1 -0
  156. package/dist/runtime/query/iteration/stream-turn.d.ts +35 -0
  157. package/dist/runtime/query/iteration/stream-turn.d.ts.map +1 -0
  158. package/dist/runtime/query/iteration/stream-turn.js +341 -0
  159. package/dist/runtime/query/iteration/stream-turn.js.map +1 -0
  160. package/dist/runtime/query/replay/list.d.ts +17 -1
  161. package/dist/runtime/query/replay/list.d.ts.map +1 -1
  162. package/dist/runtime/query/replay/list.js +21 -4
  163. package/dist/runtime/query/replay/list.js.map +1 -1
  164. package/dist/runtime/query/replay/prepare.d.ts +10 -0
  165. package/dist/runtime/query/replay/prepare.d.ts.map +1 -1
  166. package/dist/runtime/query/replay/prepare.js +20 -4
  167. package/dist/runtime/query/replay/prepare.js.map +1 -1
  168. package/dist/session/workspace/__tests__/shared-run.test.js +1 -1
  169. package/dist/session/workspace/__tests__/shared-run.test.js.map +1 -1
  170. package/dist/session/workspace/shared-run.js +1 -0
  171. package/dist/session/workspace/shared-run.js.map +1 -1
  172. package/dist/skills/loader.test.js +8 -8
  173. package/dist/skills/loader.test.js.map +1 -1
  174. package/dist/store/index.d.ts +1 -0
  175. package/dist/store/index.d.ts.map +1 -1
  176. package/dist/store/index.js +1 -0
  177. package/dist/store/index.js.map +1 -1
  178. package/dist/store/run/checkpoint-disk.d.ts +28 -0
  179. package/dist/store/run/checkpoint-disk.d.ts.map +1 -0
  180. package/dist/store/run/checkpoint-disk.js +55 -0
  181. package/dist/store/run/checkpoint-disk.js.map +1 -0
  182. package/dist/streaming/coalesce.test.js.map +1 -1
  183. package/dist/tools/builtins/__tests__/payload-budget.test.js +48 -0
  184. package/dist/tools/builtins/__tests__/payload-budget.test.js.map +1 -1
  185. package/dist/tools/builtins/__tests__/read-file.test.js +9 -0
  186. package/dist/tools/builtins/__tests__/read-file.test.js.map +1 -1
  187. package/dist/tools/builtins/read-file.d.ts.map +1 -1
  188. package/dist/tools/builtins/read-file.js +50 -2
  189. package/dist/tools/builtins/read-file.js.map +1 -1
  190. package/dist/tools/builtins/search-tools.d.ts.map +1 -1
  191. package/dist/tools/builtins/search-tools.js +32 -7
  192. package/dist/tools/builtins/search-tools.js.map +1 -1
  193. package/dist/tools/coordinator/__tests__/approve-plan.test.d.ts +2 -0
  194. package/dist/tools/coordinator/__tests__/approve-plan.test.d.ts.map +1 -0
  195. package/dist/tools/coordinator/__tests__/approve-plan.test.js +114 -0
  196. package/dist/tools/coordinator/__tests__/approve-plan.test.js.map +1 -0
  197. package/dist/tools/coordinator/__tests__/ask-user-question.test.d.ts +2 -0
  198. package/dist/tools/coordinator/__tests__/ask-user-question.test.d.ts.map +1 -0
  199. package/dist/tools/coordinator/__tests__/ask-user-question.test.js +309 -0
  200. package/dist/tools/coordinator/__tests__/ask-user-question.test.js.map +1 -0
  201. package/dist/tools/coordinator/index.d.ts +8 -0
  202. package/dist/tools/coordinator/index.d.ts.map +1 -1
  203. package/dist/tools/coordinator/index.js +195 -15
  204. package/dist/tools/coordinator/index.js.map +1 -1
  205. package/dist/types/agent/index.d.ts +1 -0
  206. package/dist/types/agent/index.d.ts.map +1 -1
  207. package/dist/types/agent/index.js +1 -0
  208. package/dist/types/agent/index.js.map +1 -1
  209. package/dist/types/agent/reactive.d.ts +15 -0
  210. package/dist/types/agent/reactive.d.ts.map +1 -1
  211. package/dist/types/agent/supervisor.d.ts +18 -0
  212. package/dist/types/agent/supervisor.d.ts.map +1 -1
  213. package/dist/types/agent/working-memory.d.ts +29 -0
  214. package/dist/types/agent/working-memory.d.ts.map +1 -0
  215. package/dist/types/agent/working-memory.js +2 -0
  216. package/dist/types/agent/working-memory.js.map +1 -0
  217. package/dist/types/common/index.d.ts +10 -0
  218. package/dist/types/common/index.d.ts.map +1 -1
  219. package/dist/types/common/index.js +18 -0
  220. package/dist/types/common/index.js.map +1 -1
  221. package/dist/types/common/usage-merge.test.d.ts +2 -0
  222. package/dist/types/common/usage-merge.test.d.ts.map +1 -0
  223. package/dist/types/common/usage-merge.test.js +48 -0
  224. package/dist/types/common/usage-merge.test.js.map +1 -0
  225. package/dist/types/connector/mcp.d.ts +9 -2
  226. package/dist/types/connector/mcp.d.ts.map +1 -1
  227. package/dist/types/hitl/index.d.ts +42 -0
  228. package/dist/types/hitl/index.d.ts.map +1 -1
  229. package/dist/types/hitl/index.js +11 -0
  230. package/dist/types/hitl/index.js.map +1 -1
  231. package/dist/types/provider/chat.d.ts +9 -0
  232. package/dist/types/provider/chat.d.ts.map +1 -1
  233. package/dist/types/provider/config.d.ts +18 -0
  234. package/dist/types/provider/config.d.ts.map +1 -1
  235. package/dist/types/provider/interface.d.ts +10 -0
  236. package/dist/types/provider/interface.d.ts.map +1 -1
  237. package/dist/types/run/checkpoint-store.d.ts +64 -0
  238. package/dist/types/run/checkpoint-store.d.ts.map +1 -0
  239. package/dist/types/run/checkpoint-store.js +12 -0
  240. package/dist/types/run/checkpoint-store.js.map +1 -0
  241. package/dist/types/run/config.d.ts +27 -0
  242. package/dist/types/run/config.d.ts.map +1 -1
  243. package/dist/types/run/events.d.ts +6 -0
  244. package/dist/types/run/events.d.ts.map +1 -1
  245. package/dist/types/run/events.js.map +1 -1
  246. package/dist/types/run/index.d.ts +1 -0
  247. package/dist/types/run/index.d.ts.map +1 -1
  248. package/dist/types/run/index.js +1 -0
  249. package/dist/types/run/index.js.map +1 -1
  250. package/dist/version.d.ts.map +1 -1
  251. package/dist/version.js +25 -4
  252. package/dist/version.js.map +1 -1
  253. package/package.json +1 -1
  254. package/src/agents/ReactiveAgent.ts +5 -0
  255. package/src/agents/SupervisorAgent.ts +56 -16
  256. package/src/agents/__tests__/supervisor-ledger.test.ts +185 -0
  257. package/src/bridge/a2a/mapper.ts +3 -0
  258. package/src/bridge/sse/mapper.ts +5 -0
  259. package/src/config/runtime.ts +9 -0
  260. package/src/connector/builtins/webhook.test.ts +21 -0
  261. package/src/connector/index.ts +1 -0
  262. package/src/connector/mcp/client.ts +4 -0
  263. package/src/connector/mcp/index.ts +1 -0
  264. package/src/connector/mcp/streamable-http.test.ts +169 -0
  265. package/src/connector/mcp/streamable-http.ts +152 -0
  266. package/src/{runtime/query → constants}/continuation.ts +5 -0
  267. package/src/constants/index.ts +1 -0
  268. package/src/manager/run/persistence.ts +34 -1
  269. package/src/persona/assembler.ts +18 -6
  270. package/src/provider/capabilities.ts +47 -0
  271. package/src/provider/collect.ts +4 -1
  272. package/src/provider/index.ts +5 -0
  273. package/src/public-runtime.ts +5 -0
  274. package/src/registry/tool/execute.test.ts +85 -10
  275. package/src/registry/tool/execute.ts +124 -21
  276. package/src/registry/toolset/catalog.test.ts +68 -0
  277. package/src/run/reporter.ts +9 -0
  278. package/src/runtime/query/__tests__/capability-negotiation.test.ts +248 -0
  279. package/src/runtime/query/__tests__/checkpoint-store.test.ts +322 -0
  280. package/src/runtime/query/__tests__/checkpoint.test.ts +15 -6
  281. package/src/runtime/query/__tests__/deferred-tools.test.ts +52 -5
  282. package/src/runtime/query/__tests__/long-document-flow.test.ts +180 -0
  283. package/src/runtime/query/__tests__/orphan-task-turn-end.test.ts +127 -0
  284. package/src/runtime/query/__tests__/prompt.test.ts +15 -11
  285. package/src/runtime/query/checkpoint.ts +23 -9
  286. package/src/runtime/query/context.ts +9 -0
  287. package/src/runtime/query/index.ts +148 -29
  288. package/src/runtime/query/iteration/index.ts +316 -831
  289. package/src/runtime/query/iteration/phases/checkpoint.ts +32 -2
  290. package/src/runtime/query/iteration/phases/compaction.ts +89 -3
  291. package/src/runtime/query/iteration/phases/context.ts +59 -3
  292. package/src/runtime/query/iteration/phases/hitl-abort-race.test.ts +86 -0
  293. package/src/runtime/query/iteration/phases/hitl-answer-question.test.ts +139 -0
  294. package/src/runtime/query/iteration/phases/tool-review.ts +7 -3
  295. package/src/runtime/query/iteration/phases/working-memory-compaction.test.ts +215 -0
  296. package/src/runtime/query/iteration/phases/working-memory.ts +86 -0
  297. package/src/runtime/query/iteration/stream-turn.ts +396 -0
  298. package/src/runtime/query/replay/list.ts +38 -4
  299. package/src/runtime/query/replay/prepare.ts +37 -4
  300. package/src/session/workspace/__tests__/shared-run.test.ts +1 -1
  301. package/src/session/workspace/shared-run.ts +1 -0
  302. package/src/skills/loader.test.ts +8 -8
  303. package/src/store/index.ts +1 -0
  304. package/src/store/run/checkpoint-disk.ts +67 -0
  305. package/src/streaming/coalesce.test.ts +3 -3
  306. package/src/tools/builtins/__tests__/payload-budget.test.ts +63 -0
  307. package/src/tools/builtins/__tests__/read-file.test.ts +12 -0
  308. package/src/tools/builtins/read-file.ts +50 -2
  309. package/src/tools/builtins/search-tools.ts +38 -7
  310. package/src/tools/coordinator/__tests__/approve-plan.test.ts +139 -0
  311. package/src/tools/coordinator/__tests__/ask-user-question.test.ts +370 -0
  312. package/src/tools/coordinator/index.ts +227 -17
  313. package/src/types/agent/index.ts +1 -0
  314. package/src/types/agent/reactive.ts +17 -0
  315. package/src/types/agent/supervisor.ts +20 -0
  316. package/src/types/agent/working-memory.ts +29 -0
  317. package/src/types/common/index.ts +19 -0
  318. package/src/types/common/usage-merge.test.ts +53 -0
  319. package/src/types/connector/mcp.ts +14 -2
  320. package/src/types/hitl/index.ts +52 -1
  321. package/src/types/provider/chat.ts +10 -0
  322. package/src/types/provider/config.ts +18 -0
  323. package/src/types/provider/interface.ts +11 -0
  324. package/src/types/run/checkpoint-store.ts +72 -0
  325. package/src/types/run/config.ts +30 -0
  326. package/src/types/run/events.ts +13 -0
  327. package/src/types/run/index.ts +1 -0
  328. package/src/version.ts +26 -4
  329. package/dist/runtime/query/continuation.d.ts.map +0 -1
  330. package/dist/runtime/query/continuation.js.map +0 -1
@@ -1,5 +1,11 @@
1
1
  import { describe, expect, it } from 'vitest';
2
2
  import { CheckpointManager, projectEmergencyToCheckpoint } from '../checkpoint.js';
3
+ const TEST_SCOPE = {
4
+ tenantId: 'tnt_test',
5
+ projectId: 'prj_test',
6
+ sessionId: 'ses_test',
7
+ runId: 'run_test',
8
+ };
3
9
  function makeCheckpoint(overrides = {}) {
4
10
  return {
5
11
  id: 'cp_test_a',
@@ -52,7 +58,7 @@ describe('CheckpointManager.listEntries', () => {
52
58
  ],
53
59
  }),
54
60
  ]);
55
- const mgr = new CheckpointManager(store);
61
+ const mgr = new CheckpointManager(store, TEST_SCOPE);
56
62
  const entries = await mgr.listEntries();
57
63
  expect(entries).toHaveLength(2);
58
64
  expect(entries[0]).toEqual({
@@ -71,7 +77,7 @@ describe('CheckpointManager.listEntries', () => {
71
77
  });
72
78
  });
73
79
  it('returns empty array when no checkpoints exist', async () => {
74
- const mgr = new CheckpointManager(makeStoreStub([]));
80
+ const mgr = new CheckpointManager(makeStoreStub([]), TEST_SCOPE);
75
81
  expect(await mgr.listEntries()).toEqual([]);
76
82
  });
77
83
  it('does not include full checkpoint payload fields', async () => {
@@ -80,7 +86,7 @@ describe('CheckpointManager.listEntries', () => {
80
86
  toolResultHashes: { call_x: 'hash' },
81
87
  branchStack: [{ agentId: 'a', decision: 'd', confidence: 1, timestamp: 0 }],
82
88
  }),
83
- ]));
89
+ ]), TEST_SCOPE);
84
90
  const [entry] = await mgr.listEntries();
85
91
  expect(entry).not.toHaveProperty('tokenUsage');
86
92
  expect(entry).not.toHaveProperty('costInfo');
@@ -1 +1 @@
1
- {"version":3,"file":"checkpoint.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/checkpoint.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAK7C,OAAO,EAAE,iBAAiB,EAAE,4BAA4B,EAAE,MAAM,kBAAkB,CAAA;AAElF,SAAS,cAAc,CAAC,YAA0C,EAAE;IACnE,OAAO;QACN,EAAE,EAAE,WAA2B;QAC/B,KAAK,EAAE,UAAmB;QAC1B,SAAS,EAAE,CAAC;QACZ,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;QAC9C,UAAU,EAAE;YACX,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;SACnB;QACD,QAAQ,EAAE;YACT,cAAc,EAAE,CAAC;YACjB,eAAe,EAAE,CAAC;YAClB,SAAS,EAAE,CAAC;YACZ,aAAa,EAAE,CAAC;SAChB;QACD,UAAU,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE;QACjD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,GAAG,SAAS;KACZ,CAAA;AACF,CAAC;AAED,SAAS,aAAa,CAAC,WAAkC;IACxD,OAAO;QACN,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,WAAW;KACb,CAAA;AAC7B,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,KAAK,GAAG,aAAa,CAAC;YAC3B,cAAc,CAAC;gBACd,EAAE,EAAE,MAAsB;gBAC1B,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,IAAI;gBACf,QAAQ,EAAE;oBACT,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;oBAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;iBACvC;aACD,CAAC;YACF,cAAc,CAAC;gBACd,EAAE,EAAE,MAAsB;gBAC1B,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,IAAI;gBACf,QAAQ,EAAE;oBACT,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;oBAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;oBACvC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;iBACjC;aACD,CAAC;SACF,CAAC,CAAA;QAEF,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,KAAK,CAAC,CAAA;QACxC,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAA;QAEvC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1B,EAAE,EAAE,MAAM;YACV,KAAK,EAAE,UAAU;YACjB,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,IAAI;YACf,YAAY,EAAE,CAAC;SACf,CAAC,CAAA;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1B,EAAE,EAAE,MAAM;YACV,KAAK,EAAE,UAAU;YACjB,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,IAAI;YACf,YAAY,EAAE,CAAC;SACf,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAA;QACpD,MAAM,CAAC,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAChC,aAAa,CAAC;YACb,cAAc,CAAC;gBACd,gBAAgB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;gBACpC,WAAW,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;aAC3E,CAAC;SACF,CAAC,CACF,CAAA;QACD,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,CAAA;QAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,CAAA;QAC5C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAA;QACpD,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,aAAa,CAAC,CAAA;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,CAAA;QAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,SAAS,iBAAiB,CAAC,YAAwC,EAAE;IACpE,OAAO;QACN,EAAE,EAAE,cAAiC;QACrC,KAAK,EAAE,YAAqB;QAC5B,QAAQ,EAAE;YACT,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE;YAC7C,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE;SACzC;QACD,UAAU,EAAE;YACX,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;SACnB;QACD,gBAAgB,EAAE,CAAC;QACnB,SAAS,EAAE,KAAK;QAChB,OAAO,EAAE,KAAK;QACd,aAAa,EAAE,SAAS;QACxB,GAAG,SAAS;KACZ,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC7C,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QACnE,MAAM,IAAI,GAAG,iBAAiB,EAAE,CAAA;QAChC,MAAM,EAAE,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAA;QAE7C,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QACnC,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC5B,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACvC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC3C,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAChC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAA;QACtE,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YAC3B,cAAc,EAAE,CAAC;YACjB,eAAe,EAAE,CAAC;YAClB,SAAS,EAAE,CAAC;YACZ,aAAa,EAAE,CAAC;SAChB,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QAC1E,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,EAAE,EAAE,cAAiC,EAAE,CAAC,CAAA;QACzE,MAAM,GAAG,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAA;QAC9C,MAAM,GAAG,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAA;QAE9C,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAA;QAC1C,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAC5B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC3E,MAAM,EAAE,GAAG,4BAA4B,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;QAChG,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;QACnF,MAAM,EAAE,GAAG,4BAA4B,CAAC,iBAAiB,EAAE,CAAC,CAAA;QAC5D,MAAM,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,aAAa,EAAE,CAAA;QAC3C,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAA;QACtC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,aAAa,EAAE,CAAA;IACtC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
1
+ {"version":3,"file":"checkpoint.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/checkpoint.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAM7C,OAAO,EAAE,iBAAiB,EAAE,4BAA4B,EAAE,MAAM,kBAAkB,CAAA;AAElF,MAAM,UAAU,GAAuB;IACtC,QAAQ,EAAE,UAAsB;IAChC,SAAS,EAAE,UAAuB;IAClC,SAAS,EAAE,UAAuB;IAClC,KAAK,EAAE,UAAmB;CAC1B,CAAA;AAED,SAAS,cAAc,CAAC,YAA0C,EAAE;IACnE,OAAO;QACN,EAAE,EAAE,WAA2B;QAC/B,KAAK,EAAE,UAAmB;QAC1B,SAAS,EAAE,CAAC;QACZ,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;QAC9C,UAAU,EAAE;YACX,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;SACnB;QACD,QAAQ,EAAE;YACT,cAAc,EAAE,CAAC;YACjB,eAAe,EAAE,CAAC;YAClB,SAAS,EAAE,CAAC;YACZ,aAAa,EAAE,CAAC;SAChB;QACD,UAAU,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE;QACjD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,GAAG,SAAS;KACZ,CAAA;AACF,CAAC;AAED,SAAS,aAAa,CAAC,WAAkC;IACxD,OAAO;QACN,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,WAAW;KACV,CAAA;AAChC,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,KAAK,GAAG,aAAa,CAAC;YAC3B,cAAc,CAAC;gBACd,EAAE,EAAE,MAAsB;gBAC1B,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,IAAI;gBACf,QAAQ,EAAE;oBACT,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;oBAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;iBACvC;aACD,CAAC;YACF,cAAc,CAAC;gBACd,EAAE,EAAE,MAAsB;gBAC1B,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,IAAI;gBACf,QAAQ,EAAE;oBACT,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;oBAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;oBACvC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;iBACjC;aACD,CAAC;SACF,CAAC,CAAA;QAEF,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAA;QACpD,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAA;QAEvC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1B,EAAE,EAAE,MAAM;YACV,KAAK,EAAE,UAAU;YACjB,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,IAAI;YACf,YAAY,EAAE,CAAC;SACf,CAAC,CAAA;QACF,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1B,EAAE,EAAE,MAAM;YACV,KAAK,EAAE,UAAU;YACjB,SAAS,EAAE,CAAC;YACZ,SAAS,EAAE,IAAI;YACf,YAAY,EAAE,CAAC;SACf,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAAC,aAAa,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC,CAAA;QAChE,MAAM,CAAC,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,GAAG,GAAG,IAAI,iBAAiB,CAChC,aAAa,CAAC;YACb,cAAc,CAAC;gBACd,gBAAgB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;gBACpC,WAAW,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;aAC3E,CAAC;SACF,CAAC,EACF,UAAU,CACV,CAAA;QACD,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,CAAA;QAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,CAAA;QAC5C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAA;QACpD,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,aAAa,CAAC,CAAA;QAC/C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,CAAA;QAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,SAAS,iBAAiB,CAAC,YAAwC,EAAE;IACpE,OAAO;QACN,EAAE,EAAE,cAAiC;QACrC,KAAK,EAAE,YAAqB;QAC5B,QAAQ,EAAE;YACT,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE;YAC7C,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE;SACzC;QACD,UAAU,EAAE;YACX,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;SACnB;QACD,gBAAgB,EAAE,CAAC;QACnB,SAAS,EAAE,KAAK;QAChB,OAAO,EAAE,KAAK;QACd,aAAa,EAAE,SAAS;QACxB,GAAG,SAAS;KACZ,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC7C,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QACnE,MAAM,IAAI,GAAG,iBAAiB,EAAE,CAAA;QAChC,MAAM,EAAE,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAA;QAE7C,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QACnC,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC5B,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACvC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC3C,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAChC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAA;QACtE,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YAC3B,cAAc,EAAE,CAAC;YACjB,eAAe,EAAE,CAAC;YAClB,SAAS,EAAE,CAAC;YACZ,aAAa,EAAE,CAAC;SAChB,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QAC1E,MAAM,IAAI,GAAG,iBAAiB,CAAC,EAAE,EAAE,EAAE,cAAiC,EAAE,CAAC,CAAA;QACzE,MAAM,GAAG,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAA;QAC9C,MAAM,GAAG,GAAG,4BAA4B,CAAC,IAAI,CAAC,CAAA;QAE9C,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAA;QAC1C,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAC5B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC3E,MAAM,EAAE,GAAG,4BAA4B,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;QAChG,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;QACnF,MAAM,EAAE,GAAG,4BAA4B,CAAC,iBAAiB,EAAE,CAAC,CAAA;QAC5D,MAAM,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,aAAa,EAAE,CAAA;QAC3C,MAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAA;QACtC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,aAAa,EAAE,CAAA;IACtC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
@@ -127,8 +127,10 @@ describe('query deferred tool discovery', () => {
127
127
  it('does not let search_tools reveal or activate deferred tools outside allowedTools', async () => {
128
128
  const tools = new ToolRegistry();
129
129
  registerDeferredDocumentTool(tools);
130
- registerDeferredDocumentTool(tools, 'dangerous_delete_document');
131
- const result = await SearchToolsTool.execute({ query: 'delete' }, {
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' }, {
132
134
  runId: 'run_deferred_allowed_tools',
133
135
  workingDirectory: '/tmp',
134
136
  abortSignal: new AbortController().signal,
@@ -138,10 +140,46 @@ describe('query deferred tool discovery', () => {
138
140
  allowedTools: ['generate_document', SearchToolsTool.name],
139
141
  });
140
142
  expect(result.success).toBe(true);
141
- expect(result.output).toContain('No deferred tools matching "delete"');
142
- expect(result.output).not.toContain('dangerous_delete_document');
143
+ expect(result.output).toContain('No deferred tools matching "dangerous"');
144
+ expect(result.output).not.toContain('dangerous_purge_document');
143
145
  expect(tools.getAvailability('generate_document')).toBe('deferred');
144
- expect(tools.getAvailability('dangerous_delete_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
+ }
145
183
  });
146
184
  });
147
185
  //# sourceMappingURL=deferred-tools.test.js.map
@@ -1 +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,2BAA2B,CAAC,CAAA;QAEhE,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,QAAQ,EAAE,EACnB;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,qCAAqC,CAAC,CAAA;QACtE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAA;QAChE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACnE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IAC5E,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
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=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"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=orphan-task-turn-end.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"orphan-task-turn-end.test.d.ts","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/orphan-task-turn-end.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,110 @@
1
+ import { 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 { createUserMessage } from '../../../types/message/index.js';
8
+ import { drainQuery } from '../index.js';
9
+ const ZERO_USAGE = {
10
+ promptTokens: 0,
11
+ completionTokens: 0,
12
+ totalTokens: 0,
13
+ cachedTokens: 0,
14
+ cacheWriteTokens: 0,
15
+ };
16
+ /** Ends its turn with plain text on the first (and only) call. */
17
+ class SingleTurnProvider {
18
+ id = 'single-turn';
19
+ name = 'Single Turn Provider';
20
+ calls = 0;
21
+ async *chatStream() {
22
+ this.calls += 1;
23
+ yield {
24
+ id: `msg_${this.calls}`,
25
+ delta: { content: 'Final answer.' },
26
+ };
27
+ yield {
28
+ id: `msg_${this.calls}`,
29
+ delta: {},
30
+ finishReason: 'stop',
31
+ usage: ZERO_USAGE,
32
+ };
33
+ }
34
+ }
35
+ /**
36
+ * Gateway that permanently reports one running task. Every dispatch
37
+ * tool is blocking, so a running task at end-of-turn is an orphan —
38
+ * there is no notification producer that could ever deliver its
39
+ * result (the listener was removed in dc16d58). The run must NOT
40
+ * busy-wait on it.
41
+ */
42
+ function orphanTaskGateway() {
43
+ const handle = {
44
+ taskId: 'task_orphan',
45
+ agentId: 'agent_worker',
46
+ state: 'running',
47
+ createdAt: Date.now(),
48
+ };
49
+ return {
50
+ createTask: async () => handle,
51
+ waitForTask: () => new Promise(() => { }),
52
+ continueTask: async () => { },
53
+ cancelTask: () => { },
54
+ getTask: () => handle,
55
+ listTasks: () => [handle],
56
+ onTaskCompleted: () => () => { },
57
+ };
58
+ }
59
+ describe('end of turn with running agent tasks', () => {
60
+ let workdirs = [];
61
+ afterEach(async () => {
62
+ await Promise.all(workdirs.map((dir) => rm(dir, { recursive: true, force: true })));
63
+ workdirs = [];
64
+ });
65
+ // Regression: the loop used to poll `pendingNotifications` every
66
+ // 250ms for up to `runConfig.timeoutMs` (120s default) whenever the
67
+ // turn ended while the gateway still listed a running task — but
68
+ // nothing has pushed onto that queue since dc16d58 removed the
69
+ // onTaskCompleted producer, so the wait always injected nothing and
70
+ // the run hung for minutes. The vitest per-test timeout is the
71
+ // hang detector here: with the busy-wait present this test times
72
+ // out instead of completing.
73
+ it('ends the run promptly instead of busy-waiting on orphan tasks', async () => {
74
+ const provider = new SingleTurnProvider();
75
+ const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-orphan-task-'));
76
+ workdirs.push(workingDirectory);
77
+ const events = [];
78
+ const run = await drainQuery({
79
+ provider,
80
+ tools: new ToolRegistry(),
81
+ runConfig: {
82
+ model: 'mock-model',
83
+ // Deliberately longer than the vitest timeout: the old
84
+ // code waited min(timeoutMs, …) polling the dead queue.
85
+ timeoutMs: 120_000,
86
+ tokenBudget: 100_000,
87
+ maxIterations: 3,
88
+ maxResponseTokens: 256,
89
+ },
90
+ agentId: 'agent_test',
91
+ agentName: 'Test Agent',
92
+ messages: [createUserMessage('do the thing')],
93
+ workingDirectory,
94
+ taskGateway: orphanTaskGateway(),
95
+ sessionId: 'ses_orphan_task',
96
+ threadId: 'thd_orphan_task',
97
+ projectId: 'prj_orphan_task',
98
+ tenantId: 'tnt_orphan_task',
99
+ }, (event) => {
100
+ events.push(event);
101
+ });
102
+ expect(run.status).toBe('completed');
103
+ expect(run.stopReason).toBe('end_turn');
104
+ expect(run.result).toBe('Final answer.');
105
+ // One turn only — no futile re-invocation loop on the orphan.
106
+ expect(provider.calls).toBe(1);
107
+ expect(events.some((event) => event.type === 'run_failed')).toBe(false);
108
+ }, 10_000);
109
+ });
110
+ //# sourceMappingURL=orphan-task-turn-end.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"orphan-task-turn-end.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/orphan-task-turn-end.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AACrC,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;AAGhE,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;AAED,kEAAkE;AAClE,MAAM,kBAAkB;IACd,EAAE,GAAG,aAAa,CAAA;IAClB,IAAI,GAAG,sBAAsB,CAAA;IACtC,KAAK,GAAG,CAAC,CAAA;IAET,KAAK,CAAC,CAAC,UAAU;QAChB,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QACf,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;SACnC,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED;;;;;;GAMG;AACH,SAAS,iBAAiB;IACzB,MAAM,MAAM,GAAe;QAC1B,MAAM,EAAE,aAAuB;QAC/B,OAAO,EAAE,cAAc;QACvB,KAAK,EAAE,SAAS;QAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,CAAA;IACD,OAAO;QACN,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM;QAC9B,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,CAAa,GAAG,EAAE,GAAE,CAAC,CAAC;QACpD,YAAY,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC5B,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;QACpB,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM;QACrB,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC;QACzB,eAAe,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAE,CAAC;KAC/B,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACrD,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,iEAAiE;IACjE,oEAAoE;IACpE,iEAAiE;IACjE,+DAA+D;IAC/D,oEAAoE;IACpE,+DAA+D;IAC/D,iEAAiE;IACjE,6BAA6B;IAC7B,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,QAAQ,GAAG,IAAI,kBAAkB,EAAE,CAAA;QACzC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,oBAAoB,CAAC,CAAC,CAAA;QAC5E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAC/B,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,QAAQ;YACR,KAAK,EAAE,IAAI,YAAY,EAAE;YACzB,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,uDAAuD;gBACvD,wDAAwD;gBACxD,SAAS,EAAE,OAAO;gBAClB,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,cAAc,CAAC,CAAC;YAC7C,gBAAgB;YAChB,WAAW,EAAE,iBAAiB,EAAE;YAChC,SAAS,EAAE,iBAA8B;YACzC,QAAQ,EAAE,iBAA6B;YACvC,SAAS,EAAE,iBAA8B;YACzC,QAAQ,EAAE,iBAA6B;SACvC,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACvC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QACxC,8DAA8D;QAC9D,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC9B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACxE,CAAC,EAAE,MAAM,CAAC,CAAA;AACX,CAAC,CAAC,CAAA"}
@@ -38,17 +38,20 @@ describe('PromptBuilder runtime context', () => {
38
38
  skills: [
39
39
  {
40
40
  metadata: {
41
- name: 'project-documents',
42
- description: 'Draft and edit project documents from grounded inputs.',
41
+ name: 'delivery-briefing',
42
+ description: 'Draft and edit delivery briefings from grounded inputs.',
43
43
  },
44
- dirPath: '/repo/.agents/skills/project-documents',
44
+ dirPath: '/repo/.agents/skills/delivery-briefing',
45
45
  },
46
46
  ],
47
47
  }).build('full', '/tmp/work');
48
48
  expect(prompt).toContain('You are a project assistant.');
49
49
  expect(prompt).toContain('## Available Skills');
50
- expect(prompt).toContain('project-documents');
51
- expect(prompt).toContain('Draft and edit project documents');
50
+ expect(prompt).toContain('<available_skills>');
51
+ expect(prompt).toContain('delivery-briefing');
52
+ expect(prompt).toContain('Draft and edit delivery briefings');
53
+ expect(prompt).toContain('<location>/repo/.agents/skills/delivery-briefing/SKILL.md</location>');
54
+ expect(prompt).toContain('The following block is a manifest');
52
55
  expect(prompt).not.toContain('## Loaded Skills');
53
56
  });
54
57
  it('includes loaded skill bodies with systemPrompt while preserving the metadata catalogue', () => {
@@ -58,21 +61,22 @@ describe('PromptBuilder runtime context', () => {
58
61
  skills: [
59
62
  {
60
63
  metadata: {
61
- name: 'long-form-files',
62
- description: 'Create long files with bounded edit chunks.',
64
+ name: 'structured-file-authoring',
65
+ description: 'Create structured files with bounded edit chunks.',
63
66
  license: 'MIT',
64
67
  compatibility: 'Requires file tools',
65
68
  allowedTools: 'read write edit',
66
69
  },
67
70
  body: 'Use skeleton-first writes and bounded edit chunks.',
68
- dirPath: '/repo/.agents/skills/long-form-files',
71
+ dirPath: '/repo/.agents/skills/structured-file-authoring',
69
72
  },
70
73
  ],
71
74
  }).build('full', '/tmp/work');
72
75
  expect(prompt).toContain('## Available Skills');
73
- expect(prompt).toContain('license: MIT');
74
- expect(prompt).toContain('compatibility: Requires file tools');
75
- expect(prompt).toContain('allowed-tools: read write edit');
76
+ expect(prompt).toContain('<license>MIT</license>');
77
+ expect(prompt).toContain('<compatibility>Requires file tools</compatibility>');
78
+ expect(prompt).toContain('<allowed_tools>read write edit</allowed_tools>');
79
+ expect(prompt).toContain('read the SKILL.md at its <location> before writing code');
76
80
  expect(prompt).toContain('## Loaded Skills');
77
81
  expect(prompt).toContain('Use skeleton-first writes');
78
82
  });
@@ -1 +1 @@
1
- {"version":3,"file":"prompt.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/prompt.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEjD,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAA;AAE5C,SAAS,gBAAgB;IACxB,OAAO;QACN,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;QAChB,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAC3B,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;QACvB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAChC,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;KACI,CAAA;AACrC,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC9E,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;YAChC,YAAY,EAAE,mBAAmB;YACjC,KAAK,EAAE,gBAAgB,EAAE;YACzB,cAAc,EAAE;gBACf,KAAK,EAAE,cAAc;gBACrB,eAAe,EAAE,UAAU;gBAC3B,gBAAgB,EAAE,yCAAyC;gBAC3D,KAAK,EAAE,CAAC,0CAA0C,CAAC;aACnD;SACD,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QAE7B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAA;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAA;QACxD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAA;QACtD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yCAAyC,CAAC,CAAA;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0CAA0C,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAChF,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;YAChC,YAAY,EAAE,8BAA8B;YAC5C,KAAK,EAAE,gBAAgB,EAAE;YACzB,MAAM,EAAE;gBACP;oBACC,QAAQ,EAAE;wBACT,IAAI,EAAE,mBAAmB;wBACzB,WAAW,EAAE,wDAAwD;qBACrE;oBACD,OAAO,EAAE,wCAAwC;iBACjD;aACD;SACD,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QAE7B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAA;QACxD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAA;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAA;QAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wFAAwF,EAAE,GAAG,EAAE;QACjG,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;YAChC,YAAY,EAAE,8BAA8B;YAC5C,KAAK,EAAE,gBAAgB,EAAE;YACzB,MAAM,EAAE;gBACP;oBACC,QAAQ,EAAE;wBACT,IAAI,EAAE,iBAAiB;wBACvB,WAAW,EAAE,6CAA6C;wBAC1D,OAAO,EAAE,KAAK;wBACd,aAAa,EAAE,qBAAqB;wBACpC,YAAY,EAAE,iBAAiB;qBAC/B;oBACD,IAAI,EAAE,oDAAoD;oBAC1D,OAAO,EAAE,sCAAsC;iBAC/C;aACD;SACD,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QAE7B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,oCAAoC,CAAC,CAAA;QAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAA;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAA;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
1
+ {"version":3,"file":"prompt.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/prompt.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEjD,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAA;AAE5C,SAAS,gBAAgB;IACxB,OAAO;QACN,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;QACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;QACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;QAChB,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAC3B,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC;QACvB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;QAChC,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;KACI,CAAA;AACrC,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC9E,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;YAChC,YAAY,EAAE,mBAAmB;YACjC,KAAK,EAAE,gBAAgB,EAAE;YACzB,cAAc,EAAE;gBACf,KAAK,EAAE,cAAc;gBACrB,eAAe,EAAE,UAAU;gBAC3B,gBAAgB,EAAE,yCAAyC;gBAC3D,KAAK,EAAE,CAAC,0CAA0C,CAAC;aACnD;SACD,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QAE7B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAA;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAA;QACxD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAA;QACtD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yCAAyC,CAAC,CAAA;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0CAA0C,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAChF,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;YAChC,YAAY,EAAE,8BAA8B;YAC5C,KAAK,EAAE,gBAAgB,EAAE;YACzB,MAAM,EAAE;gBACP;oBACC,QAAQ,EAAE;wBACT,IAAI,EAAE,mBAAmB;wBACzB,WAAW,EAAE,yDAAyD;qBACtE;oBACD,OAAO,EAAE,wCAAwC;iBACjD;aACD;SACD,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QAE7B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAA;QACxD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAA;QAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAA;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAA;QAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,sEAAsE,CAAC,CAAA;QAChG,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAA;QAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wFAAwF,EAAE,GAAG,EAAE;QACjG,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC;YAChC,YAAY,EAAE,8BAA8B;YAC5C,KAAK,EAAE,gBAAgB,EAAE;YACzB,MAAM,EAAE;gBACP;oBACC,QAAQ,EAAE;wBACT,IAAI,EAAE,2BAA2B;wBACjC,WAAW,EAAE,mDAAmD;wBAChE,OAAO,EAAE,KAAK;wBACd,aAAa,EAAE,qBAAqB;wBACpC,YAAY,EAAE,iBAAiB;qBAC/B;oBACD,IAAI,EAAE,oDAAoD;oBAC1D,OAAO,EAAE,gDAAgD;iBACzD;aACD;SACD,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QAE7B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAA;QAClD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,oDAAoD,CAAC,CAAA;QAC9E,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gDAAgD,CAAC,CAAA;QAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yDAAyD,CAAC,CAAA;QACnF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAA;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
@@ -1,8 +1,14 @@
1
1
  import type { RunPersistence } from '../../manager/run/persistence.js';
2
- import type { RunDiskStore } from '../../store/run/disk.js';
3
2
  import type { ActiveNodeInfo, BranchStackEntry, CheckpointId, CheckpointSummary, IterationCheckpoint } from '../../types/hitl/index.js';
3
+ import type { CheckpointRunScope, CheckpointStore } from '../../types/run/checkpoint-store.js';
4
4
  import type { EmergencySaveData } from '../../types/run/emergency.js';
5
5
  import type { CheckpointListEntry } from '../../types/run/replay.js';
6
+ /**
7
+ * Projection from a full checkpoint payload to the public listing entry.
8
+ * Exported for the replay `listCheckpoints` entry point, which projects
9
+ * store results without constructing a `CheckpointManager`.
10
+ */
11
+ export declare function toCheckpointListEntry(cp: IterationCheckpoint): CheckpointListEntry;
6
12
  /**
7
13
  * Project an {@link EmergencySaveData} dump to an {@link IterationCheckpoint}
8
14
  * shape so `replay({ fromCheckpoint: 'emergency' })` can consume it through
@@ -19,7 +25,15 @@ import type { CheckpointListEntry } from '../../types/run/replay.js';
19
25
  export declare function projectEmergencyToCheckpoint(dump: EmergencySaveData): IterationCheckpoint;
20
26
  export declare class CheckpointManager {
21
27
  private store;
22
- constructor(store: RunDiskStore);
28
+ private scope;
29
+ /**
30
+ * @param store scope-keyed checkpoint persistence. The default query
31
+ * pipeline passes the run's disk-backed store
32
+ * ({@link import('../../store/run/checkpoint-disk.js').DiskCheckpointStore});
33
+ * hosts inject their own via `QueryParams.checkpointStore`.
34
+ * @param scope the run every operation of this manager is keyed to.
35
+ */
36
+ constructor(store: CheckpointStore, scope: CheckpointRunScope);
23
37
  create(runMgr: RunPersistence, iteration: number, extra?: {
24
38
  toolResults?: Array<{
25
39
  toolCallId: string;
@@ -1 +1 @@
1
- {"version":3,"file":"checkpoint.d.ts","sourceRoot":"","sources":["../../../src/runtime/query/checkpoint.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAA;AACtE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAA;AAC3D,OAAO,KAAK,EACX,cAAc,EACd,gBAAgB,EAChB,YAAY,EACZ,iBAAiB,EACjB,mBAAmB,EACnB,MAAM,2BAA2B,CAAA;AAElC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAA;AACrE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAA;AAepE;;;;;;;;;;;;GAYG;AACH,wBAAgB,4BAA4B,CAAC,IAAI,EAAE,iBAAiB,GAAG,mBAAmB,CAezF;AAED,qBAAa,iBAAiB;IAC7B,OAAO,CAAC,KAAK,CAAc;gBAEf,KAAK,EAAE,YAAY;IAIzB,MAAM,CACX,MAAM,EAAE,cAAc,EACtB,SAAS,EAAE,MAAM,EACjB,KAAK,CAAC,EAAE;QACP,WAAW,CAAC,EAAE,KAAK,CAAC;YAAE,UAAU,EAAE,MAAM,CAAC;YAAC,QAAQ,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,OAAO,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC,CAAA;QAC7F,WAAW,CAAC,EAAE,gBAAgB,EAAE,CAAA;QAChC,UAAU,CAAC,EAAE,cAAc,CAAA;KAC3B,GACC,OAAO,CAAC,mBAAmB,CAAC;IAsBzB,OAAO,CAAC,YAAY,EAAE,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAQjE,IAAI,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAI5C;;;;;OAKG;IACG,WAAW,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAK7C,KAAK,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAW5C,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,iBAAiB;CAajF"}
1
+ {"version":3,"file":"checkpoint.d.ts","sourceRoot":"","sources":["../../../src/runtime/query/checkpoint.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAA;AACtE,OAAO,KAAK,EACX,cAAc,EACd,gBAAgB,EAChB,YAAY,EACZ,iBAAiB,EACjB,mBAAmB,EACnB,MAAM,2BAA2B,CAAA;AAElC,OAAO,KAAK,EAAE,kBAAkB,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAA;AAC9F,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAA;AACrE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAA;AAKpE;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,EAAE,EAAE,mBAAmB,GAAG,mBAAmB,CAQlF;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,4BAA4B,CAAC,IAAI,EAAE,iBAAiB,GAAG,mBAAmB,CAezF;AAED,qBAAa,iBAAiB;IAC7B,OAAO,CAAC,KAAK,CAAiB;IAC9B,OAAO,CAAC,KAAK,CAAoB;IAEjC;;;;;;OAMG;gBACS,KAAK,EAAE,eAAe,EAAE,KAAK,EAAE,kBAAkB;IAKvD,MAAM,CACX,MAAM,EAAE,cAAc,EACtB,SAAS,EAAE,MAAM,EACjB,KAAK,CAAC,EAAE;QACP,WAAW,CAAC,EAAE,KAAK,CAAC;YAAE,UAAU,EAAE,MAAM,CAAC;YAAC,QAAQ,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,OAAO,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC,CAAA;QAC7F,WAAW,CAAC,EAAE,gBAAgB,EAAE,CAAA;QAChC,UAAU,CAAC,EAAE,cAAc,CAAA;KAC3B,GACC,OAAO,CAAC,mBAAmB,CAAC;IAsBzB,OAAO,CAAC,YAAY,EAAE,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAQjE,IAAI,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAI5C;;;;;OAKG;IACG,WAAW,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAK7C,KAAK,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAW5C,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,iBAAiB;CAajF"}