@stigmer/runner 3.0.1

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 (904) hide show
  1. package/README.md +212 -0
  2. package/dist/.build-fingerprint +1 -0
  3. package/dist/activities/call-agent-status.d.ts +37 -0
  4. package/dist/activities/call-agent-status.js +91 -0
  5. package/dist/activities/call-agent-status.js.map +1 -0
  6. package/dist/activities/call-agent.d.ts +25 -0
  7. package/dist/activities/call-agent.js +233 -0
  8. package/dist/activities/call-agent.js.map +1 -0
  9. package/dist/activities/call-eval.d.ts +50 -0
  10. package/dist/activities/call-eval.js +244 -0
  11. package/dist/activities/call-eval.js.map +1 -0
  12. package/dist/activities/call-function.d.ts +21 -0
  13. package/dist/activities/call-function.js +54 -0
  14. package/dist/activities/call-function.js.map +1 -0
  15. package/dist/activities/call-grpc.d.ts +22 -0
  16. package/dist/activities/call-grpc.js +101 -0
  17. package/dist/activities/call-grpc.js.map +1 -0
  18. package/dist/activities/call-http.d.ts +32 -0
  19. package/dist/activities/call-http.js +134 -0
  20. package/dist/activities/call-http.js.map +1 -0
  21. package/dist/activities/call-llm.d.ts +39 -0
  22. package/dist/activities/call-llm.js +260 -0
  23. package/dist/activities/call-llm.js.map +1 -0
  24. package/dist/activities/call-transform.d.ts +20 -0
  25. package/dist/activities/call-transform.js +33 -0
  26. package/dist/activities/call-transform.js.map +1 -0
  27. package/dist/activities/call-validate.d.ts +41 -0
  28. package/dist/activities/call-validate.js +163 -0
  29. package/dist/activities/call-validate.js.map +1 -0
  30. package/dist/activities/classify-tool-approvals.d.ts +47 -0
  31. package/dist/activities/classify-tool-approvals.js +207 -0
  32. package/dist/activities/classify-tool-approvals.js.map +1 -0
  33. package/dist/activities/discover-mcp-server.d.ts +65 -0
  34. package/dist/activities/discover-mcp-server.js +269 -0
  35. package/dist/activities/discover-mcp-server.js.map +1 -0
  36. package/dist/activities/emit-event.d.ts +46 -0
  37. package/dist/activities/emit-event.js +125 -0
  38. package/dist/activities/emit-event.js.map +1 -0
  39. package/dist/activities/ensure-thread.d.ts +24 -0
  40. package/dist/activities/ensure-thread.js +44 -0
  41. package/dist/activities/ensure-thread.js.map +1 -0
  42. package/dist/activities/evaluate-expressions.d.ts +21 -0
  43. package/dist/activities/evaluate-expressions.js +39 -0
  44. package/dist/activities/evaluate-expressions.js.map +1 -0
  45. package/dist/activities/execute-cursor/approval-policy.d.ts +104 -0
  46. package/dist/activities/execute-cursor/approval-policy.js +193 -0
  47. package/dist/activities/execute-cursor/approval-policy.js.map +1 -0
  48. package/dist/activities/execute-cursor/approval-state.d.ts +157 -0
  49. package/dist/activities/execute-cursor/approval-state.js +223 -0
  50. package/dist/activities/execute-cursor/approval-state.js.map +1 -0
  51. package/dist/activities/execute-cursor/attachment-resolver.d.ts +19 -0
  52. package/dist/activities/execute-cursor/attachment-resolver.js +52 -0
  53. package/dist/activities/execute-cursor/attachment-resolver.js.map +1 -0
  54. package/dist/activities/execute-cursor/blueprint-resolver.d.ts +66 -0
  55. package/dist/activities/execute-cursor/blueprint-resolver.js +169 -0
  56. package/dist/activities/execute-cursor/blueprint-resolver.js.map +1 -0
  57. package/dist/activities/execute-cursor/connect-backfill.d.ts +18 -0
  58. package/dist/activities/execute-cursor/connect-backfill.js +27 -0
  59. package/dist/activities/execute-cursor/connect-backfill.js.map +1 -0
  60. package/dist/activities/execute-cursor/cursor-event-recorder.d.ts +24 -0
  61. package/dist/activities/execute-cursor/cursor-event-recorder.js +66 -0
  62. package/dist/activities/execute-cursor/cursor-event-recorder.js.map +1 -0
  63. package/dist/activities/execute-cursor/cursor-mode.d.ts +32 -0
  64. package/dist/activities/execute-cursor/cursor-mode.js +36 -0
  65. package/dist/activities/execute-cursor/cursor-mode.js.map +1 -0
  66. package/dist/activities/execute-cursor/delta-enricher.d.ts +87 -0
  67. package/dist/activities/execute-cursor/delta-enricher.js +265 -0
  68. package/dist/activities/execute-cursor/delta-enricher.js.map +1 -0
  69. package/dist/activities/execute-cursor/env-resolver.d.ts +19 -0
  70. package/dist/activities/execute-cursor/env-resolver.js +47 -0
  71. package/dist/activities/execute-cursor/env-resolver.js.map +1 -0
  72. package/dist/activities/execute-cursor/error-classifier.d.ts +73 -0
  73. package/dist/activities/execute-cursor/error-classifier.js +185 -0
  74. package/dist/activities/execute-cursor/error-classifier.js.map +1 -0
  75. package/dist/activities/execute-cursor/fetch-interceptor.d.ts +59 -0
  76. package/dist/activities/execute-cursor/fetch-interceptor.js +317 -0
  77. package/dist/activities/execute-cursor/fetch-interceptor.js.map +1 -0
  78. package/dist/activities/execute-cursor/hitl-diagnostics.d.ts +32 -0
  79. package/dist/activities/execute-cursor/hitl-diagnostics.js +73 -0
  80. package/dist/activities/execute-cursor/hitl-diagnostics.js.map +1 -0
  81. package/dist/activities/execute-cursor/hook-script.d.ts +47 -0
  82. package/dist/activities/execute-cursor/hook-script.js +156 -0
  83. package/dist/activities/execute-cursor/hook-script.js.map +1 -0
  84. package/dist/activities/execute-cursor/http2-interceptor.d.ts +94 -0
  85. package/dist/activities/execute-cursor/http2-interceptor.js +209 -0
  86. package/dist/activities/execute-cursor/http2-interceptor.js.map +1 -0
  87. package/dist/activities/execute-cursor/index.d.ts +67 -0
  88. package/dist/activities/execute-cursor/index.js +1176 -0
  89. package/dist/activities/execute-cursor/index.js.map +1 -0
  90. package/dist/activities/execute-cursor/mcp-config.d.ts +30 -0
  91. package/dist/activities/execute-cursor/mcp-config.js +39 -0
  92. package/dist/activities/execute-cursor/mcp-config.js.map +1 -0
  93. package/dist/activities/execute-cursor/mcp-resolver.d.ts +91 -0
  94. package/dist/activities/execute-cursor/mcp-resolver.js +178 -0
  95. package/dist/activities/execute-cursor/mcp-resolver.js.map +1 -0
  96. package/dist/activities/execute-cursor/message-translator.d.ts +211 -0
  97. package/dist/activities/execute-cursor/message-translator.js +786 -0
  98. package/dist/activities/execute-cursor/message-translator.js.map +1 -0
  99. package/dist/activities/execute-cursor/model-pricing-data.d.ts +40 -0
  100. package/dist/activities/execute-cursor/model-pricing-data.js +114 -0
  101. package/dist/activities/execute-cursor/model-pricing-data.js.map +1 -0
  102. package/dist/activities/execute-cursor/model-pricing.d.ts +42 -0
  103. package/dist/activities/execute-cursor/model-pricing.js +141 -0
  104. package/dist/activities/execute-cursor/model-pricing.js.map +1 -0
  105. package/dist/activities/execute-cursor/placeholder-resolver.d.ts +34 -0
  106. package/dist/activities/execute-cursor/placeholder-resolver.js +82 -0
  107. package/dist/activities/execute-cursor/placeholder-resolver.js.map +1 -0
  108. package/dist/activities/execute-cursor/prompt-builder.d.ts +80 -0
  109. package/dist/activities/execute-cursor/prompt-builder.js +280 -0
  110. package/dist/activities/execute-cursor/prompt-builder.js.map +1 -0
  111. package/dist/activities/execute-cursor/rejection-capture.d.ts +30 -0
  112. package/dist/activities/execute-cursor/rejection-capture.js +80 -0
  113. package/dist/activities/execute-cursor/rejection-capture.js.map +1 -0
  114. package/dist/activities/execute-cursor/session-lifecycle.d.ts +189 -0
  115. package/dist/activities/execute-cursor/session-lifecycle.js +285 -0
  116. package/dist/activities/execute-cursor/session-lifecycle.js.map +1 -0
  117. package/dist/activities/execute-cursor/skill-resolver.d.ts +29 -0
  118. package/dist/activities/execute-cursor/skill-resolver.js +134 -0
  119. package/dist/activities/execute-cursor/skill-resolver.js.map +1 -0
  120. package/dist/activities/execute-cursor/subagent-config.d.ts +34 -0
  121. package/dist/activities/execute-cursor/subagent-config.js +55 -0
  122. package/dist/activities/execute-cursor/subagent-config.js.map +1 -0
  123. package/dist/activities/execute-cursor/todo-tracker.d.ts +35 -0
  124. package/dist/activities/execute-cursor/todo-tracker.js +108 -0
  125. package/dist/activities/execute-cursor/todo-tracker.js.map +1 -0
  126. package/dist/activities/execute-cursor/usage-accumulator.d.ts +55 -0
  127. package/dist/activities/execute-cursor/usage-accumulator.js +89 -0
  128. package/dist/activities/execute-cursor/usage-accumulator.js.map +1 -0
  129. package/dist/activities/execute-cursor/workspace-provision.d.ts +22 -0
  130. package/dist/activities/execute-cursor/workspace-provision.js +37 -0
  131. package/dist/activities/execute-cursor/workspace-provision.js.map +1 -0
  132. package/dist/activities/execute-cursor/workspace-setup.d.ts +27 -0
  133. package/dist/activities/execute-cursor/workspace-setup.js +61 -0
  134. package/dist/activities/execute-cursor/workspace-setup.js.map +1 -0
  135. package/dist/activities/execute-deep-agent/__test-utils__/v3-event-fixtures.d.ts +71 -0
  136. package/dist/activities/execute-deep-agent/__test-utils__/v3-event-fixtures.js +182 -0
  137. package/dist/activities/execute-deep-agent/__test-utils__/v3-event-fixtures.js.map +1 -0
  138. package/dist/activities/execute-deep-agent/attachment-injector.d.ts +67 -0
  139. package/dist/activities/execute-deep-agent/attachment-injector.js +306 -0
  140. package/dist/activities/execute-deep-agent/attachment-injector.js.map +1 -0
  141. package/dist/activities/execute-deep-agent/auto-publish.d.ts +17 -0
  142. package/dist/activities/execute-deep-agent/auto-publish.js +71 -0
  143. package/dist/activities/execute-deep-agent/auto-publish.js.map +1 -0
  144. package/dist/activities/execute-deep-agent/environment.d.ts +24 -0
  145. package/dist/activities/execute-deep-agent/environment.js +50 -0
  146. package/dist/activities/execute-deep-agent/environment.js.map +1 -0
  147. package/dist/activities/execute-deep-agent/event-recorder.d.ts +21 -0
  148. package/dist/activities/execute-deep-agent/event-recorder.js +67 -0
  149. package/dist/activities/execute-deep-agent/event-recorder.js.map +1 -0
  150. package/dist/activities/execute-deep-agent/execution-state.d.ts +61 -0
  151. package/dist/activities/execute-deep-agent/execution-state.js +76 -0
  152. package/dist/activities/execute-deep-agent/execution-state.js.map +1 -0
  153. package/dist/activities/execute-deep-agent/execution-status-writer.d.ts +17 -0
  154. package/dist/activities/execute-deep-agent/execution-status-writer.js +9 -0
  155. package/dist/activities/execute-deep-agent/execution-status-writer.js.map +1 -0
  156. package/dist/activities/execute-deep-agent/hitl.d.ts +58 -0
  157. package/dist/activities/execute-deep-agent/hitl.js +155 -0
  158. package/dist/activities/execute-deep-agent/hitl.js.map +1 -0
  159. package/dist/activities/execute-deep-agent/index.d.ts +14 -0
  160. package/dist/activities/execute-deep-agent/index.js +286 -0
  161. package/dist/activities/execute-deep-agent/index.js.map +1 -0
  162. package/dist/activities/execute-deep-agent/inline-publisher.d.ts +36 -0
  163. package/dist/activities/execute-deep-agent/inline-publisher.js +105 -0
  164. package/dist/activities/execute-deep-agent/inline-publisher.js.map +1 -0
  165. package/dist/activities/execute-deep-agent/post-stream.d.ts +23 -0
  166. package/dist/activities/execute-deep-agent/post-stream.js +71 -0
  167. package/dist/activities/execute-deep-agent/post-stream.js.map +1 -0
  168. package/dist/activities/execute-deep-agent/prompt-builder.d.ts +27 -0
  169. package/dist/activities/execute-deep-agent/prompt-builder.js +200 -0
  170. package/dist/activities/execute-deep-agent/prompt-builder.js.map +1 -0
  171. package/dist/activities/execute-deep-agent/setup.d.ts +55 -0
  172. package/dist/activities/execute-deep-agent/setup.js +411 -0
  173. package/dist/activities/execute-deep-agent/setup.js.map +1 -0
  174. package/dist/activities/execute-deep-agent/status-builder-shared.d.ts +39 -0
  175. package/dist/activities/execute-deep-agent/status-builder-shared.js +120 -0
  176. package/dist/activities/execute-deep-agent/status-builder-shared.js.map +1 -0
  177. package/dist/activities/execute-deep-agent/status-builder.d.ts +81 -0
  178. package/dist/activities/execute-deep-agent/status-builder.js +312 -0
  179. package/dist/activities/execute-deep-agent/status-builder.js.map +1 -0
  180. package/dist/activities/execute-deep-agent/streaming-scheduler.d.ts +60 -0
  181. package/dist/activities/execute-deep-agent/streaming-scheduler.js +114 -0
  182. package/dist/activities/execute-deep-agent/streaming-scheduler.js.map +1 -0
  183. package/dist/activities/execute-deep-agent/streaming-side-effects.d.ts +22 -0
  184. package/dist/activities/execute-deep-agent/streaming-side-effects.js +83 -0
  185. package/dist/activities/execute-deep-agent/streaming-side-effects.js.map +1 -0
  186. package/dist/activities/execute-deep-agent/streaming-terminal.d.ts +12 -0
  187. package/dist/activities/execute-deep-agent/streaming-terminal.js +71 -0
  188. package/dist/activities/execute-deep-agent/streaming-terminal.js.map +1 -0
  189. package/dist/activities/execute-deep-agent/streaming-v3.d.ts +13 -0
  190. package/dist/activities/execute-deep-agent/streaming-v3.js +170 -0
  191. package/dist/activities/execute-deep-agent/streaming-v3.js.map +1 -0
  192. package/dist/activities/execute-deep-agent/streaming.d.ts +66 -0
  193. package/dist/activities/execute-deep-agent/streaming.js +169 -0
  194. package/dist/activities/execute-deep-agent/streaming.js.map +1 -0
  195. package/dist/activities/execute-deep-agent/subagent-tracker.d.ts +90 -0
  196. package/dist/activities/execute-deep-agent/subagent-tracker.js +364 -0
  197. package/dist/activities/execute-deep-agent/subagent-tracker.js.map +1 -0
  198. package/dist/activities/execute-deep-agent/subagent-transformer.d.ts +150 -0
  199. package/dist/activities/execute-deep-agent/subagent-transformer.js +450 -0
  200. package/dist/activities/execute-deep-agent/subagent-transformer.js.map +1 -0
  201. package/dist/activities/execute-deep-agent/subagent-wiring.d.ts +28 -0
  202. package/dist/activities/execute-deep-agent/subagent-wiring.js +40 -0
  203. package/dist/activities/execute-deep-agent/subagent-wiring.js.map +1 -0
  204. package/dist/activities/execute-deep-agent/v3-event-recorder.d.ts +31 -0
  205. package/dist/activities/execute-deep-agent/v3-event-recorder.js +71 -0
  206. package/dist/activities/execute-deep-agent/v3-event-recorder.js.map +1 -0
  207. package/dist/activities/execute-deep-agent/v3-events.d.ts +102 -0
  208. package/dist/activities/execute-deep-agent/v3-events.js +35 -0
  209. package/dist/activities/execute-deep-agent/v3-events.js.map +1 -0
  210. package/dist/activities/execute-deep-agent/v3-protocol-normalizer.d.ts +15 -0
  211. package/dist/activities/execute-deep-agent/v3-protocol-normalizer.js +235 -0
  212. package/dist/activities/execute-deep-agent/v3-protocol-normalizer.js.map +1 -0
  213. package/dist/activities/execute-deep-agent/v3-status-builder.d.ts +68 -0
  214. package/dist/activities/execute-deep-agent/v3-status-builder.js +394 -0
  215. package/dist/activities/execute-deep-agent/v3-status-builder.js.map +1 -0
  216. package/dist/activities/execute-deep-agent/writeback-coordinator.d.ts +71 -0
  217. package/dist/activities/execute-deep-agent/writeback-coordinator.js +295 -0
  218. package/dist/activities/execute-deep-agent/writeback-coordinator.js.map +1 -0
  219. package/dist/activities/hydrate-workflow-execution.d.ts +32 -0
  220. package/dist/activities/hydrate-workflow-execution.js +212 -0
  221. package/dist/activities/hydrate-workflow-execution.js.map +1 -0
  222. package/dist/activities/notification.d.ts +19 -0
  223. package/dist/activities/notification.js +47 -0
  224. package/dist/activities/notification.js.map +1 -0
  225. package/dist/activities/promote-task-output.d.ts +38 -0
  226. package/dist/activities/promote-task-output.js +90 -0
  227. package/dist/activities/promote-task-output.js.map +1 -0
  228. package/dist/activities/run-command.d.ts +15 -0
  229. package/dist/activities/run-command.js +123 -0
  230. package/dist/activities/run-command.js.map +1 -0
  231. package/dist/activities/workflow-event-activities.d.ts +48 -0
  232. package/dist/activities/workflow-event-activities.js +415 -0
  233. package/dist/activities/workflow-event-activities.js.map +1 -0
  234. package/dist/bootstrap.d.ts +80 -0
  235. package/dist/bootstrap.js +114 -0
  236. package/dist/bootstrap.js.map +1 -0
  237. package/dist/budget/index.d.ts +1 -0
  238. package/dist/budget/index.js +2 -0
  239. package/dist/budget/index.js.map +1 -0
  240. package/dist/budget/tracker.d.ts +52 -0
  241. package/dist/budget/tracker.js +123 -0
  242. package/dist/budget/tracker.js.map +1 -0
  243. package/dist/claimcheck/compressor.d.ts +2 -0
  244. package/dist/claimcheck/compressor.js +8 -0
  245. package/dist/claimcheck/compressor.js.map +1 -0
  246. package/dist/claimcheck/config.d.ts +7 -0
  247. package/dist/claimcheck/config.js +10 -0
  248. package/dist/claimcheck/config.js.map +1 -0
  249. package/dist/claimcheck/index.d.ts +3 -0
  250. package/dist/claimcheck/index.js +4 -0
  251. package/dist/claimcheck/index.js.map +1 -0
  252. package/dist/claimcheck/payload-codec.d.ts +23 -0
  253. package/dist/claimcheck/payload-codec.js +81 -0
  254. package/dist/claimcheck/payload-codec.js.map +1 -0
  255. package/dist/client/server-contracts.d.ts +52 -0
  256. package/dist/client/server-contracts.js +72 -0
  257. package/dist/client/server-contracts.js.map +1 -0
  258. package/dist/client/stigmer-client.d.ts +131 -0
  259. package/dist/client/stigmer-client.js +239 -0
  260. package/dist/client/stigmer-client.js.map +1 -0
  261. package/dist/config.d.ts +64 -0
  262. package/dist/config.js +123 -0
  263. package/dist/config.js.map +1 -0
  264. package/dist/idle-watchdog.d.ts +11 -0
  265. package/dist/idle-watchdog.js +24 -0
  266. package/dist/idle-watchdog.js.map +1 -0
  267. package/dist/index.d.ts +43 -0
  268. package/dist/index.js +42 -0
  269. package/dist/index.js.map +1 -0
  270. package/dist/interceptors/workflow-metrics-sink.d.ts +11 -0
  271. package/dist/interceptors/workflow-metrics-sink.js +51 -0
  272. package/dist/interceptors/workflow-metrics-sink.js.map +1 -0
  273. package/dist/ipc-protocol-fixtures.d.ts +32 -0
  274. package/dist/ipc-protocol-fixtures.js +69 -0
  275. package/dist/ipc-protocol-fixtures.js.map +1 -0
  276. package/dist/ipc-protocol.d.ts +60 -0
  277. package/dist/ipc-protocol.js +19 -0
  278. package/dist/ipc-protocol.js.map +1 -0
  279. package/dist/main.d.ts +19 -0
  280. package/dist/main.js +292 -0
  281. package/dist/main.js.map +1 -0
  282. package/dist/middleware/approval-gate.d.ts +30 -0
  283. package/dist/middleware/approval-gate.js +125 -0
  284. package/dist/middleware/approval-gate.js.map +1 -0
  285. package/dist/middleware/cost-cap.d.ts +22 -0
  286. package/dist/middleware/cost-cap.js +159 -0
  287. package/dist/middleware/cost-cap.js.map +1 -0
  288. package/dist/middleware/error-hints.d.ts +27 -0
  289. package/dist/middleware/error-hints.js +116 -0
  290. package/dist/middleware/error-hints.js.map +1 -0
  291. package/dist/middleware/execution-budget.d.ts +20 -0
  292. package/dist/middleware/execution-budget.js +151 -0
  293. package/dist/middleware/execution-budget.js.map +1 -0
  294. package/dist/middleware/graceful-stop.d.ts +17 -0
  295. package/dist/middleware/graceful-stop.js +63 -0
  296. package/dist/middleware/graceful-stop.js.map +1 -0
  297. package/dist/middleware/index.d.ts +27 -0
  298. package/dist/middleware/index.js +45 -0
  299. package/dist/middleware/index.js.map +1 -0
  300. package/dist/middleware/loop-detection.d.ts +14 -0
  301. package/dist/middleware/loop-detection.js +156 -0
  302. package/dist/middleware/loop-detection.js.map +1 -0
  303. package/dist/middleware/otel-spans.d.ts +11 -0
  304. package/dist/middleware/otel-spans.js +177 -0
  305. package/dist/middleware/otel-spans.js.map +1 -0
  306. package/dist/middleware/think-tool.d.ts +23 -0
  307. package/dist/middleware/think-tool.js +33 -0
  308. package/dist/middleware/think-tool.js.map +1 -0
  309. package/dist/middleware/tool-truncation.d.ts +16 -0
  310. package/dist/middleware/tool-truncation.js +67 -0
  311. package/dist/middleware/tool-truncation.js.map +1 -0
  312. package/dist/middleware/types.d.ts +100 -0
  313. package/dist/middleware/types.js +9 -0
  314. package/dist/middleware/types.js.map +1 -0
  315. package/dist/notification/index.d.ts +2 -0
  316. package/dist/notification/index.js +6 -0
  317. package/dist/notification/index.js.map +1 -0
  318. package/dist/notification/provider.d.ts +29 -0
  319. package/dist/notification/provider.js +25 -0
  320. package/dist/notification/provider.js.map +1 -0
  321. package/dist/notification/webhook.d.ts +13 -0
  322. package/dist/notification/webhook.js +55 -0
  323. package/dist/notification/webhook.js.map +1 -0
  324. package/dist/otel-metrics.d.ts +21 -0
  325. package/dist/otel-metrics.js +54 -0
  326. package/dist/otel-metrics.js.map +1 -0
  327. package/dist/otel.d.ts +57 -0
  328. package/dist/otel.js +164 -0
  329. package/dist/otel.js.map +1 -0
  330. package/dist/runner-manager.d.ts +113 -0
  331. package/dist/runner-manager.js +412 -0
  332. package/dist/runner-manager.js.map +1 -0
  333. package/dist/runner-token-coordinator.d.ts +56 -0
  334. package/dist/runner-token-coordinator.js +84 -0
  335. package/dist/runner-token-coordinator.js.map +1 -0
  336. package/dist/runner.d.ts +104 -0
  337. package/dist/runner.js +234 -0
  338. package/dist/runner.js.map +1 -0
  339. package/dist/shared/approval-policy.d.ts +45 -0
  340. package/dist/shared/approval-policy.js +122 -0
  341. package/dist/shared/approval-policy.js.map +1 -0
  342. package/dist/shared/artifact-storage.d.ts +44 -0
  343. package/dist/shared/artifact-storage.js +162 -0
  344. package/dist/shared/artifact-storage.js.map +1 -0
  345. package/dist/shared/checkpointer/factory.d.ts +28 -0
  346. package/dist/shared/checkpointer/factory.js +55 -0
  347. package/dist/shared/checkpointer/factory.js.map +1 -0
  348. package/dist/shared/checkpointer/http-saver.d.ts +34 -0
  349. package/dist/shared/checkpointer/http-saver.js +274 -0
  350. package/dist/shared/checkpointer/http-saver.js.map +1 -0
  351. package/dist/shared/checkpointer/types.d.ts +12 -0
  352. package/dist/shared/checkpointer/types.js +2 -0
  353. package/dist/shared/checkpointer/types.js.map +1 -0
  354. package/dist/shared/connect-backfill.d.ts +58 -0
  355. package/dist/shared/connect-backfill.js +119 -0
  356. package/dist/shared/connect-backfill.js.map +1 -0
  357. package/dist/shared/extract-json.d.ts +26 -0
  358. package/dist/shared/extract-json.js +140 -0
  359. package/dist/shared/extract-json.js.map +1 -0
  360. package/dist/shared/grpc-retry.d.ts +35 -0
  361. package/dist/shared/grpc-retry.js +78 -0
  362. package/dist/shared/grpc-retry.js.map +1 -0
  363. package/dist/shared/heartbeat.d.ts +22 -0
  364. package/dist/shared/heartbeat.js +55 -0
  365. package/dist/shared/heartbeat.js.map +1 -0
  366. package/dist/shared/json-schema-to-zod.d.ts +13 -0
  367. package/dist/shared/json-schema-to-zod.js +49 -0
  368. package/dist/shared/json-schema-to-zod.js.map +1 -0
  369. package/dist/shared/llm-proxy.d.ts +57 -0
  370. package/dist/shared/llm-proxy.js +116 -0
  371. package/dist/shared/llm-proxy.js.map +1 -0
  372. package/dist/shared/mcp-manager.d.ts +47 -0
  373. package/dist/shared/mcp-manager.js +118 -0
  374. package/dist/shared/mcp-manager.js.map +1 -0
  375. package/dist/shared/mcp-resolver.d.ts +41 -0
  376. package/dist/shared/mcp-resolver.js +96 -0
  377. package/dist/shared/mcp-resolver.js.map +1 -0
  378. package/dist/shared/model-pricing-data.d.ts +18 -0
  379. package/dist/shared/model-pricing-data.js +78 -0
  380. package/dist/shared/model-pricing-data.js.map +1 -0
  381. package/dist/shared/model-pricing.d.ts +24 -0
  382. package/dist/shared/model-pricing.js +58 -0
  383. package/dist/shared/model-pricing.js.map +1 -0
  384. package/dist/shared/model-registry.d.ts +55 -0
  385. package/dist/shared/model-registry.js +178 -0
  386. package/dist/shared/model-registry.js.map +1 -0
  387. package/dist/shared/placeholder-resolver.d.ts +27 -0
  388. package/dist/shared/placeholder-resolver.js +75 -0
  389. package/dist/shared/placeholder-resolver.js.map +1 -0
  390. package/dist/shared/plan-artifact.d.ts +56 -0
  391. package/dist/shared/plan-artifact.js +98 -0
  392. package/dist/shared/plan-artifact.js.map +1 -0
  393. package/dist/shared/skill-relevance.d.ts +65 -0
  394. package/dist/shared/skill-relevance.js +175 -0
  395. package/dist/shared/skill-relevance.js.map +1 -0
  396. package/dist/shared/skill-writer.d.ts +73 -0
  397. package/dist/shared/skill-writer.js +230 -0
  398. package/dist/shared/skill-writer.js.map +1 -0
  399. package/dist/shared/status.d.ts +37 -0
  400. package/dist/shared/status.js +73 -0
  401. package/dist/shared/status.js.map +1 -0
  402. package/dist/shared/subagent-gate.d.ts +41 -0
  403. package/dist/shared/subagent-gate.js +93 -0
  404. package/dist/shared/subagent-gate.js.map +1 -0
  405. package/dist/shared/tool-kind.d.ts +22 -0
  406. package/dist/shared/tool-kind.js +79 -0
  407. package/dist/shared/tool-kind.js.map +1 -0
  408. package/dist/shared/workspace/file-tree.d.ts +13 -0
  409. package/dist/shared/workspace/file-tree.js +101 -0
  410. package/dist/shared/workspace/file-tree.js.map +1 -0
  411. package/dist/shared/workspace/local-backend.d.ts +41 -0
  412. package/dist/shared/workspace/local-backend.js +113 -0
  413. package/dist/shared/workspace/local-backend.js.map +1 -0
  414. package/dist/shared/workspace/platform-dir.d.ts +25 -0
  415. package/dist/shared/workspace/platform-dir.js +36 -0
  416. package/dist/shared/workspace/platform-dir.js.map +1 -0
  417. package/dist/shared/workspace/platform-mount.d.ts +95 -0
  418. package/dist/shared/workspace/platform-mount.js +157 -0
  419. package/dist/shared/workspace/platform-mount.js.map +1 -0
  420. package/dist/shared/workspace/provisioner.d.ts +47 -0
  421. package/dist/shared/workspace/provisioner.js +84 -0
  422. package/dist/shared/workspace/provisioner.js.map +1 -0
  423. package/dist/shared/workspace/sources/empty.d.ts +8 -0
  424. package/dist/shared/workspace/sources/empty.js +18 -0
  425. package/dist/shared/workspace/sources/empty.js.map +1 -0
  426. package/dist/shared/workspace/sources/git.d.ts +22 -0
  427. package/dist/shared/workspace/sources/git.js +207 -0
  428. package/dist/shared/workspace/sources/git.js.map +1 -0
  429. package/dist/shared/workspace/sources/local-path.d.ts +17 -0
  430. package/dist/shared/workspace/sources/local-path.js +57 -0
  431. package/dist/shared/workspace/sources/local-path.js.map +1 -0
  432. package/dist/shared/workspace/types.d.ts +58 -0
  433. package/dist/shared/workspace/types.js +25 -0
  434. package/dist/shared/workspace/types.js.map +1 -0
  435. package/dist/shared/zip-extract.d.ts +30 -0
  436. package/dist/shared/zip-extract.js +150 -0
  437. package/dist/shared/zip-extract.js.map +1 -0
  438. package/dist/worker.d.ts +27 -0
  439. package/dist/worker.js +65 -0
  440. package/dist/worker.js.map +1 -0
  441. package/dist/workflow-engine/clone.d.ts +11 -0
  442. package/dist/workflow-engine/clone.js +21 -0
  443. package/dist/workflow-engine/clone.js.map +1 -0
  444. package/dist/workflow-engine/do-executor.d.ts +27 -0
  445. package/dist/workflow-engine/do-executor.js +418 -0
  446. package/dist/workflow-engine/do-executor.js.map +1 -0
  447. package/dist/workflow-engine/duration.d.ts +12 -0
  448. package/dist/workflow-engine/duration.js +25 -0
  449. package/dist/workflow-engine/duration.js.map +1 -0
  450. package/dist/workflow-engine/error-utils.d.ts +42 -0
  451. package/dist/workflow-engine/error-utils.js +77 -0
  452. package/dist/workflow-engine/error-utils.js.map +1 -0
  453. package/dist/workflow-engine/errors.d.ts +46 -0
  454. package/dist/workflow-engine/errors.js +105 -0
  455. package/dist/workflow-engine/errors.js.map +1 -0
  456. package/dist/workflow-engine/expression-utils.d.ts +60 -0
  457. package/dist/workflow-engine/expression-utils.js +108 -0
  458. package/dist/workflow-engine/expression-utils.js.map +1 -0
  459. package/dist/workflow-engine/expression.d.ts +132 -0
  460. package/dist/workflow-engine/expression.js +366 -0
  461. package/dist/workflow-engine/expression.js.map +1 -0
  462. package/dist/workflow-engine/loader.d.ts +23 -0
  463. package/dist/workflow-engine/loader.js +429 -0
  464. package/dist/workflow-engine/loader.js.map +1 -0
  465. package/dist/workflow-engine/recovery.d.ts +53 -0
  466. package/dist/workflow-engine/recovery.js +46 -0
  467. package/dist/workflow-engine/recovery.js.map +1 -0
  468. package/dist/workflow-engine/resolve.d.ts +83 -0
  469. package/dist/workflow-engine/resolve.js +257 -0
  470. package/dist/workflow-engine/resolve.js.map +1 -0
  471. package/dist/workflow-engine/retry.d.ts +30 -0
  472. package/dist/workflow-engine/retry.js +97 -0
  473. package/dist/workflow-engine/retry.js.map +1 -0
  474. package/dist/workflow-engine/state.d.ts +26 -0
  475. package/dist/workflow-engine/state.js +49 -0
  476. package/dist/workflow-engine/state.js.map +1 -0
  477. package/dist/workflow-engine/task-factory.d.ts +20 -0
  478. package/dist/workflow-engine/task-factory.js +133 -0
  479. package/dist/workflow-engine/task-factory.js.map +1 -0
  480. package/dist/workflow-engine/task-status-accumulator.d.ts +59 -0
  481. package/dist/workflow-engine/task-status-accumulator.js +164 -0
  482. package/dist/workflow-engine/task-status-accumulator.js.map +1 -0
  483. package/dist/workflow-engine/tasks/call-agent-output.d.ts +26 -0
  484. package/dist/workflow-engine/tasks/call-agent-output.js +109 -0
  485. package/dist/workflow-engine/tasks/call-agent-output.js.map +1 -0
  486. package/dist/workflow-engine/tasks/call-agent.d.ts +31 -0
  487. package/dist/workflow-engine/tasks/call-agent.js +161 -0
  488. package/dist/workflow-engine/tasks/call-agent.js.map +1 -0
  489. package/dist/workflow-engine/tasks/call-function.d.ts +19 -0
  490. package/dist/workflow-engine/tasks/call-function.js +64 -0
  491. package/dist/workflow-engine/tasks/call-function.js.map +1 -0
  492. package/dist/workflow-engine/tasks/call-grpc.d.ts +15 -0
  493. package/dist/workflow-engine/tasks/call-grpc.js +27 -0
  494. package/dist/workflow-engine/tasks/call-grpc.js.map +1 -0
  495. package/dist/workflow-engine/tasks/call-http.d.ts +19 -0
  496. package/dist/workflow-engine/tasks/call-http.js +31 -0
  497. package/dist/workflow-engine/tasks/call-http.js.map +1 -0
  498. package/dist/workflow-engine/tasks/for.d.ts +39 -0
  499. package/dist/workflow-engine/tasks/for.js +154 -0
  500. package/dist/workflow-engine/tasks/for.js.map +1 -0
  501. package/dist/workflow-engine/tasks/fork.d.ts +42 -0
  502. package/dist/workflow-engine/tasks/fork.js +142 -0
  503. package/dist/workflow-engine/tasks/fork.js.map +1 -0
  504. package/dist/workflow-engine/tasks/human-input.d.ts +33 -0
  505. package/dist/workflow-engine/tasks/human-input.js +109 -0
  506. package/dist/workflow-engine/tasks/human-input.js.map +1 -0
  507. package/dist/workflow-engine/tasks/listen.d.ts +34 -0
  508. package/dist/workflow-engine/tasks/listen.js +119 -0
  509. package/dist/workflow-engine/tasks/listen.js.map +1 -0
  510. package/dist/workflow-engine/tasks/raise.d.ts +18 -0
  511. package/dist/workflow-engine/tasks/raise.js +60 -0
  512. package/dist/workflow-engine/tasks/raise.js.map +1 -0
  513. package/dist/workflow-engine/tasks/run.d.ts +39 -0
  514. package/dist/workflow-engine/tasks/run.js +114 -0
  515. package/dist/workflow-engine/tasks/run.js.map +1 -0
  516. package/dist/workflow-engine/tasks/set.d.ts +15 -0
  517. package/dist/workflow-engine/tasks/set.js +31 -0
  518. package/dist/workflow-engine/tasks/set.js.map +1 -0
  519. package/dist/workflow-engine/tasks/switch.d.ts +25 -0
  520. package/dist/workflow-engine/tasks/switch.js +76 -0
  521. package/dist/workflow-engine/tasks/switch.js.map +1 -0
  522. package/dist/workflow-engine/tasks/try.d.ts +49 -0
  523. package/dist/workflow-engine/tasks/try.js +189 -0
  524. package/dist/workflow-engine/tasks/try.js.map +1 -0
  525. package/dist/workflow-engine/tasks/wait.d.ts +24 -0
  526. package/dist/workflow-engine/tasks/wait.js +39 -0
  527. package/dist/workflow-engine/tasks/wait.js.map +1 -0
  528. package/dist/workflow-engine/types.d.ts +682 -0
  529. package/dist/workflow-engine/types.js +47 -0
  530. package/dist/workflow-engine/types.js.map +1 -0
  531. package/dist/workflows/call-agent-orchestrator.d.ts +31 -0
  532. package/dist/workflows/call-agent-orchestrator.js +214 -0
  533. package/dist/workflows/call-agent-orchestrator.js.map +1 -0
  534. package/dist/workflows/connect-mcp-server.d.ts +20 -0
  535. package/dist/workflows/connect-mcp-server.js +113 -0
  536. package/dist/workflows/connect-mcp-server.js.map +1 -0
  537. package/dist/workflows/engine-core.d.ts +36 -0
  538. package/dist/workflows/engine-core.js +272 -0
  539. package/dist/workflows/engine-core.js.map +1 -0
  540. package/dist/workflows/execute-from-execution.d.ts +32 -0
  541. package/dist/workflows/execute-from-execution.js +71 -0
  542. package/dist/workflows/execute-from-execution.js.map +1 -0
  543. package/dist/workflows/execute-serverless-workflow.d.ts +32 -0
  544. package/dist/workflows/execute-serverless-workflow.js +36 -0
  545. package/dist/workflows/execute-serverless-workflow.js.map +1 -0
  546. package/dist/workflows/human-input-orchestrator.d.ts +19 -0
  547. package/dist/workflows/human-input-orchestrator.js +59 -0
  548. package/dist/workflows/human-input-orchestrator.js.map +1 -0
  549. package/dist/workflows/index.d.ts +22 -0
  550. package/dist/workflows/index.js +23 -0
  551. package/dist/workflows/index.js.map +1 -0
  552. package/dist/workflows/listen-orchestrator.d.ts +29 -0
  553. package/dist/workflows/listen-orchestrator.js +143 -0
  554. package/dist/workflows/listen-orchestrator.js.map +1 -0
  555. package/dist/workflows/metrics-sink.d.ts +33 -0
  556. package/dist/workflows/metrics-sink.js +21 -0
  557. package/dist/workflows/metrics-sink.js.map +1 -0
  558. package/dist/workflows/run-orchestrator.d.ts +15 -0
  559. package/dist/workflows/run-orchestrator.js +27 -0
  560. package/dist/workflows/run-orchestrator.js.map +1 -0
  561. package/dist/workflows/types.d.ts +46 -0
  562. package/dist/workflows/types.js +15 -0
  563. package/dist/workflows/types.js.map +1 -0
  564. package/dist/workflows/workflow-signals.d.ts +29 -0
  565. package/dist/workflows/workflow-signals.js +46 -0
  566. package/dist/workflows/workflow-signals.js.map +1 -0
  567. package/package.json +108 -0
  568. package/src/__test-utils__/__tests__/replay-fetch.test.ts +155 -0
  569. package/src/__test-utils__/mock-client.ts +44 -0
  570. package/src/__test-utils__/mock-workspace.ts +28 -0
  571. package/src/__test-utils__/proto-helpers.ts +41 -0
  572. package/src/__test-utils__/replay-fetch.ts +523 -0
  573. package/src/__tests__/bootstrap.test.ts +221 -0
  574. package/src/__tests__/claimcheck-codec.test.ts +257 -0
  575. package/src/__tests__/config.test.ts +150 -0
  576. package/src/__tests__/deterministic-eval-llm.test.ts +269 -0
  577. package/src/__tests__/deterministic-mcp-hitl.test.ts +405 -0
  578. package/src/__tests__/golden-e2e.test.ts +250 -0
  579. package/src/__tests__/ipc-protocol-fixtures.test.ts +66 -0
  580. package/src/__tests__/ipc-protocol.test.ts +32 -0
  581. package/src/__tests__/otel-metrics.test.ts +40 -0
  582. package/src/__tests__/runner-manager.test.ts +55 -0
  583. package/src/__tests__/runner-token-coordinator.test.ts +166 -0
  584. package/src/__tests__/runner.test.ts +182 -0
  585. package/src/__tests__/worker.test.ts +18 -0
  586. package/src/activities/__tests__/call-agent-contracts.test.ts +483 -0
  587. package/src/activities/__tests__/call-agent.test.ts +263 -0
  588. package/src/activities/__tests__/call-function.test.ts +47 -0
  589. package/src/activities/__tests__/call-grpc.test.ts +39 -0
  590. package/src/activities/__tests__/call-http.test.ts +288 -0
  591. package/src/activities/__tests__/call-llm.test.ts +301 -0
  592. package/src/activities/__tests__/classify-tool-approvals.test.ts +430 -0
  593. package/src/activities/__tests__/discover-mcp-server.test.ts +641 -0
  594. package/src/activities/__tests__/ensure-thread.test.ts +96 -0
  595. package/src/activities/__tests__/error-classifier.test.ts +372 -0
  596. package/src/activities/__tests__/evaluate-expressions.test.ts +114 -0
  597. package/src/activities/__tests__/hydrate-workflow-execution.test.ts +321 -0
  598. package/src/activities/__tests__/notification.test.ts +151 -0
  599. package/src/activities/__tests__/workflow-event-activities.test.ts +664 -0
  600. package/src/activities/call-agent-status.ts +130 -0
  601. package/src/activities/call-agent.ts +302 -0
  602. package/src/activities/call-eval.ts +333 -0
  603. package/src/activities/call-function.ts +73 -0
  604. package/src/activities/call-grpc.ts +140 -0
  605. package/src/activities/call-http.ts +185 -0
  606. package/src/activities/call-llm.ts +379 -0
  607. package/src/activities/call-transform.ts +54 -0
  608. package/src/activities/call-validate.ts +223 -0
  609. package/src/activities/classify-tool-approvals.ts +319 -0
  610. package/src/activities/discover-mcp-server.ts +411 -0
  611. package/src/activities/emit-event.ts +195 -0
  612. package/src/activities/ensure-thread.ts +45 -0
  613. package/src/activities/evaluate-expressions.ts +47 -0
  614. package/src/activities/execute-cursor/__tests__/approval-gate.test.ts +188 -0
  615. package/src/activities/execute-cursor/__tests__/build-prompt.test.ts +111 -0
  616. package/src/activities/execute-cursor/__tests__/cursor-baseurl-routing.test.ts +86 -0
  617. package/src/activities/execute-cursor/__tests__/cursor-fetch-interceptor-bypass.test.ts +64 -0
  618. package/src/activities/execute-cursor/__tests__/cursor-mode.test.ts +95 -0
  619. package/src/activities/execute-cursor/__tests__/cursor-sdk-auth-smoke.test.ts +90 -0
  620. package/src/activities/execute-cursor/__tests__/delta-enricher.test.ts +242 -0
  621. package/src/activities/execute-cursor/__tests__/error-classifier-introspection.test.ts +156 -0
  622. package/src/activities/execute-cursor/__tests__/fetch-interceptor.test.ts +211 -0
  623. package/src/activities/execute-cursor/__tests__/hitl-ledger.test.ts +298 -0
  624. package/src/activities/execute-cursor/__tests__/http2-interceptor.test.ts +360 -0
  625. package/src/activities/execute-cursor/__tests__/message-translator.test.ts +657 -0
  626. package/src/activities/execute-cursor/__tests__/model-pricing.test.ts +92 -0
  627. package/src/activities/execute-cursor/__tests__/prompt-builder-delegation.test.ts +101 -0
  628. package/src/activities/execute-cursor/__tests__/runner-error-regressions.test.ts +144 -0
  629. package/src/activities/execute-cursor/__tests__/session-lifecycle.test.ts +65 -0
  630. package/src/activities/execute-cursor/__tests__/skill-resolver.test.ts +265 -0
  631. package/src/activities/execute-cursor/__tests__/subagent-config.test.ts +107 -0
  632. package/src/activities/execute-cursor/__tests__/todo-tracker.test.ts +498 -0
  633. package/src/activities/execute-cursor/__tests__/workspace-provision.test.ts +283 -0
  634. package/src/activities/execute-cursor/approval-policy.ts +224 -0
  635. package/src/activities/execute-cursor/approval-state.ts +311 -0
  636. package/src/activities/execute-cursor/attachment-resolver.ts +78 -0
  637. package/src/activities/execute-cursor/blueprint-resolver.ts +234 -0
  638. package/src/activities/execute-cursor/connect-backfill.ts +49 -0
  639. package/src/activities/execute-cursor/cursor-event-recorder.ts +83 -0
  640. package/src/activities/execute-cursor/cursor-mode.ts +42 -0
  641. package/src/activities/execute-cursor/delta-enricher.ts +307 -0
  642. package/src/activities/execute-cursor/env-resolver.ts +64 -0
  643. package/src/activities/execute-cursor/error-classifier.ts +247 -0
  644. package/src/activities/execute-cursor/fetch-interceptor.ts +382 -0
  645. package/src/activities/execute-cursor/hitl-diagnostics.ts +82 -0
  646. package/src/activities/execute-cursor/hook-script.ts +159 -0
  647. package/src/activities/execute-cursor/http2-interceptor.ts +253 -0
  648. package/src/activities/execute-cursor/index.ts +1439 -0
  649. package/src/activities/execute-cursor/mcp-config.ts +66 -0
  650. package/src/activities/execute-cursor/mcp-resolver.ts +271 -0
  651. package/src/activities/execute-cursor/message-translator.ts +896 -0
  652. package/src/activities/execute-cursor/model-pricing-data.ts +167 -0
  653. package/src/activities/execute-cursor/model-pricing.ts +167 -0
  654. package/src/activities/execute-cursor/placeholder-resolver.ts +109 -0
  655. package/src/activities/execute-cursor/prompt-builder.ts +349 -0
  656. package/src/activities/execute-cursor/rejection-capture.ts +100 -0
  657. package/src/activities/execute-cursor/session-lifecycle.ts +429 -0
  658. package/src/activities/execute-cursor/skill-resolver.ts +176 -0
  659. package/src/activities/execute-cursor/subagent-config.ts +62 -0
  660. package/src/activities/execute-cursor/todo-tracker.ts +133 -0
  661. package/src/activities/execute-cursor/usage-accumulator.ts +126 -0
  662. package/src/activities/execute-cursor/workspace-provision.ts +55 -0
  663. package/src/activities/execute-cursor/workspace-setup.ts +75 -0
  664. package/src/activities/execute-deep-agent/__test-utils__/v3-event-fixtures.ts +281 -0
  665. package/src/activities/execute-deep-agent/__tests__/attachment-injector.test.ts +720 -0
  666. package/src/activities/execute-deep-agent/__tests__/auto-publish.test.ts +146 -0
  667. package/src/activities/execute-deep-agent/__tests__/environment.test.ts +103 -0
  668. package/src/activities/execute-deep-agent/__tests__/event-recorder.test.ts +150 -0
  669. package/src/activities/execute-deep-agent/__tests__/execution-state-extended.test.ts +150 -0
  670. package/src/activities/execute-deep-agent/__tests__/execution-state.test.ts +157 -0
  671. package/src/activities/execute-deep-agent/__tests__/hitl-integration.test.ts +223 -0
  672. package/src/activities/execute-deep-agent/__tests__/hitl.test.ts +244 -0
  673. package/src/activities/execute-deep-agent/__tests__/index.test.ts +91 -0
  674. package/src/activities/execute-deep-agent/__tests__/inline-publisher.test.ts +240 -0
  675. package/src/activities/execute-deep-agent/__tests__/post-stream.test.ts +112 -0
  676. package/src/activities/execute-deep-agent/__tests__/prompt-builder.test.ts +208 -0
  677. package/src/activities/execute-deep-agent/__tests__/status-builder.test.ts +1771 -0
  678. package/src/activities/execute-deep-agent/__tests__/streaming-scheduler.test.ts +199 -0
  679. package/src/activities/execute-deep-agent/__tests__/streaming-v3.test.ts +527 -0
  680. package/src/activities/execute-deep-agent/__tests__/streaming.test.ts +508 -0
  681. package/src/activities/execute-deep-agent/__tests__/subagent-tracker.test.ts +474 -0
  682. package/src/activities/execute-deep-agent/__tests__/subagent-transformer.test.ts +734 -0
  683. package/src/activities/execute-deep-agent/__tests__/subagent-wiring.test.ts +71 -0
  684. package/src/activities/execute-deep-agent/__tests__/summarization-verification.test.ts +323 -0
  685. package/src/activities/execute-deep-agent/__tests__/v3-event-recorder.test.ts +186 -0
  686. package/src/activities/execute-deep-agent/__tests__/v3-protocol-normalizer.test.ts +324 -0
  687. package/src/activities/execute-deep-agent/__tests__/v3-status-builder.test.ts +504 -0
  688. package/src/activities/execute-deep-agent/__tests__/writeback-coordinator.test.ts +399 -0
  689. package/src/activities/execute-deep-agent/attachment-injector.ts +470 -0
  690. package/src/activities/execute-deep-agent/auto-publish.ts +80 -0
  691. package/src/activities/execute-deep-agent/environment.ts +67 -0
  692. package/src/activities/execute-deep-agent/event-recorder.ts +95 -0
  693. package/src/activities/execute-deep-agent/execution-state.ts +87 -0
  694. package/src/activities/execute-deep-agent/execution-status-writer.ts +19 -0
  695. package/src/activities/execute-deep-agent/hitl.ts +221 -0
  696. package/src/activities/execute-deep-agent/index.ts +342 -0
  697. package/src/activities/execute-deep-agent/inline-publisher.ts +134 -0
  698. package/src/activities/execute-deep-agent/post-stream.ts +109 -0
  699. package/src/activities/execute-deep-agent/prompt-builder.ts +264 -0
  700. package/src/activities/execute-deep-agent/setup.ts +599 -0
  701. package/src/activities/execute-deep-agent/status-builder-shared.ts +136 -0
  702. package/src/activities/execute-deep-agent/status-builder.ts +412 -0
  703. package/src/activities/execute-deep-agent/streaming-scheduler.ts +159 -0
  704. package/src/activities/execute-deep-agent/streaming-side-effects.ts +89 -0
  705. package/src/activities/execute-deep-agent/streaming-terminal.ts +96 -0
  706. package/src/activities/execute-deep-agent/streaming-v3.ts +272 -0
  707. package/src/activities/execute-deep-agent/streaming.ts +303 -0
  708. package/src/activities/execute-deep-agent/subagent-tracker.ts +445 -0
  709. package/src/activities/execute-deep-agent/subagent-transformer.ts +648 -0
  710. package/src/activities/execute-deep-agent/subagent-wiring.ts +56 -0
  711. package/src/activities/execute-deep-agent/v3-event-recorder.ts +111 -0
  712. package/src/activities/execute-deep-agent/v3-events.ts +153 -0
  713. package/src/activities/execute-deep-agent/v3-protocol-normalizer.ts +264 -0
  714. package/src/activities/execute-deep-agent/v3-status-builder.ts +490 -0
  715. package/src/activities/execute-deep-agent/writeback-coordinator.ts +420 -0
  716. package/src/activities/hydrate-workflow-execution.ts +306 -0
  717. package/src/activities/notification.ts +71 -0
  718. package/src/activities/promote-task-output.ts +126 -0
  719. package/src/activities/run-command.ts +148 -0
  720. package/src/activities/workflow-event-activities.ts +481 -0
  721. package/src/bootstrap.ts +173 -0
  722. package/src/budget/__tests__/tracker.test.ts +293 -0
  723. package/src/budget/index.ts +9 -0
  724. package/src/budget/tracker.ts +171 -0
  725. package/src/claimcheck/compressor.ts +9 -0
  726. package/src/claimcheck/config.ts +20 -0
  727. package/src/claimcheck/index.ts +3 -0
  728. package/src/claimcheck/payload-codec.ts +107 -0
  729. package/src/client/__tests__/server-contracts.test.ts +149 -0
  730. package/src/client/__tests__/stigmer-client.test.ts +142 -0
  731. package/src/client/server-contracts.ts +125 -0
  732. package/src/client/stigmer-client.ts +339 -0
  733. package/src/config.ts +185 -0
  734. package/src/idle-watchdog.ts +28 -0
  735. package/src/index.ts +48 -0
  736. package/src/interceptors/workflow-metrics-sink.ts +56 -0
  737. package/src/ipc-protocol-fixtures.ts +117 -0
  738. package/src/ipc-protocol.ts +113 -0
  739. package/src/main.ts +324 -0
  740. package/src/middleware/__tests__/approval-gate.test.ts +231 -0
  741. package/src/middleware/__tests__/cost-cap.test.ts +192 -0
  742. package/src/middleware/__tests__/error-hints.test.ts +90 -0
  743. package/src/middleware/__tests__/execution-budget.test.ts +138 -0
  744. package/src/middleware/__tests__/graceful-stop.test.ts +105 -0
  745. package/src/middleware/__tests__/loop-detection.test.ts +137 -0
  746. package/src/middleware/__tests__/otel-spans.test.ts +89 -0
  747. package/src/middleware/__tests__/think-tool.test.ts +26 -0
  748. package/src/middleware/__tests__/tool-truncation.test.ts +112 -0
  749. package/src/middleware/approval-gate.ts +179 -0
  750. package/src/middleware/cost-cap.ts +213 -0
  751. package/src/middleware/error-hints.ts +136 -0
  752. package/src/middleware/execution-budget.ts +176 -0
  753. package/src/middleware/graceful-stop.ts +86 -0
  754. package/src/middleware/index.ts +70 -0
  755. package/src/middleware/loop-detection.ts +192 -0
  756. package/src/middleware/otel-spans.ts +205 -0
  757. package/src/middleware/think-tool.ts +38 -0
  758. package/src/middleware/tool-truncation.ts +94 -0
  759. package/src/middleware/types.ts +114 -0
  760. package/src/notification/__tests__/provider.test.ts +85 -0
  761. package/src/notification/__tests__/webhook.test.ts +127 -0
  762. package/src/notification/index.ts +15 -0
  763. package/src/notification/provider.ts +52 -0
  764. package/src/notification/webhook.ts +61 -0
  765. package/src/otel-metrics.ts +73 -0
  766. package/src/otel.ts +194 -0
  767. package/src/runner-manager.ts +652 -0
  768. package/src/runner-token-coordinator.ts +135 -0
  769. package/src/runner.ts +380 -0
  770. package/src/shared/__tests__/approval-policy.test.ts +256 -0
  771. package/src/shared/__tests__/artifact-storage-extended.test.ts +208 -0
  772. package/src/shared/__tests__/artifact-storage.test.ts +365 -0
  773. package/src/shared/__tests__/connect-backfill.test.ts +346 -0
  774. package/src/shared/__tests__/extract-json.test.ts +153 -0
  775. package/src/shared/__tests__/grpc-retry-extended.test.ts +176 -0
  776. package/src/shared/__tests__/grpc-retry.test.ts +172 -0
  777. package/src/shared/__tests__/json-schema-to-zod.test.ts +227 -0
  778. package/src/shared/__tests__/llm-proxy.test.ts +179 -0
  779. package/src/shared/__tests__/mcp-manager.test.ts +154 -0
  780. package/src/shared/__tests__/model-pricing.test.ts +85 -0
  781. package/src/shared/__tests__/model-registry.test.ts +197 -0
  782. package/src/shared/__tests__/placeholder-resolver.test.ts +210 -0
  783. package/src/shared/__tests__/plan-artifact.test.ts +142 -0
  784. package/src/shared/__tests__/skill-relevance.test.ts +292 -0
  785. package/src/shared/__tests__/skill-writer.test.ts +349 -0
  786. package/src/shared/__tests__/status.test.ts +142 -0
  787. package/src/shared/__tests__/subagent-gate.test.ts +112 -0
  788. package/src/shared/__tests__/tool-kind.test.ts +58 -0
  789. package/src/shared/__tests__/zip-extract.test.ts +204 -0
  790. package/src/shared/approval-policy.ts +146 -0
  791. package/src/shared/artifact-storage.ts +207 -0
  792. package/src/shared/checkpointer/__tests__/factory.test.ts +42 -0
  793. package/src/shared/checkpointer/__tests__/http-saver.test.ts +176 -0
  794. package/src/shared/checkpointer/factory.ts +73 -0
  795. package/src/shared/checkpointer/http-saver.ts +349 -0
  796. package/src/shared/checkpointer/types.ts +12 -0
  797. package/src/shared/connect-backfill.ts +162 -0
  798. package/src/shared/extract-json.ts +153 -0
  799. package/src/shared/grpc-retry.ts +113 -0
  800. package/src/shared/heartbeat.ts +70 -0
  801. package/src/shared/json-schema-to-zod.ts +53 -0
  802. package/src/shared/llm-proxy.ts +138 -0
  803. package/src/shared/mcp-manager.ts +150 -0
  804. package/src/shared/mcp-resolver.ts +150 -0
  805. package/src/shared/model-pricing-data.ts +109 -0
  806. package/src/shared/model-pricing.ts +81 -0
  807. package/src/shared/model-registry.ts +214 -0
  808. package/src/shared/placeholder-resolver.ts +102 -0
  809. package/src/shared/plan-artifact.ts +120 -0
  810. package/src/shared/skill-relevance.ts +222 -0
  811. package/src/shared/skill-writer.ts +300 -0
  812. package/src/shared/status.ts +94 -0
  813. package/src/shared/subagent-gate.ts +117 -0
  814. package/src/shared/tool-kind.ts +91 -0
  815. package/src/shared/workspace/__tests__/file-tree.test.ts +210 -0
  816. package/src/shared/workspace/__tests__/git-source.test.ts +423 -0
  817. package/src/shared/workspace/__tests__/local-backend-platform.test.ts +259 -0
  818. package/src/shared/workspace/__tests__/local-backend.test.ts +154 -0
  819. package/src/shared/workspace/__tests__/platform-mount.test.ts +378 -0
  820. package/src/shared/workspace/__tests__/provisioner.test.ts +145 -0
  821. package/src/shared/workspace/file-tree.ts +116 -0
  822. package/src/shared/workspace/local-backend.ts +140 -0
  823. package/src/shared/workspace/platform-dir.ts +38 -0
  824. package/src/shared/workspace/platform-mount.ts +190 -0
  825. package/src/shared/workspace/provisioner.ts +150 -0
  826. package/src/shared/workspace/sources/empty.ts +20 -0
  827. package/src/shared/workspace/sources/git.ts +285 -0
  828. package/src/shared/workspace/sources/local-path.ts +89 -0
  829. package/src/shared/workspace/types.ts +69 -0
  830. package/src/shared/zip-extract.ts +193 -0
  831. package/src/worker.ts +98 -0
  832. package/src/workflow-engine/__tests__/do-executor-recovery.test.ts +382 -0
  833. package/src/workflow-engine/__tests__/do-executor.test.ts +963 -0
  834. package/src/workflow-engine/__tests__/errors.test.ts +174 -0
  835. package/src/workflow-engine/__tests__/expression.test.ts +776 -0
  836. package/src/workflow-engine/__tests__/for.test.ts +575 -0
  837. package/src/workflow-engine/__tests__/fork.test.ts +838 -0
  838. package/src/workflow-engine/__tests__/golden-execution.test.ts +1085 -0
  839. package/src/workflow-engine/__tests__/jq-wasm-spike.test.ts +90 -0
  840. package/src/workflow-engine/__tests__/loader.test.ts +1393 -0
  841. package/src/workflow-engine/__tests__/pause-resume.test.ts +267 -0
  842. package/src/workflow-engine/__tests__/recovery.test.ts +115 -0
  843. package/src/workflow-engine/__tests__/resolve.test.ts +432 -0
  844. package/src/workflow-engine/__tests__/retry.test.ts +306 -0
  845. package/src/workflow-engine/__tests__/state.test.ts +174 -0
  846. package/src/workflow-engine/__tests__/task-status-accumulator.test.ts +373 -0
  847. package/src/workflow-engine/__tests__/tasks/call-agent-output.test.ts +120 -0
  848. package/src/workflow-engine/__tests__/tasks/call-agent.test.ts +816 -0
  849. package/src/workflow-engine/__tests__/tasks/call-function.test.ts +205 -0
  850. package/src/workflow-engine/__tests__/tasks/call-grpc.test.ts +133 -0
  851. package/src/workflow-engine/__tests__/tasks/call-http.test.ts +150 -0
  852. package/src/workflow-engine/__tests__/tasks/emit-event.test.ts +322 -0
  853. package/src/workflow-engine/__tests__/tasks/human-input.test.ts +416 -0
  854. package/src/workflow-engine/__tests__/tasks/listen.test.ts +422 -0
  855. package/src/workflow-engine/__tests__/tasks/raise.test.ts +166 -0
  856. package/src/workflow-engine/__tests__/tasks/run.test.ts +272 -0
  857. package/src/workflow-engine/__tests__/tasks/set.test.ts +127 -0
  858. package/src/workflow-engine/__tests__/tasks/switch.test.ts +277 -0
  859. package/src/workflow-engine/__tests__/tasks/try.test.ts +590 -0
  860. package/src/workflow-engine/__tests__/tasks/wait.test.ts +173 -0
  861. package/src/workflow-engine/clone.ts +18 -0
  862. package/src/workflow-engine/do-executor.ts +569 -0
  863. package/src/workflow-engine/duration.ts +22 -0
  864. package/src/workflow-engine/error-utils.ts +97 -0
  865. package/src/workflow-engine/errors.ts +130 -0
  866. package/src/workflow-engine/expression-utils.ts +129 -0
  867. package/src/workflow-engine/expression.ts +430 -0
  868. package/src/workflow-engine/loader.ts +524 -0
  869. package/src/workflow-engine/recovery.ts +80 -0
  870. package/src/workflow-engine/resolve.ts +342 -0
  871. package/src/workflow-engine/retry.ts +109 -0
  872. package/src/workflow-engine/state.ts +56 -0
  873. package/src/workflow-engine/task-factory.ts +160 -0
  874. package/src/workflow-engine/task-status-accumulator.ts +204 -0
  875. package/src/workflow-engine/tasks/call-agent-output.ts +132 -0
  876. package/src/workflow-engine/tasks/call-agent.ts +221 -0
  877. package/src/workflow-engine/tasks/call-function.ts +107 -0
  878. package/src/workflow-engine/tasks/call-grpc.ts +47 -0
  879. package/src/workflow-engine/tasks/call-http.ts +51 -0
  880. package/src/workflow-engine/tasks/for.ts +244 -0
  881. package/src/workflow-engine/tasks/fork.ts +228 -0
  882. package/src/workflow-engine/tasks/human-input.ts +147 -0
  883. package/src/workflow-engine/tasks/listen.ts +166 -0
  884. package/src/workflow-engine/tasks/raise.ts +81 -0
  885. package/src/workflow-engine/tasks/run.ts +142 -0
  886. package/src/workflow-engine/tasks/set.ts +47 -0
  887. package/src/workflow-engine/tasks/switch.ts +102 -0
  888. package/src/workflow-engine/tasks/try.ts +274 -0
  889. package/src/workflow-engine/tasks/wait.ts +53 -0
  890. package/src/workflow-engine/types.ts +911 -0
  891. package/src/workflows/__tests__/connect-mcp-server.test.ts +359 -0
  892. package/src/workflows/__tests__/execute-serverless-workflow.test.ts +277 -0
  893. package/src/workflows/call-agent-orchestrator.ts +283 -0
  894. package/src/workflows/connect-mcp-server.ts +152 -0
  895. package/src/workflows/engine-core.ts +406 -0
  896. package/src/workflows/execute-from-execution.ts +101 -0
  897. package/src/workflows/execute-serverless-workflow.ts +60 -0
  898. package/src/workflows/human-input-orchestrator.ts +76 -0
  899. package/src/workflows/index.ts +32 -0
  900. package/src/workflows/listen-orchestrator.ts +200 -0
  901. package/src/workflows/metrics-sink.ts +48 -0
  902. package/src/workflows/run-orchestrator.ts +34 -0
  903. package/src/workflows/types.ts +64 -0
  904. package/src/workflows/workflow-signals.ts +55 -0
@@ -0,0 +1,1771 @@
1
+ import { describe, it, expect, vi } from "vitest";
2
+ import { create } from "@bufbuild/protobuf";
3
+ import { AgentExecutionStatusSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
4
+ import {
5
+ ExecutionPhase,
6
+ MessageType,
7
+ ToolCallStatus,
8
+ } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/enum_pb";
9
+ import { StatusBuilder, type StreamEvent, type ApprovalPolicyProvider } from "../status-builder.js";
10
+ import type { MergedToolPolicy } from "../../../shared/approval-policy.js";
11
+
12
+ function makeBuilder() {
13
+ const status = create(AgentExecutionStatusSchema, {});
14
+ return new StatusBuilder("exec-test", status);
15
+ }
16
+
17
+ function chatStreamEvent(
18
+ runId: string,
19
+ content: string | Record<string, unknown>[],
20
+ metadata?: Record<string, unknown>,
21
+ ): StreamEvent {
22
+ return {
23
+ event: "on_chat_model_stream",
24
+ name: "ChatAnthropic",
25
+ run_id: runId,
26
+ data: { chunk: { content } },
27
+ metadata,
28
+ };
29
+ }
30
+
31
+ function chatEndEvent(
32
+ runId: string,
33
+ usageMetadata?: Record<string, unknown>,
34
+ ): StreamEvent {
35
+ return {
36
+ event: "on_chat_model_end",
37
+ name: "ChatAnthropic",
38
+ run_id: runId,
39
+ data: {
40
+ output: {
41
+ content: "final text",
42
+ usage_metadata: usageMetadata,
43
+ },
44
+ },
45
+ };
46
+ }
47
+
48
+ function toolStartEvent(
49
+ runId: string,
50
+ toolName: string,
51
+ input?: Record<string, unknown>,
52
+ ): StreamEvent {
53
+ return {
54
+ event: "on_tool_start",
55
+ name: toolName,
56
+ run_id: runId,
57
+ data: { input: input ?? {} },
58
+ };
59
+ }
60
+
61
+ function toolEndEvent(
62
+ runId: string,
63
+ output: unknown,
64
+ ): StreamEvent {
65
+ return {
66
+ event: "on_tool_end",
67
+ name: "some_tool",
68
+ run_id: runId,
69
+ data: { output },
70
+ };
71
+ }
72
+
73
+ describe("StatusBuilder", () => {
74
+ describe("initialization", () => {
75
+ it("sets phase to IN_PROGRESS", () => {
76
+ const sb = makeBuilder();
77
+ expect(sb.currentStatus.phase).toBe(ExecutionPhase.EXECUTION_IN_PROGRESS);
78
+ });
79
+
80
+ it("sets startedAt timestamp", () => {
81
+ const sb = makeBuilder();
82
+ expect(sb.currentStatus.startedAt).toBeTruthy();
83
+ });
84
+
85
+ it("starts with forceNextUpdate = false", () => {
86
+ expect(makeBuilder().forceNextUpdate).toBe(false);
87
+ });
88
+ });
89
+
90
+ describe("on_chat_model_stream — text tokens", () => {
91
+ it("creates an AI message and appends text", () => {
92
+ const sb = makeBuilder();
93
+ sb.processEvent(chatStreamEvent("run-1", "Hello"));
94
+ sb.processEvent(chatStreamEvent("run-1", " world"));
95
+
96
+ expect(sb.currentStatus.messages).toHaveLength(1);
97
+ const msg = sb.currentStatus.messages[0];
98
+ expect(msg.type).toBe(MessageType.MESSAGE_AI);
99
+ expect(msg.content).toBe("Hello world");
100
+ expect(msg.isStreaming).toBe(true);
101
+ });
102
+
103
+ it("handles content block array with text blocks", () => {
104
+ const sb = makeBuilder();
105
+ sb.processEvent(chatStreamEvent("run-1", [
106
+ { type: "text", text: "block text" },
107
+ ]));
108
+
109
+ expect(sb.currentStatus.messages).toHaveLength(1);
110
+ expect(sb.currentStatus.messages[0].content).toBe("block text");
111
+ });
112
+
113
+ it("ignores empty string content", () => {
114
+ const sb = makeBuilder();
115
+ sb.processEvent(chatStreamEvent("run-1", ""));
116
+
117
+ expect(sb.currentStatus.messages).toHaveLength(0);
118
+ });
119
+
120
+ it("ignores events with no chunk data", () => {
121
+ const sb = makeBuilder();
122
+ sb.processEvent({
123
+ event: "on_chat_model_stream",
124
+ run_id: "run-1",
125
+ data: {},
126
+ });
127
+
128
+ expect(sb.currentStatus.messages).toHaveLength(0);
129
+ });
130
+ });
131
+
132
+ describe("on_chat_model_stream — thinking blocks", () => {
133
+ it("creates a THINKING message for thinking content", () => {
134
+ const sb = makeBuilder();
135
+ sb.processEvent(chatStreamEvent("run-1", [
136
+ { type: "thinking", thinking: "Let me reason..." },
137
+ ]));
138
+
139
+ expect(sb.currentStatus.messages).toHaveLength(1);
140
+ const msg = sb.currentStatus.messages[0];
141
+ expect(msg.type).toBe(MessageType.MESSAGE_THINKING);
142
+ expect(msg.content).toBe("Let me reason...");
143
+ expect(msg.isStreaming).toBe(true);
144
+ });
145
+
146
+ it("appends to existing thinking message in same namespace", () => {
147
+ const sb = makeBuilder();
148
+ sb.processEvent(chatStreamEvent("run-1", [
149
+ { type: "thinking", thinking: "Step 1. " },
150
+ ]));
151
+ sb.processEvent(chatStreamEvent("run-1", [
152
+ { type: "thinking", thinking: "Step 2." },
153
+ ]));
154
+
155
+ expect(sb.currentStatus.messages).toHaveLength(1);
156
+ expect(sb.currentStatus.messages[0].content).toBe("Step 1. Step 2.");
157
+ });
158
+
159
+ it("handles mixed thinking and text in same event", () => {
160
+ const sb = makeBuilder();
161
+ sb.processEvent(chatStreamEvent("run-1", [
162
+ { type: "thinking", thinking: "hmm..." },
163
+ { type: "text", text: "answer" },
164
+ ]));
165
+
166
+ expect(sb.currentStatus.messages).toHaveLength(2);
167
+ expect(sb.currentStatus.messages[0].type).toBe(MessageType.MESSAGE_THINKING);
168
+ expect(sb.currentStatus.messages[1].type).toBe(MessageType.MESSAGE_AI);
169
+ });
170
+ });
171
+
172
+ describe("turn boundary detection", () => {
173
+ it("creates new message when run_id changes", () => {
174
+ const sb = makeBuilder();
175
+ sb.processEvent(chatStreamEvent("run-1", "First turn"));
176
+ sb.processEvent(chatStreamEvent("run-2", "Second turn"));
177
+
178
+ expect(sb.currentStatus.messages).toHaveLength(2);
179
+ expect(sb.currentStatus.messages[0].content).toBe("First turn");
180
+ expect(sb.currentStatus.messages[0].isStreaming).toBe(false);
181
+ expect(sb.currentStatus.messages[1].content).toBe("Second turn");
182
+ expect(sb.currentStatus.messages[1].isStreaming).toBe(true);
183
+ });
184
+
185
+ it("reuses same message for same run_id", () => {
186
+ const sb = makeBuilder();
187
+ sb.processEvent(chatStreamEvent("run-1", "a"));
188
+ sb.processEvent(chatStreamEvent("run-1", "b"));
189
+
190
+ expect(sb.currentStatus.messages).toHaveLength(1);
191
+ expect(sb.currentStatus.messages[0].content).toBe("ab");
192
+ });
193
+ });
194
+
195
+ describe("on_chat_model_end", () => {
196
+ it("finalizes message streaming flag", () => {
197
+ const sb = makeBuilder();
198
+ sb.processEvent(chatStreamEvent("run-1", "text"));
199
+ expect(sb.currentStatus.messages[0].isStreaming).toBe(true);
200
+
201
+ sb.processEvent(chatEndEvent("run-1"));
202
+ expect(sb.currentStatus.messages[0].isStreaming).toBe(false);
203
+ });
204
+
205
+ it("accumulates usage metadata", () => {
206
+ const sb = makeBuilder();
207
+ sb.processEvent(chatStreamEvent("run-1", "text"));
208
+ sb.processEvent(chatEndEvent("run-1", {
209
+ input_tokens: 100,
210
+ output_tokens: 50,
211
+ cache_read_input_tokens: 10,
212
+ cache_creation_input_tokens: 5,
213
+ }));
214
+
215
+ const usage = sb.currentStatus.streamingUsage;
216
+ expect(usage).toBeDefined();
217
+ expect(usage!.inputTokens).toBe(100n);
218
+ expect(usage!.outputTokens).toBe(50n);
219
+ expect(usage!.cacheReadTokens).toBe(10n);
220
+ expect(usage!.cacheWriteTokens).toBe(5n);
221
+ expect(usage!.totalTokens).toBe(165n);
222
+ expect(usage!.turnCount).toBe(1);
223
+ });
224
+
225
+ it("accumulates across multiple turns", () => {
226
+ const sb = makeBuilder();
227
+ sb.processEvent(chatStreamEvent("run-1", "turn 1"));
228
+ sb.processEvent(chatEndEvent("run-1", {
229
+ input_tokens: 100,
230
+ output_tokens: 50,
231
+ }));
232
+ sb.processEvent(chatStreamEvent("run-2", "turn 2"));
233
+ sb.processEvent(chatEndEvent("run-2", {
234
+ input_tokens: 200,
235
+ output_tokens: 100,
236
+ }));
237
+
238
+ const usage = sb.currentStatus.streamingUsage!;
239
+ expect(usage.inputTokens).toBe(300n);
240
+ expect(usage.outputTokens).toBe(150n);
241
+ expect(usage.turnCount).toBe(2);
242
+ });
243
+ });
244
+
245
+ describe("on_tool_start", () => {
246
+ it("creates a RUNNING tool call on the current AI message", () => {
247
+ const sb = makeBuilder();
248
+ sb.processEvent(chatStreamEvent("run-1", "I'll read a file"));
249
+ sb.processEvent(toolStartEvent("tool-run-1", "read", { path: "/foo.ts" }));
250
+
251
+ const msg = sb.currentStatus.messages[0];
252
+ expect(msg.toolCalls).toHaveLength(1);
253
+ expect(msg.toolCalls[0].id).toBe("tool-run-1");
254
+ expect(msg.toolCalls[0].name).toBe("read");
255
+ expect(msg.toolCalls[0].status).toBe(ToolCallStatus.TOOL_CALL_RUNNING);
256
+ expect(msg.toolCalls[0].startedAt).toBeTruthy();
257
+ expect(msg.toolCalls[0].args).toEqual({ path: "/foo.ts" });
258
+ });
259
+
260
+ it("sets forceNextUpdate", () => {
261
+ const sb = makeBuilder();
262
+ sb.processEvent(chatStreamEvent("run-1", "text"));
263
+ sb.processEvent(toolStartEvent("tool-run-1", "read"));
264
+
265
+ expect(sb.forceNextUpdate).toBe(true);
266
+ });
267
+
268
+ it("creates AI message for tool-only turn when no prior AI message exists", () => {
269
+ const sb = makeBuilder();
270
+ sb.processEvent(toolStartEvent("tool-run-1", "read"));
271
+
272
+ expect(sb.currentStatus.messages).toHaveLength(1);
273
+ expect(sb.currentStatus.messages[0].toolCalls).toHaveLength(1);
274
+ expect(sb.currentStatus.messages[0].toolCalls[0].name).toBe("read");
275
+ });
276
+ });
277
+
278
+ describe("on_tool_end", () => {
279
+ it("marks tool call as COMPLETED with result", () => {
280
+ const sb = makeBuilder();
281
+ sb.processEvent(chatStreamEvent("run-1", "reading file"));
282
+ sb.processEvent(toolStartEvent("tool-run-1", "read"));
283
+ sb.processEvent(toolEndEvent("tool-run-1", "file contents here"));
284
+
285
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
286
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
287
+ expect(tc.result).toBe("file contents here");
288
+ expect(tc.completedAt).toBeTruthy();
289
+ expect(tc.isStreaming).toBe(false);
290
+ });
291
+
292
+ it("marks tool call as FAILED when output has error", () => {
293
+ const sb = makeBuilder();
294
+ sb.processEvent(chatStreamEvent("run-1", "text"));
295
+ sb.processEvent(toolStartEvent("tool-run-1", "write"));
296
+ sb.processEvent(toolEndEvent("tool-run-1", { error: "permission denied" }));
297
+
298
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
299
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_FAILED);
300
+ expect(tc.error).toBe("permission denied");
301
+ });
302
+
303
+ it("truncates long results", () => {
304
+ const sb = makeBuilder();
305
+ sb.processEvent(chatStreamEvent("run-1", "text"));
306
+ sb.processEvent(toolStartEvent("tool-run-1", "read"));
307
+
308
+ const longResult = "x".repeat(60_000);
309
+ sb.processEvent(toolEndEvent("tool-run-1", longResult));
310
+
311
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
312
+ expect(tc.result.length).toBeLessThan(longResult.length);
313
+ expect(tc.result).toContain("[truncated:");
314
+ });
315
+
316
+ it("sets forceNextUpdate", () => {
317
+ const sb = makeBuilder();
318
+ sb.processEvent(chatStreamEvent("run-1", "text"));
319
+ sb.processEvent(toolStartEvent("tool-run-1", "read"));
320
+ sb.clearForceFlag();
321
+ sb.processEvent(toolEndEvent("tool-run-1", "done"));
322
+
323
+ expect(sb.forceNextUpdate).toBe(true);
324
+ });
325
+
326
+ it("ignores unknown tool run_id", () => {
327
+ const sb = makeBuilder();
328
+ sb.processEvent(toolEndEvent("unknown-run", "result"));
329
+ // Should not throw
330
+ expect(sb.currentStatus.messages).toHaveLength(0);
331
+ });
332
+
333
+ it("handles object output with content field", () => {
334
+ const sb = makeBuilder();
335
+ sb.processEvent(chatStreamEvent("run-1", "text"));
336
+ sb.processEvent(toolStartEvent("tool-run-1", "shell"));
337
+ sb.processEvent(toolEndEvent("tool-run-1", { content: "stdout output" }));
338
+
339
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
340
+ expect(tc.result).toBe("stdout output");
341
+ });
342
+ });
343
+
344
+ describe("unknown events", () => {
345
+ it("silently ignores unhandled event types", () => {
346
+ const sb = makeBuilder();
347
+ sb.processEvent({
348
+ event: "on_chain_start",
349
+ run_id: "run-1",
350
+ data: {},
351
+ });
352
+ expect(sb.currentStatus.messages).toHaveLength(0);
353
+ });
354
+ });
355
+
356
+ describe("clearForceFlag", () => {
357
+ it("resets forceNextUpdate to false", () => {
358
+ const sb = makeBuilder();
359
+ sb.processEvent(chatStreamEvent("run-1", "text"));
360
+ sb.processEvent(toolStartEvent("tool-run-1", "read"));
361
+ expect(sb.forceNextUpdate).toBe(true);
362
+
363
+ sb.clearForceFlag();
364
+ expect(sb.forceNextUpdate).toBe(false);
365
+ });
366
+ });
367
+
368
+ describe("addArtifact", () => {
369
+ it("appends a new artifact and sets forceNextUpdate", () => {
370
+ const sb = makeBuilder();
371
+ sb.clearForceFlag();
372
+
373
+ sb.addArtifact({
374
+ name: "report.txt",
375
+ sandboxPath: "workspace/report.txt",
376
+ kind: 0,
377
+ sizeBytes: 100n,
378
+ storageKey: "artifacts/exec-1/report.txt",
379
+ downloadUrl: "http://localhost/dl",
380
+ createdAt: "2026-05-19T12:00:00Z",
381
+ expiresAt: "",
382
+ entries: [],
383
+ contentHash: "abc123",
384
+ $typeName: "ai.stigmer.agentic.agentexecution.v1.ExecutionArtifact",
385
+ } as any);
386
+
387
+ expect(sb.currentStatus.artifacts).toHaveLength(1);
388
+ expect(sb.currentStatus.artifacts[0].name).toBe("report.txt");
389
+ expect(sb.forceNextUpdate).toBe(true);
390
+ });
391
+
392
+ it("deduplicates by sandboxPath — skips same contentHash", () => {
393
+ const sb = makeBuilder();
394
+ const artifact = {
395
+ name: "f.txt",
396
+ sandboxPath: "ws/f.txt",
397
+ kind: 0,
398
+ sizeBytes: 10n,
399
+ storageKey: "k",
400
+ downloadUrl: "u",
401
+ createdAt: "",
402
+ expiresAt: "",
403
+ entries: [],
404
+ contentHash: "hash1",
405
+ $typeName: "ai.stigmer.agentic.agentexecution.v1.ExecutionArtifact",
406
+ } as any;
407
+
408
+ sb.addArtifact(artifact);
409
+ sb.clearForceFlag();
410
+ sb.addArtifact({ ...artifact });
411
+
412
+ expect(sb.currentStatus.artifacts).toHaveLength(1);
413
+ expect(sb.forceNextUpdate).toBe(false);
414
+ });
415
+
416
+ it("replaces artifact when contentHash changes", () => {
417
+ const sb = makeBuilder();
418
+ const artifact = {
419
+ name: "f.txt",
420
+ sandboxPath: "ws/f.txt",
421
+ kind: 0,
422
+ sizeBytes: 10n,
423
+ storageKey: "k",
424
+ downloadUrl: "u",
425
+ createdAt: "",
426
+ expiresAt: "",
427
+ entries: [],
428
+ contentHash: "hash1",
429
+ $typeName: "ai.stigmer.agentic.agentexecution.v1.ExecutionArtifact",
430
+ } as any;
431
+
432
+ sb.addArtifact(artifact);
433
+ sb.clearForceFlag();
434
+ sb.addArtifact({ ...artifact, contentHash: "hash2", sizeBytes: 20n });
435
+
436
+ expect(sb.currentStatus.artifacts).toHaveLength(1);
437
+ expect(sb.currentStatus.artifacts[0].contentHash).toBe("hash2");
438
+ expect(sb.forceNextUpdate).toBe(true);
439
+ });
440
+ });
441
+
442
+ describe("addWriteBack", () => {
443
+ it("appends a new writeback entry and sets forceNextUpdate", () => {
444
+ const sb = makeBuilder();
445
+ sb.clearForceFlag();
446
+
447
+ sb.addWriteBack({
448
+ workspaceEntryName: "my-app",
449
+ branchName: "stigmer/a1b2c3d4",
450
+ baseBranch: "main",
451
+ commitSha: "abc",
452
+ pullRequestUrl: "",
453
+ pullRequestNumber: 0,
454
+ diffSummary: "1 file changed",
455
+ phase: 1,
456
+ error: "",
457
+ $typeName: "ai.stigmer.agentic.agentexecution.v1.WorkspaceWriteBack",
458
+ } as any);
459
+
460
+ expect(sb.currentStatus.workspaceWriteBacks).toHaveLength(1);
461
+ expect(sb.forceNextUpdate).toBe(true);
462
+ });
463
+
464
+ it("upserts by workspaceEntryName", () => {
465
+ const sb = makeBuilder();
466
+
467
+ sb.addWriteBack({
468
+ workspaceEntryName: "my-app",
469
+ branchName: "stigmer/a1b2",
470
+ baseBranch: "main",
471
+ commitSha: "sha1",
472
+ pullRequestUrl: "",
473
+ pullRequestNumber: 0,
474
+ diffSummary: "",
475
+ phase: 1,
476
+ error: "",
477
+ $typeName: "ai.stigmer.agentic.agentexecution.v1.WorkspaceWriteBack",
478
+ } as any);
479
+
480
+ sb.addWriteBack({
481
+ workspaceEntryName: "my-app",
482
+ branchName: "stigmer/a1b2",
483
+ baseBranch: "main",
484
+ commitSha: "sha2",
485
+ pullRequestUrl: "https://github.com/pr/1",
486
+ pullRequestNumber: 1,
487
+ diffSummary: "2 files changed",
488
+ phase: 3,
489
+ error: "",
490
+ $typeName: "ai.stigmer.agentic.agentexecution.v1.WorkspaceWriteBack",
491
+ } as any);
492
+
493
+ expect(sb.currentStatus.workspaceWriteBacks).toHaveLength(1);
494
+ expect(sb.currentStatus.workspaceWriteBacks[0].commitSha).toBe("sha2");
495
+ expect(sb.currentStatus.workspaceWriteBacks[0].phase).toBe(3);
496
+ });
497
+
498
+ it("tracks multiple workspace entries independently", () => {
499
+ const sb = makeBuilder();
500
+
501
+ sb.addWriteBack({
502
+ workspaceEntryName: "frontend",
503
+ branchName: "stigmer/x",
504
+ baseBranch: "main",
505
+ commitSha: "a",
506
+ pullRequestUrl: "",
507
+ pullRequestNumber: 0,
508
+ diffSummary: "",
509
+ phase: 1,
510
+ error: "",
511
+ $typeName: "ai.stigmer.agentic.agentexecution.v1.WorkspaceWriteBack",
512
+ } as any);
513
+
514
+ sb.addWriteBack({
515
+ workspaceEntryName: "backend",
516
+ branchName: "stigmer/x",
517
+ baseBranch: "develop",
518
+ commitSha: "b",
519
+ pullRequestUrl: "",
520
+ pullRequestNumber: 0,
521
+ diffSummary: "",
522
+ phase: 2,
523
+ error: "",
524
+ $typeName: "ai.stigmer.agentic.agentexecution.v1.WorkspaceWriteBack",
525
+ } as any);
526
+
527
+ expect(sb.currentStatus.workspaceWriteBacks).toHaveLength(2);
528
+ });
529
+ });
530
+
531
+ // ─────────────────────────────────────────────────────────────────────────
532
+ // Approval provider integration
533
+ // ─────────────────────────────────────────────────────────────────────────
534
+
535
+ describe("approval provider integration", () => {
536
+ function makeApprovalProvider(
537
+ overrides: Partial<ApprovalPolicyProvider> = {},
538
+ ): ApprovalPolicyProvider {
539
+ return {
540
+ policies: new Map(),
541
+ toolServerMap: new Map(),
542
+ autoApproveAll: false,
543
+ ...overrides,
544
+ };
545
+ }
546
+
547
+ function policyFor(
548
+ serverSlug: string,
549
+ toolName: string,
550
+ message = "Approve?",
551
+ ): [string, MergedToolPolicy] {
552
+ return [
553
+ `${serverSlug}/${toolName}`,
554
+ {
555
+ toolName,
556
+ mcpServerSlug: serverSlug,
557
+ requiresApproval: true,
558
+ approvalMessage: message,
559
+ },
560
+ ];
561
+ }
562
+
563
+ it("sets tool status to WAITING_APPROVAL when policy requires it", () => {
564
+ const sb = makeBuilder();
565
+ const [key, policy] = policyFor("github", "create_issue");
566
+ sb.setApprovalProvider(makeApprovalProvider({
567
+ policies: new Map([[key, policy]]),
568
+ toolServerMap: new Map([["create_issue", "github"]]),
569
+ }));
570
+
571
+ sb.processEvent(chatStreamEvent("run-1", "I'll create an issue"));
572
+ sb.processEvent(toolStartEvent("tool-1", "create_issue", { title: "Bug fix" }));
573
+
574
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
575
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_WAITING_APPROVAL);
576
+ expect(tc.requiresApproval).toBe(true);
577
+ expect(tc.approvalMessage).toBe("Approve?");
578
+ expect(tc.mcpServerSlug).toBe("github");
579
+ });
580
+
581
+ it("sets phase to WAITING_FOR_APPROVAL when tool requires approval", () => {
582
+ const sb = makeBuilder();
583
+ const [key, policy] = policyFor("github", "push");
584
+ sb.setApprovalProvider(makeApprovalProvider({
585
+ policies: new Map([[key, policy]]),
586
+ toolServerMap: new Map([["push", "github"]]),
587
+ }));
588
+
589
+ sb.processEvent(chatStreamEvent("run-1", "pushing"));
590
+ sb.processEvent(toolStartEvent("tool-1", "push"));
591
+
592
+ expect(sb.currentStatus.phase).toBe(ExecutionPhase.EXECUTION_WAITING_FOR_APPROVAL);
593
+ });
594
+
595
+ it("does not require approval when autoApproveAll is true", () => {
596
+ const sb = makeBuilder();
597
+ const [key, policy] = policyFor("github", "delete_repo");
598
+ sb.setApprovalProvider(makeApprovalProvider({
599
+ policies: new Map([[key, policy]]),
600
+ toolServerMap: new Map([["delete_repo", "github"]]),
601
+ autoApproveAll: true,
602
+ }));
603
+
604
+ sb.processEvent(chatStreamEvent("run-1", "deleting"));
605
+ sb.processEvent(toolStartEvent("tool-1", "delete_repo"));
606
+
607
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
608
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_RUNNING);
609
+ expect(tc.requiresApproval).toBe(false);
610
+ });
611
+
612
+ it("sets mcpServerSlug even when autoApproveAll is true", () => {
613
+ const sb = makeBuilder();
614
+ sb.setApprovalProvider(makeApprovalProvider({
615
+ toolServerMap: new Map([["echo", "test-mcp-server"]]),
616
+ autoApproveAll: true,
617
+ }));
618
+
619
+ sb.processEvent(chatStreamEvent("run-1", "echoing"));
620
+ sb.processEvent(toolStartEvent("tool-1", "echo", { input: "hello" }));
621
+
622
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
623
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_RUNNING);
624
+ expect(tc.requiresApproval).toBe(false);
625
+ expect(tc.mcpServerSlug).toBe("test-mcp-server");
626
+ });
627
+
628
+ it("does not require approval for tools without policies", () => {
629
+ const sb = makeBuilder();
630
+ sb.setApprovalProvider(makeApprovalProvider({
631
+ policies: new Map(),
632
+ toolServerMap: new Map([["read", "filesystem"]]),
633
+ }));
634
+
635
+ sb.processEvent(chatStreamEvent("run-1", "reading"));
636
+ sb.processEvent(toolStartEvent("tool-1", "read"));
637
+
638
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
639
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_RUNNING);
640
+ expect(tc.mcpServerSlug).toBe("filesystem");
641
+ });
642
+
643
+ it("does not require approval for tools not in toolServerMap", () => {
644
+ const sb = makeBuilder();
645
+ const [key, policy] = policyFor("github", "create_pr");
646
+ sb.setApprovalProvider(makeApprovalProvider({
647
+ policies: new Map([[key, policy]]),
648
+ toolServerMap: new Map(),
649
+ }));
650
+
651
+ sb.processEvent(chatStreamEvent("run-1", "creating PR"));
652
+ sb.processEvent(toolStartEvent("tool-1", "create_pr"));
653
+
654
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
655
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_RUNNING);
656
+ });
657
+
658
+ it("resolves {{args.field}} placeholders in approval message", () => {
659
+ const sb = makeBuilder();
660
+ const [key, policy] = policyFor(
661
+ "github", "create_issue",
662
+ "Create issue '{{args.title}}' in {{args.repo}}?",
663
+ );
664
+ sb.setApprovalProvider(makeApprovalProvider({
665
+ policies: new Map([[key, policy]]),
666
+ toolServerMap: new Map([["create_issue", "github"]]),
667
+ }));
668
+
669
+ sb.processEvent(chatStreamEvent("run-1", "creating issue"));
670
+ sb.processEvent(toolStartEvent("tool-1", "create_issue", {
671
+ title: "Fix crash",
672
+ repo: "stigmer/stigmer",
673
+ }));
674
+
675
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
676
+ expect(tc.approvalMessage).toBe("Create issue 'Fix crash' in stigmer/stigmer?");
677
+ });
678
+
679
+ it("sets approvalRequestedAt when approval required", () => {
680
+ const sb = makeBuilder();
681
+ const [key, policy] = policyFor("db", "drop_table");
682
+ sb.setApprovalProvider(makeApprovalProvider({
683
+ policies: new Map([[key, policy]]),
684
+ toolServerMap: new Map([["drop_table", "db"]]),
685
+ }));
686
+
687
+ sb.processEvent(chatStreamEvent("run-1", "dropping"));
688
+ sb.processEvent(toolStartEvent("tool-1", "drop_table"));
689
+
690
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
691
+ expect(tc.approvalRequestedAt).toBeTruthy();
692
+ expect(tc.approvalRequestedAt).toContain("T");
693
+ });
694
+ });
695
+
696
+ // ─────────────────────────────────────────────────────────────────────────
697
+ // Args sanitization (argsPreview)
698
+ // ─────────────────────────────────────────────────────────────────────────
699
+
700
+ describe("args preview sanitization", () => {
701
+ function builderWithApproval(
702
+ toolName: string,
703
+ serverSlug: string,
704
+ ): StatusBuilder {
705
+ const sb = makeBuilder();
706
+ sb.setApprovalProvider({
707
+ policies: new Map([
708
+ [`${serverSlug}/${toolName}`, {
709
+ toolName,
710
+ mcpServerSlug: serverSlug,
711
+ requiresApproval: true,
712
+ approvalMessage: "Approve?",
713
+ }],
714
+ ]),
715
+ toolServerMap: new Map([[toolName, serverSlug]]),
716
+ autoApproveAll: false,
717
+ });
718
+ return sb;
719
+ }
720
+
721
+ it("redacts sensitive arg keys", () => {
722
+ const sb = builderWithApproval("connect", "db");
723
+ sb.processEvent(chatStreamEvent("run-1", "connecting"));
724
+ sb.processEvent(toolStartEvent("tool-1", "connect", {
725
+ host: "localhost",
726
+ password: "super-secret",
727
+ token: "jwt-token",
728
+ }));
729
+
730
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
731
+ const preview = JSON.parse(tc.argsPreview);
732
+ expect(preview.host).toBe("localhost");
733
+ expect(preview.password).toBe("[REDACTED]");
734
+ expect(preview.token).toBe("[REDACTED]");
735
+ });
736
+
737
+ it("redacts case-insensitively", () => {
738
+ const sb = builderWithApproval("call", "api");
739
+ sb.processEvent(chatStreamEvent("run-1", "calling"));
740
+ sb.processEvent(toolStartEvent("tool-1", "call", {
741
+ Authorization: "Bearer xyz",
742
+ API_KEY: "key-123",
743
+ }));
744
+
745
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
746
+ const preview = JSON.parse(tc.argsPreview);
747
+ expect(preview.Authorization).toBe("[REDACTED]");
748
+ expect(preview.API_KEY).toBe("[REDACTED]");
749
+ });
750
+
751
+ it("truncates long args preview", () => {
752
+ const sb = builderWithApproval("write", "fs");
753
+ sb.processEvent(chatStreamEvent("run-1", "writing"));
754
+ sb.processEvent(toolStartEvent("tool-1", "write", {
755
+ content: "x".repeat(1000),
756
+ }));
757
+
758
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
759
+ expect(tc.argsPreview.length).toBeLessThanOrEqual(501);
760
+ expect(tc.argsPreview.endsWith("…")).toBe(true);
761
+ });
762
+
763
+ it("does not set argsPreview for non-approval tools", () => {
764
+ const sb = makeBuilder();
765
+ sb.processEvent(chatStreamEvent("run-1", "reading"));
766
+ sb.processEvent(toolStartEvent("tool-1", "read", { path: "/foo" }));
767
+
768
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
769
+ expect(tc.argsPreview).toBe("");
770
+ });
771
+
772
+ it("sets argsPreview as empty when args serialization fails", () => {
773
+ const sb = builderWithApproval("call", "api");
774
+ sb.processEvent(chatStreamEvent("run-1", "calling"));
775
+
776
+ const circular: Record<string, unknown> = {};
777
+ circular.self = circular;
778
+ sb.processEvent(toolStartEvent("tool-1", "call", circular));
779
+
780
+ const tc = sb.currentStatus.messages[0].toolCalls[0];
781
+ expect(tc.argsPreview).toBe("");
782
+ });
783
+ });
784
+
785
+ // ─────────────────────────────────────────────────────────────────────────
786
+ // Namespace extraction
787
+ // ─────────────────────────────────────────────────────────────────────────
788
+
789
+ describe("namespace extraction", () => {
790
+ it("extracts namespace from langgraph_checkpoint_ns", () => {
791
+ const sb = makeBuilder();
792
+ sb.processEvent({
793
+ event: "on_chat_model_stream",
794
+ name: "ChatAnthropic",
795
+ run_id: "run-ns-1",
796
+ data: { chunk: { content: "hello" } },
797
+ metadata: { langgraph_checkpoint_ns: "tools:task-123|agent_node:inner" },
798
+ });
799
+
800
+ expect(sb.currentStatus.messages).toHaveLength(1);
801
+ });
802
+
803
+ it("extracts namespace from checkpoint_ns fallback", () => {
804
+ const sb = makeBuilder();
805
+ sb.processEvent({
806
+ event: "on_chat_model_stream",
807
+ name: "ChatAnthropic",
808
+ run_id: "run-ns-2",
809
+ data: { chunk: { content: "hello" } },
810
+ metadata: { checkpoint_ns: "some-namespace" },
811
+ });
812
+
813
+ expect(sb.currentStatus.messages).toHaveLength(1);
814
+ });
815
+
816
+ it("uses empty namespace when no metadata", () => {
817
+ const sb = makeBuilder();
818
+ sb.processEvent({
819
+ event: "on_chat_model_stream",
820
+ run_id: "run-ns-3",
821
+ data: { chunk: { content: "hello" } },
822
+ });
823
+
824
+ expect(sb.currentStatus.messages).toHaveLength(1);
825
+ });
826
+
827
+ it("different namespaces create separate message trees", () => {
828
+ const sb = makeBuilder();
829
+
830
+ sb.processEvent({
831
+ event: "on_chat_model_stream",
832
+ name: "ChatAnthropic",
833
+ run_id: "main-run",
834
+ data: { chunk: { content: "main text" } },
835
+ metadata: {},
836
+ });
837
+
838
+ sb.processEvent({
839
+ event: "on_chat_model_stream",
840
+ name: "ChatAnthropic",
841
+ run_id: "sub-run",
842
+ data: { chunk: { content: "sub text" } },
843
+ metadata: { langgraph_checkpoint_ns: "tools:task-1|agent_node:inner" },
844
+ });
845
+
846
+ expect(sb.currentStatus.messages).toHaveLength(2);
847
+ expect(sb.currentStatus.messages[0].content).toBe("main text");
848
+ expect(sb.currentStatus.messages[1].content).toBe("sub text");
849
+ });
850
+
851
+ it("same namespace reuses message within the same run_id", () => {
852
+ const sb = makeBuilder();
853
+ const ns = "tools:task-1|agent_node:inner";
854
+
855
+ sb.processEvent({
856
+ event: "on_chat_model_stream",
857
+ name: "ChatAnthropic",
858
+ run_id: "sub-run-1",
859
+ data: { chunk: { content: "part 1" } },
860
+ metadata: { langgraph_checkpoint_ns: ns },
861
+ });
862
+
863
+ sb.processEvent({
864
+ event: "on_chat_model_stream",
865
+ name: "ChatAnthropic",
866
+ run_id: "sub-run-1",
867
+ data: { chunk: { content: " part 2" } },
868
+ metadata: { langgraph_checkpoint_ns: ns },
869
+ });
870
+
871
+ expect(sb.currentStatus.messages).toHaveLength(1);
872
+ expect(sb.currentStatus.messages[0].content).toBe("part 1 part 2");
873
+ });
874
+ });
875
+
876
+ // ─────────────────────────────────────────────────────────────────────────
877
+ // Usage accumulation edge cases
878
+ // ─────────────────────────────────────────────────────────────────────────
879
+
880
+ describe("usage accumulation edge cases", () => {
881
+ it("handles missing usage_metadata gracefully", () => {
882
+ const sb = makeBuilder();
883
+ sb.processEvent(chatStreamEvent("run-1", "text"));
884
+ sb.processEvent(chatEndEvent("run-1"));
885
+
886
+ expect(sb.currentStatus.streamingUsage).toBeUndefined();
887
+ });
888
+
889
+ it("handles usage with zero tokens", () => {
890
+ const sb = makeBuilder();
891
+ sb.processEvent(chatStreamEvent("run-1", "text"));
892
+ sb.processEvent(chatEndEvent("run-1", {
893
+ input_tokens: 0,
894
+ output_tokens: 0,
895
+ }));
896
+
897
+ const usage = sb.currentStatus.streamingUsage!;
898
+ expect(usage.inputTokens).toBe(0n);
899
+ expect(usage.outputTokens).toBe(0n);
900
+ expect(usage.turnCount).toBe(1);
901
+ });
902
+
903
+ it("handles non-numeric token values gracefully", () => {
904
+ const sb = makeBuilder();
905
+ sb.processEvent(chatStreamEvent("run-1", "text"));
906
+ sb.processEvent(chatEndEvent("run-1", {
907
+ input_tokens: "not a number",
908
+ output_tokens: null,
909
+ cache_read_input_tokens: undefined,
910
+ }));
911
+
912
+ const usage = sb.currentStatus.streamingUsage!;
913
+ expect(usage.inputTokens).toBe(0n);
914
+ expect(usage.outputTokens).toBe(0n);
915
+ expect(usage.cacheReadTokens).toBe(0n);
916
+ expect(usage.turnCount).toBe(1);
917
+ });
918
+
919
+ it("handles bigint token values", () => {
920
+ const sb = makeBuilder();
921
+ sb.processEvent(chatStreamEvent("run-1", "text"));
922
+ sb.processEvent(chatEndEvent("run-1", {
923
+ input_tokens: 100n,
924
+ output_tokens: 50n,
925
+ }));
926
+
927
+ const usage = sb.currentStatus.streamingUsage!;
928
+ expect(usage.inputTokens).toBe(100n);
929
+ expect(usage.outputTokens).toBe(50n);
930
+ });
931
+
932
+ it("computes totalTokens as sum of all token fields", () => {
933
+ const sb = makeBuilder();
934
+ sb.processEvent(chatStreamEvent("run-1", "text"));
935
+ sb.processEvent(chatEndEvent("run-1", {
936
+ input_tokens: 100,
937
+ output_tokens: 50,
938
+ cache_read_input_tokens: 20,
939
+ cache_creation_input_tokens: 10,
940
+ }));
941
+
942
+ const usage = sb.currentStatus.streamingUsage!;
943
+ expect(usage.totalTokens).toBe(180n);
944
+ });
945
+
946
+ it("usage from data.usage_metadata takes precedence when output missing", () => {
947
+ const sb = makeBuilder();
948
+ sb.processEvent(chatStreamEvent("run-1", "text"));
949
+
950
+ sb.processEvent({
951
+ event: "on_chat_model_end",
952
+ name: "ChatAnthropic",
953
+ run_id: "run-1",
954
+ data: {
955
+ output: {},
956
+ usage_metadata: {
957
+ input_tokens: 42,
958
+ output_tokens: 13,
959
+ },
960
+ },
961
+ });
962
+
963
+ const usage = sb.currentStatus.streamingUsage!;
964
+ expect(usage.inputTokens).toBe(42n);
965
+ expect(usage.outputTokens).toBe(13n);
966
+ });
967
+
968
+ it("sets observedAt timestamp on usage snapshot", () => {
969
+ const sb = makeBuilder();
970
+ sb.processEvent(chatStreamEvent("run-1", "text"));
971
+ sb.processEvent(chatEndEvent("run-1", {
972
+ input_tokens: 10,
973
+ output_tokens: 5,
974
+ }));
975
+
976
+ const usage = sb.currentStatus.streamingUsage!;
977
+ expect(usage.observedAt).toBeTruthy();
978
+ expect(usage.observedAt).toContain("T");
979
+ });
980
+ });
981
+
982
+ // ─────────────────────────────────────────────────────────────────────────
983
+ // Error handling in event handlers
984
+ // ─────────────────────────────────────────────────────────────────────────
985
+
986
+ describe("event handler error resilience", () => {
987
+ it("does not throw when handler encounters an error", () => {
988
+ const sb = makeBuilder();
989
+ const errorSpy = vi.spyOn(console, "error").mockImplementation(() => {});
990
+
991
+ sb.processEvent({
992
+ event: "on_chat_model_stream",
993
+ run_id: "run-err",
994
+ data: { chunk: { content: { toString() { throw new Error("boom"); } } } },
995
+ });
996
+
997
+ errorSpy.mockRestore();
998
+ });
999
+
1000
+ it("logs error with execution context on handler failure", () => {
1001
+ const sb = makeBuilder();
1002
+ const errorSpy = vi.spyOn(console, "error").mockImplementation(() => {});
1003
+
1004
+ // Need a tool call in the map so handleToolEnd reaches the code
1005
+ // that accesses event.data.output (which we make throw)
1006
+ sb.processEvent(chatStreamEvent("run-1", "text"));
1007
+ sb.processEvent(toolStartEvent("bad-run", "read"));
1008
+
1009
+ sb.processEvent({
1010
+ event: "on_tool_end",
1011
+ name: "read",
1012
+ run_id: "bad-run",
1013
+ data: { get output() { throw new Error("property access failed"); } } as any,
1014
+ });
1015
+
1016
+ expect(errorSpy).toHaveBeenCalled();
1017
+ const logMsg = errorSpy.mock.calls[0][0] as string;
1018
+ expect(logMsg).toContain("exec-test");
1019
+ expect(logMsg).toContain("on_tool_end");
1020
+ expect(logMsg).toContain("bad-run");
1021
+
1022
+ errorSpy.mockRestore();
1023
+ });
1024
+ });
1025
+
1026
+ // ─────────────────────────────────────────────────────────────────────────
1027
+ // Tool result extraction edge cases
1028
+ // ─────────────────────────────────────────────────────────────────────────
1029
+
1030
+ describe("tool result extraction", () => {
1031
+ it("extracts string result directly", () => {
1032
+ const sb = makeBuilder();
1033
+ sb.processEvent(chatStreamEvent("run-1", "text"));
1034
+ sb.processEvent(toolStartEvent("tool-1", "read"));
1035
+ sb.processEvent(toolEndEvent("tool-1", "file contents"));
1036
+
1037
+ expect(sb.currentStatus.messages[0].toolCalls[0].result).toBe("file contents");
1038
+ });
1039
+
1040
+ it("extracts content from object result", () => {
1041
+ const sb = makeBuilder();
1042
+ sb.processEvent(chatStreamEvent("run-1", "text"));
1043
+ sb.processEvent(toolStartEvent("tool-1", "shell"));
1044
+ sb.processEvent(toolEndEvent("tool-1", { content: "stdout output" }));
1045
+
1046
+ expect(sb.currentStatus.messages[0].toolCalls[0].result).toBe("stdout output");
1047
+ });
1048
+
1049
+ it("JSON-serializes object without content field", () => {
1050
+ const sb = makeBuilder();
1051
+ sb.processEvent(chatStreamEvent("run-1", "text"));
1052
+ sb.processEvent(toolStartEvent("tool-1", "api"));
1053
+ sb.processEvent(toolEndEvent("tool-1", { status: 200, body: "ok" }));
1054
+
1055
+ const result = sb.currentStatus.messages[0].toolCalls[0].result;
1056
+ expect(JSON.parse(result)).toEqual({ status: 200, body: "ok" });
1057
+ });
1058
+
1059
+ it("JSON-serializes full data when output is undefined", () => {
1060
+ const sb = makeBuilder();
1061
+ sb.processEvent(chatStreamEvent("run-1", "text"));
1062
+ sb.processEvent(toolStartEvent("tool-1", "noop"));
1063
+ sb.processEvent({
1064
+ event: "on_tool_end",
1065
+ name: "noop",
1066
+ run_id: "tool-1",
1067
+ data: { something: "else" },
1068
+ });
1069
+
1070
+ const result = sb.currentStatus.messages[0].toolCalls[0].result;
1071
+ const parsed = JSON.parse(result);
1072
+ expect(parsed.something).toBe("else");
1073
+ });
1074
+
1075
+ it("returns serialization error for circular references", () => {
1076
+ const sb = makeBuilder();
1077
+ sb.processEvent(chatStreamEvent("run-1", "text"));
1078
+ sb.processEvent(toolStartEvent("tool-1", "bad"));
1079
+
1080
+ const circular: Record<string, unknown> = { a: 1 };
1081
+ circular.self = circular;
1082
+
1083
+ sb.processEvent({
1084
+ event: "on_tool_end",
1085
+ name: "bad",
1086
+ run_id: "tool-1",
1087
+ data: { output: circular },
1088
+ });
1089
+
1090
+ expect(sb.currentStatus.messages[0].toolCalls[0].result).toBe("[serialization error]");
1091
+ });
1092
+ });
1093
+
1094
+ // ─────────────────────────────────────────────────────────────────────────
1095
+ // Multiple concurrent tool calls
1096
+ // ─────────────────────────────────────────────────────────────────────────
1097
+
1098
+ describe("multiple concurrent tool calls", () => {
1099
+ it("tracks multiple running tools on the same message", () => {
1100
+ const sb = makeBuilder();
1101
+ sb.processEvent(chatStreamEvent("run-1", "I'll do two things"));
1102
+ sb.processEvent(toolStartEvent("tool-a", "read"));
1103
+ sb.processEvent(toolStartEvent("tool-b", "search"));
1104
+
1105
+ const msg = sb.currentStatus.messages[0];
1106
+ expect(msg.toolCalls).toHaveLength(2);
1107
+ expect(msg.toolCalls[0].name).toBe("read");
1108
+ expect(msg.toolCalls[1].name).toBe("search");
1109
+ expect(msg.toolCalls[0].status).toBe(ToolCallStatus.TOOL_CALL_RUNNING);
1110
+ expect(msg.toolCalls[1].status).toBe(ToolCallStatus.TOOL_CALL_RUNNING);
1111
+ });
1112
+
1113
+ it("completes tools independently", () => {
1114
+ const sb = makeBuilder();
1115
+ sb.processEvent(chatStreamEvent("run-1", "parallel work"));
1116
+ sb.processEvent(toolStartEvent("tool-a", "read"));
1117
+ sb.processEvent(toolStartEvent("tool-b", "search"));
1118
+
1119
+ sb.processEvent(toolEndEvent("tool-a", "file data"));
1120
+
1121
+ const msg = sb.currentStatus.messages[0];
1122
+ expect(msg.toolCalls[0].status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
1123
+ expect(msg.toolCalls[1].status).toBe(ToolCallStatus.TOOL_CALL_RUNNING);
1124
+
1125
+ sb.processEvent(toolEndEvent("tool-b", "search results"));
1126
+ expect(msg.toolCalls[1].status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
1127
+ });
1128
+
1129
+ it("handles interleaved tool start and end events", () => {
1130
+ const sb = makeBuilder();
1131
+ sb.processEvent(chatStreamEvent("run-1", "working"));
1132
+ sb.processEvent(toolStartEvent("tool-1", "read"));
1133
+ sb.processEvent(toolEndEvent("tool-1", "data 1"));
1134
+ sb.processEvent(toolStartEvent("tool-2", "write"));
1135
+ sb.processEvent(toolEndEvent("tool-2", "done"));
1136
+
1137
+ const msg = sb.currentStatus.messages[0];
1138
+ expect(msg.toolCalls).toHaveLength(2);
1139
+ expect(msg.toolCalls[0].status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
1140
+ expect(msg.toolCalls[1].status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
1141
+ });
1142
+ });
1143
+
1144
+ // ─────────────────────────────────────────────────────────────────────────
1145
+ // ExecutionState lifecycle
1146
+ // ─────────────────────────────────────────────────────────────────────────
1147
+
1148
+ describe("execution state lifecycle", () => {
1149
+ it("tool start times are cleaned up on tool end", () => {
1150
+ const sb = makeBuilder();
1151
+ sb.processEvent(chatStreamEvent("run-1", "text"));
1152
+ sb.processEvent(toolStartEvent("tool-1", "read"));
1153
+ sb.processEvent(toolEndEvent("tool-1", "done"));
1154
+
1155
+ // Internal state should have cleaned up the start time
1156
+ // Verified by the fact that there's no leak — the TS implementation
1157
+ // deletes from toolStartTimes in handleToolEnd
1158
+ });
1159
+
1160
+ it("handles chat_model_end without prior stream (no message exists)", () => {
1161
+ const sb = makeBuilder();
1162
+ sb.processEvent(chatEndEvent("orphan-run", {
1163
+ input_tokens: 50,
1164
+ output_tokens: 25,
1165
+ }));
1166
+
1167
+ // Usage should still accumulate even without a prior message
1168
+ const usage = sb.currentStatus.streamingUsage!;
1169
+ expect(usage.inputTokens).toBe(50n);
1170
+ expect(usage.outputTokens).toBe(25n);
1171
+ expect(usage.turnCount).toBe(1);
1172
+ });
1173
+
1174
+ it("handles tool_end for unknown run_id gracefully", () => {
1175
+ const sb = makeBuilder();
1176
+ sb.processEvent(toolEndEvent("never-started", "result"));
1177
+ // Should not throw, no messages or tool calls created
1178
+ expect(sb.currentStatus.messages).toHaveLength(0);
1179
+ });
1180
+
1181
+ it("creates AI message for tool_start without prior text stream", () => {
1182
+ const sb = makeBuilder();
1183
+ sb.processEvent(toolStartEvent("orphan-tool", "read"));
1184
+ expect(sb.currentStatus.messages).toHaveLength(1);
1185
+ expect(sb.currentStatus.messages[0].toolCalls).toHaveLength(1);
1186
+ expect(sb.currentStatus.messages[0].toolCalls[0].name).toBe("read");
1187
+ });
1188
+ });
1189
+
1190
+ // ─────────────────────────────────────────────────────────────────────────
1191
+ // Thinking + text interleaving
1192
+ // ─────────────────────────────────────────────────────────────────────────
1193
+
1194
+ describe("thinking and text interleaving", () => {
1195
+ it("creates separate messages for thinking and text in sequence", () => {
1196
+ const sb = makeBuilder();
1197
+
1198
+ sb.processEvent(chatStreamEvent("run-1", [
1199
+ { type: "thinking", thinking: "Let me think..." },
1200
+ ]));
1201
+ sb.processEvent(chatStreamEvent("run-1", [
1202
+ { type: "text", text: "Here is my answer" },
1203
+ ]));
1204
+
1205
+ expect(sb.currentStatus.messages).toHaveLength(2);
1206
+ expect(sb.currentStatus.messages[0].type).toBe(MessageType.MESSAGE_THINKING);
1207
+ expect(sb.currentStatus.messages[0].content).toBe("Let me think...");
1208
+ expect(sb.currentStatus.messages[1].type).toBe(MessageType.MESSAGE_AI);
1209
+ expect(sb.currentStatus.messages[1].content).toBe("Here is my answer");
1210
+ });
1211
+
1212
+ it("multiple thinking blocks accumulate in same thinking message", () => {
1213
+ const sb = makeBuilder();
1214
+
1215
+ sb.processEvent(chatStreamEvent("run-1", [
1216
+ { type: "thinking", thinking: "Step 1. " },
1217
+ ]));
1218
+ sb.processEvent(chatStreamEvent("run-1", [
1219
+ { type: "thinking", thinking: "Step 2. " },
1220
+ ]));
1221
+ sb.processEvent(chatStreamEvent("run-1", [
1222
+ { type: "thinking", thinking: "Step 3." },
1223
+ ]));
1224
+
1225
+ expect(sb.currentStatus.messages).toHaveLength(1);
1226
+ expect(sb.currentStatus.messages[0].content).toBe("Step 1. Step 2. Step 3.");
1227
+ });
1228
+
1229
+ it("thinking from different namespaces creates separate thinking messages", () => {
1230
+ const sb = makeBuilder();
1231
+
1232
+ sb.processEvent({
1233
+ event: "on_chat_model_stream",
1234
+ name: "ChatAnthropic",
1235
+ run_id: "run-main",
1236
+ data: { chunk: { content: [{ type: "thinking", thinking: "main thinking" }] } },
1237
+ metadata: {},
1238
+ });
1239
+
1240
+ sb.processEvent({
1241
+ event: "on_chat_model_stream",
1242
+ name: "ChatAnthropic",
1243
+ run_id: "run-sub",
1244
+ data: { chunk: { content: [{ type: "thinking", thinking: "sub thinking" }] } },
1245
+ metadata: { langgraph_checkpoint_ns: "tools:task-1" },
1246
+ });
1247
+
1248
+ expect(sb.currentStatus.messages).toHaveLength(2);
1249
+ expect(sb.currentStatus.messages[0].content).toBe("main thinking");
1250
+ expect(sb.currentStatus.messages[1].content).toBe("sub thinking");
1251
+ });
1252
+ });
1253
+
1254
+ // ─────────────────────────────────────────────────────────────────────────
1255
+ // Content block edge cases
1256
+ // ─────────────────────────────────────────────────────────────────────────
1257
+
1258
+ describe("content block edge cases", () => {
1259
+ it("ignores non-text non-thinking blocks", () => {
1260
+ const sb = makeBuilder();
1261
+ sb.processEvent(chatStreamEvent("run-1", [
1262
+ { type: "image_url", image_url: "https://example.com/img.png" },
1263
+ ]));
1264
+
1265
+ expect(sb.currentStatus.messages).toHaveLength(0);
1266
+ });
1267
+
1268
+ it("handles null content blocks gracefully", () => {
1269
+ const sb = makeBuilder();
1270
+ sb.processEvent(chatStreamEvent("run-1", [null as any]));
1271
+
1272
+ expect(sb.currentStatus.messages).toHaveLength(0);
1273
+ });
1274
+
1275
+ it("handles empty array content", () => {
1276
+ const sb = makeBuilder();
1277
+ sb.processEvent(chatStreamEvent("run-1", []));
1278
+
1279
+ expect(sb.currentStatus.messages).toHaveLength(0);
1280
+ });
1281
+
1282
+ it("handles numeric content gracefully", () => {
1283
+ const sb = makeBuilder();
1284
+ sb.processEvent(chatStreamEvent("run-1", 42 as any));
1285
+
1286
+ expect(sb.currentStatus.messages).toHaveLength(0);
1287
+ });
1288
+ });
1289
+
1290
+ // ═══════════════════════════════════════════════════════════════════════
1291
+ // GOLDEN SEQUENCES
1292
+ //
1293
+ // Realistic multi-event conversation sequences that assert on the
1294
+ // complete AgentExecutionStatus shape. These serve as the regression
1295
+ // contract for V3StatusBuilder in Phase 2 of the v3 streaming migration.
1296
+ //
1297
+ // Each sequence represents a real-world conversation pattern and
1298
+ // validates the final proto output end-to-end through StatusBuilder.
1299
+ // ═══════════════════════════════════════════════════════════════════════
1300
+
1301
+ describe("golden sequences", () => {
1302
+
1303
+ describe("plain chat — 2-turn text-only conversation", () => {
1304
+ it("produces 2 AI messages with accumulated usage", () => {
1305
+ const sb = makeBuilder();
1306
+
1307
+ // Turn 1: stream tokens → end
1308
+ sb.processEvent(chatStreamEvent("run-1", "Hello, "));
1309
+ sb.processEvent(chatStreamEvent("run-1", "I can help "));
1310
+ sb.processEvent(chatStreamEvent("run-1", "with that."));
1311
+ sb.processEvent(chatEndEvent("run-1", {
1312
+ input_tokens: 50,
1313
+ output_tokens: 12,
1314
+ }));
1315
+
1316
+ // Turn 2: stream tokens → end
1317
+ sb.processEvent(chatStreamEvent("run-2", "Here is "));
1318
+ sb.processEvent(chatStreamEvent("run-2", "more detail."));
1319
+ sb.processEvent(chatEndEvent("run-2", {
1320
+ input_tokens: 80,
1321
+ output_tokens: 8,
1322
+ }));
1323
+
1324
+ const status = sb.currentStatus;
1325
+
1326
+ // 2 AI messages, both finished streaming
1327
+ expect(status.messages).toHaveLength(2);
1328
+ expect(status.messages[0].type).toBe(MessageType.MESSAGE_AI);
1329
+ expect(status.messages[0].content).toBe("Hello, I can help with that.");
1330
+ expect(status.messages[0].isStreaming).toBe(false);
1331
+ expect(status.messages[0].toolCalls).toHaveLength(0);
1332
+
1333
+ expect(status.messages[1].type).toBe(MessageType.MESSAGE_AI);
1334
+ expect(status.messages[1].content).toBe("Here is more detail.");
1335
+ expect(status.messages[1].isStreaming).toBe(false);
1336
+ expect(status.messages[1].toolCalls).toHaveLength(0);
1337
+
1338
+ // Usage accumulated across turns
1339
+ const usage = status.streamingUsage!;
1340
+ expect(usage.inputTokens).toBe(130n);
1341
+ expect(usage.outputTokens).toBe(20n);
1342
+ expect(usage.turnCount).toBe(2);
1343
+ expect(usage.totalTokens).toBe(150n);
1344
+
1345
+ // Phase stays IN_PROGRESS (terminal phase set by caller)
1346
+ expect(status.phase).toBe(ExecutionPhase.EXECUTION_IN_PROGRESS);
1347
+ });
1348
+ });
1349
+
1350
+ describe("Anthropic thinking + text", () => {
1351
+ it("produces THINKING then AI messages with correct turn boundaries", () => {
1352
+ const sb = makeBuilder();
1353
+
1354
+ // Turn 1: thinking blocks → text blocks (same run_id)
1355
+ sb.processEvent(chatStreamEvent("run-1", [
1356
+ { type: "thinking", thinking: "Let me analyze " },
1357
+ ]));
1358
+ sb.processEvent(chatStreamEvent("run-1", [
1359
+ { type: "thinking", thinking: "this problem carefully." },
1360
+ ]));
1361
+ sb.processEvent(chatStreamEvent("run-1", [
1362
+ { type: "text", text: "Based on my analysis, " },
1363
+ ]));
1364
+ sb.processEvent(chatStreamEvent("run-1", [
1365
+ { type: "text", text: "here is the answer." },
1366
+ ]));
1367
+ sb.processEvent(chatEndEvent("run-1", {
1368
+ input_tokens: 200,
1369
+ output_tokens: 80,
1370
+ cache_read_input_tokens: 50,
1371
+ }));
1372
+
1373
+ // Turn 2: new run_id, text only (follow-up)
1374
+ sb.processEvent(chatStreamEvent("run-2", "Let me elaborate."));
1375
+ sb.processEvent(chatEndEvent("run-2", {
1376
+ input_tokens: 300,
1377
+ output_tokens: 40,
1378
+ }));
1379
+
1380
+ const status = sb.currentStatus;
1381
+
1382
+ // 3 messages: thinking, AI (turn 1), AI (turn 2)
1383
+ expect(status.messages).toHaveLength(3);
1384
+
1385
+ expect(status.messages[0].type).toBe(MessageType.MESSAGE_THINKING);
1386
+ expect(status.messages[0].content).toBe("Let me analyze this problem carefully.");
1387
+ expect(status.messages[0].isStreaming).toBe(true); // thinking messages aren't finalized by chat_model_end
1388
+
1389
+ expect(status.messages[1].type).toBe(MessageType.MESSAGE_AI);
1390
+ expect(status.messages[1].content).toBe("Based on my analysis, here is the answer.");
1391
+ expect(status.messages[1].isStreaming).toBe(false);
1392
+
1393
+ expect(status.messages[2].type).toBe(MessageType.MESSAGE_AI);
1394
+ expect(status.messages[2].content).toBe("Let me elaborate.");
1395
+ expect(status.messages[2].isStreaming).toBe(false);
1396
+
1397
+ // Usage across both turns with cache tokens
1398
+ const usage = status.streamingUsage!;
1399
+ expect(usage.inputTokens).toBe(500n);
1400
+ expect(usage.outputTokens).toBe(120n);
1401
+ expect(usage.cacheReadTokens).toBe(50n);
1402
+ expect(usage.turnCount).toBe(2);
1403
+ expect(usage.totalTokens).toBe(670n);
1404
+ });
1405
+ });
1406
+
1407
+ describe("single tool call — ReAct pattern", () => {
1408
+ it("produces AI message with tool call followed by response", () => {
1409
+ const sb = makeBuilder();
1410
+
1411
+ // Turn 1: model decides to call a tool
1412
+ sb.processEvent(chatStreamEvent("run-1", "I'll read the file for you."));
1413
+ sb.processEvent(chatEndEvent("run-1", {
1414
+ input_tokens: 100,
1415
+ output_tokens: 15,
1416
+ }));
1417
+
1418
+ // Tool execution
1419
+ sb.processEvent(toolStartEvent("tool-run-1", "read_file", { path: "/src/main.ts" }));
1420
+ sb.processEvent(toolEndEvent("tool-run-1", "export function main() { console.log('hello'); }"));
1421
+
1422
+ // Turn 2: model responds with tool result
1423
+ sb.processEvent(chatStreamEvent("run-2", "The file contains a main function that logs 'hello'."));
1424
+ sb.processEvent(chatEndEvent("run-2", {
1425
+ input_tokens: 200,
1426
+ output_tokens: 20,
1427
+ }));
1428
+
1429
+ const status = sb.currentStatus;
1430
+
1431
+ // 2 messages: AI with tool call, AI response
1432
+ expect(status.messages).toHaveLength(2);
1433
+
1434
+ const firstMsg = status.messages[0];
1435
+ expect(firstMsg.type).toBe(MessageType.MESSAGE_AI);
1436
+ expect(firstMsg.content).toBe("I'll read the file for you.");
1437
+ expect(firstMsg.isStreaming).toBe(false);
1438
+ expect(firstMsg.toolCalls).toHaveLength(1);
1439
+
1440
+ const tc = firstMsg.toolCalls[0];
1441
+ expect(tc.id).toBe("tool-run-1");
1442
+ expect(tc.name).toBe("read_file");
1443
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
1444
+ expect(tc.args).toEqual({ path: "/src/main.ts" });
1445
+ expect(tc.result).toBe("export function main() { console.log('hello'); }");
1446
+ expect(tc.error).toBe("");
1447
+ expect(tc.isStreaming).toBe(false);
1448
+ expect(tc.startedAt).toBeTruthy();
1449
+ expect(tc.completedAt).toBeTruthy();
1450
+
1451
+ const secondMsg = status.messages[1];
1452
+ expect(secondMsg.type).toBe(MessageType.MESSAGE_AI);
1453
+ expect(secondMsg.content).toBe("The file contains a main function that logs 'hello'.");
1454
+ expect(secondMsg.toolCalls).toHaveLength(0);
1455
+
1456
+ // Usage accumulated across both LLM turns
1457
+ const usage = status.streamingUsage!;
1458
+ expect(usage.inputTokens).toBe(300n);
1459
+ expect(usage.outputTokens).toBe(35n);
1460
+ expect(usage.turnCount).toBe(2);
1461
+ });
1462
+ });
1463
+
1464
+ describe("tool error — failed tool call", () => {
1465
+ it("marks tool as FAILED with error string", () => {
1466
+ const sb = makeBuilder();
1467
+
1468
+ sb.processEvent(chatStreamEvent("run-1", "I'll write to the file."));
1469
+ sb.processEvent(chatEndEvent("run-1", {
1470
+ input_tokens: 50,
1471
+ output_tokens: 10,
1472
+ }));
1473
+
1474
+ sb.processEvent(toolStartEvent("tool-run-1", "write_file", {
1475
+ path: "/etc/passwd",
1476
+ content: "malicious",
1477
+ }));
1478
+ sb.processEvent(toolEndEvent("tool-run-1", {
1479
+ error: "EACCES: permission denied, open '/etc/passwd'",
1480
+ }));
1481
+
1482
+ // Model recovers with error explanation
1483
+ sb.processEvent(chatStreamEvent("run-2", "I don't have permission to write to that file."));
1484
+ sb.processEvent(chatEndEvent("run-2", {
1485
+ input_tokens: 120,
1486
+ output_tokens: 18,
1487
+ }));
1488
+
1489
+ const status = sb.currentStatus;
1490
+
1491
+ expect(status.messages).toHaveLength(2);
1492
+
1493
+ const tc = status.messages[0].toolCalls[0];
1494
+ expect(tc.name).toBe("write_file");
1495
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_FAILED);
1496
+ expect(tc.error).toBe("EACCES: permission denied, open '/etc/passwd'");
1497
+ expect(tc.result).toBe("");
1498
+ expect(tc.completedAt).toBeTruthy();
1499
+
1500
+ expect(status.messages[1].content).toBe(
1501
+ "I don't have permission to write to that file.",
1502
+ );
1503
+
1504
+ const usage = status.streamingUsage!;
1505
+ expect(usage.turnCount).toBe(2);
1506
+ });
1507
+ });
1508
+
1509
+ describe("multi-tool concurrent — parallel tool execution", () => {
1510
+ it("tracks concurrent tools independently on same message", () => {
1511
+ const sb = makeBuilder();
1512
+
1513
+ // Model decides to call two tools in parallel
1514
+ sb.processEvent(chatStreamEvent("run-1", "I'll search both files."));
1515
+ sb.processEvent(chatEndEvent("run-1", {
1516
+ input_tokens: 60,
1517
+ output_tokens: 10,
1518
+ }));
1519
+
1520
+ sb.processEvent(toolStartEvent("tool-a", "read_file", { path: "/a.ts" }));
1521
+ sb.processEvent(toolStartEvent("tool-b", "read_file", { path: "/b.ts" }));
1522
+
1523
+ // Tool B finishes first (out of order)
1524
+ sb.processEvent(toolEndEvent("tool-b", "content of b.ts"));
1525
+ sb.processEvent(toolEndEvent("tool-a", "content of a.ts"));
1526
+
1527
+ // Model synthesizes results
1528
+ sb.processEvent(chatStreamEvent("run-2", "Both files have been read."));
1529
+ sb.processEvent(chatEndEvent("run-2", {
1530
+ input_tokens: 180,
1531
+ output_tokens: 12,
1532
+ }));
1533
+
1534
+ const status = sb.currentStatus;
1535
+
1536
+ expect(status.messages).toHaveLength(2);
1537
+
1538
+ const toolMsg = status.messages[0];
1539
+ expect(toolMsg.toolCalls).toHaveLength(2);
1540
+
1541
+ const tcA = toolMsg.toolCalls.find(t => t.id === "tool-a")!;
1542
+ const tcB = toolMsg.toolCalls.find(t => t.id === "tool-b")!;
1543
+
1544
+ expect(tcA.status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
1545
+ expect(tcA.result).toBe("content of a.ts");
1546
+ expect(tcB.status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
1547
+ expect(tcB.result).toBe("content of b.ts");
1548
+
1549
+ // Both completed independently
1550
+ expect(tcA.completedAt).toBeTruthy();
1551
+ expect(tcB.completedAt).toBeTruthy();
1552
+ });
1553
+ });
1554
+
1555
+ describe("HITL approval gate — tool requiring approval", () => {
1556
+ it("transitions phase to WAITING_FOR_APPROVAL with approval fields", () => {
1557
+ const sb = makeBuilder();
1558
+
1559
+ sb.setApprovalProvider({
1560
+ policies: new Map([
1561
+ ["github/create_pull_request", {
1562
+ toolName: "create_pull_request",
1563
+ mcpServerSlug: "github",
1564
+ requiresApproval: true,
1565
+ approvalMessage: "Create PR '{{args.title}}' in {{args.repo}}?",
1566
+ }],
1567
+ ]),
1568
+ toolServerMap: new Map([["create_pull_request", "github"]]),
1569
+ autoApproveAll: false,
1570
+ });
1571
+
1572
+ // Model streams text then calls the approval-gated tool
1573
+ sb.processEvent(chatStreamEvent("run-1", "I'll create a pull request."));
1574
+ sb.processEvent(chatEndEvent("run-1", {
1575
+ input_tokens: 90,
1576
+ output_tokens: 12,
1577
+ }));
1578
+
1579
+ sb.processEvent(toolStartEvent("tool-pr-1", "create_pull_request", {
1580
+ title: "Fix login bug",
1581
+ repo: "stigmer/stigmer",
1582
+ }));
1583
+
1584
+ const status = sb.currentStatus;
1585
+
1586
+ // Phase transitioned to waiting
1587
+ expect(status.phase).toBe(ExecutionPhase.EXECUTION_WAITING_FOR_APPROVAL);
1588
+
1589
+ const tc = status.messages[0].toolCalls[0];
1590
+ expect(tc.status).toBe(ToolCallStatus.TOOL_CALL_WAITING_APPROVAL);
1591
+ expect(tc.requiresApproval).toBe(true);
1592
+ expect(tc.approvalMessage).toBe(
1593
+ "Create PR 'Fix login bug' in stigmer/stigmer?",
1594
+ );
1595
+ expect(tc.mcpServerSlug).toBe("github");
1596
+ expect(tc.approvalRequestedAt).toBeTruthy();
1597
+ expect(tc.argsPreview).toBeTruthy();
1598
+
1599
+ // Args preview has sensitive fields redacted
1600
+ const preview = JSON.parse(tc.argsPreview);
1601
+ expect(preview.title).toBe("Fix login bug");
1602
+ expect(preview.repo).toBe("stigmer/stigmer");
1603
+ });
1604
+ });
1605
+
1606
+ describe("usage accumulation — 3-turn with cache tokens", () => {
1607
+ it("accumulates input, output, cache read/write across 3 turns", () => {
1608
+ const sb = makeBuilder();
1609
+
1610
+ // Turn 1: initial prompt
1611
+ sb.processEvent(chatStreamEvent("run-1", "First."));
1612
+ sb.processEvent(chatEndEvent("run-1", {
1613
+ input_tokens: 100,
1614
+ output_tokens: 5,
1615
+ cache_read_input_tokens: 0,
1616
+ cache_creation_input_tokens: 80,
1617
+ }));
1618
+
1619
+ // Turn 2: tool call turn
1620
+ sb.processEvent(chatStreamEvent("run-2", "Reading."));
1621
+ sb.processEvent(chatEndEvent("run-2", {
1622
+ input_tokens: 200,
1623
+ output_tokens: 8,
1624
+ cache_read_input_tokens: 80,
1625
+ cache_creation_input_tokens: 0,
1626
+ }));
1627
+
1628
+ // Turn 3: final response
1629
+ sb.processEvent(chatStreamEvent("run-3", "Done."));
1630
+ sb.processEvent(chatEndEvent("run-3", {
1631
+ input_tokens: 350,
1632
+ output_tokens: 30,
1633
+ cache_read_input_tokens: 80,
1634
+ cache_creation_input_tokens: 20,
1635
+ }));
1636
+
1637
+ const usage = sb.currentStatus.streamingUsage!;
1638
+
1639
+ expect(usage.inputTokens).toBe(650n);
1640
+ expect(usage.outputTokens).toBe(43n);
1641
+ expect(usage.cacheReadTokens).toBe(160n);
1642
+ expect(usage.cacheWriteTokens).toBe(100n);
1643
+ expect(usage.totalTokens).toBe(953n);
1644
+ expect(usage.turnCount).toBe(3);
1645
+ expect(usage.observedAt).toBeTruthy();
1646
+ });
1647
+ });
1648
+
1649
+ describe("namespace isolation — parent + subagent", () => {
1650
+ it("creates separate message trees by namespace", () => {
1651
+ const sb = makeBuilder();
1652
+
1653
+ const parentNs = "";
1654
+ const subagentNs = "tools:delegate-123|agent_node:researcher";
1655
+
1656
+ // Parent agent starts
1657
+ sb.processEvent({
1658
+ event: "on_chat_model_stream",
1659
+ name: "ChatAnthropic",
1660
+ run_id: "parent-run-1",
1661
+ data: { chunk: { content: "I'll delegate this research task." } },
1662
+ metadata: {},
1663
+ });
1664
+ sb.processEvent({
1665
+ event: "on_chat_model_end",
1666
+ name: "ChatAnthropic",
1667
+ run_id: "parent-run-1",
1668
+ data: {
1669
+ output: { content: "done" },
1670
+ usage_metadata: { input_tokens: 100, output_tokens: 15 },
1671
+ },
1672
+ });
1673
+
1674
+ // Subagent thinking
1675
+ sb.processEvent({
1676
+ event: "on_chat_model_stream",
1677
+ name: "ChatAnthropic",
1678
+ run_id: "sub-run-1",
1679
+ data: { chunk: { content: [{ type: "thinking", thinking: "Researching the topic..." }] } },
1680
+ metadata: { langgraph_checkpoint_ns: subagentNs },
1681
+ });
1682
+
1683
+ // Subagent text response
1684
+ sb.processEvent({
1685
+ event: "on_chat_model_stream",
1686
+ name: "ChatAnthropic",
1687
+ run_id: "sub-run-1",
1688
+ data: { chunk: { content: [{ type: "text", text: "I found the following results." }] } },
1689
+ metadata: { langgraph_checkpoint_ns: subagentNs },
1690
+ });
1691
+ sb.processEvent({
1692
+ event: "on_chat_model_end",
1693
+ name: "ChatAnthropic",
1694
+ run_id: "sub-run-1",
1695
+ data: {
1696
+ output: { content: "done" },
1697
+ usage_metadata: { input_tokens: 150, output_tokens: 25 },
1698
+ },
1699
+ metadata: { langgraph_checkpoint_ns: subagentNs },
1700
+ });
1701
+
1702
+ // Subagent tool call
1703
+ sb.processEvent({
1704
+ event: "on_tool_start",
1705
+ name: "web_search",
1706
+ run_id: "sub-tool-1",
1707
+ data: { input: { query: "LangGraph v3 streaming" } },
1708
+ metadata: { langgraph_checkpoint_ns: subagentNs },
1709
+ });
1710
+ sb.processEvent({
1711
+ event: "on_tool_end",
1712
+ name: "web_search",
1713
+ run_id: "sub-tool-1",
1714
+ data: { output: "3 results found" },
1715
+ metadata: { langgraph_checkpoint_ns: subagentNs },
1716
+ });
1717
+
1718
+ // Parent resumes after delegation
1719
+ sb.processEvent({
1720
+ event: "on_chat_model_stream",
1721
+ name: "ChatAnthropic",
1722
+ run_id: "parent-run-2",
1723
+ data: { chunk: { content: "The research is complete." } },
1724
+ metadata: {},
1725
+ });
1726
+ sb.processEvent({
1727
+ event: "on_chat_model_end",
1728
+ name: "ChatAnthropic",
1729
+ run_id: "parent-run-2",
1730
+ data: {
1731
+ output: { content: "done" },
1732
+ usage_metadata: { input_tokens: 400, output_tokens: 10 },
1733
+ },
1734
+ });
1735
+
1736
+ const status = sb.currentStatus;
1737
+
1738
+ // 5 messages: parent-text, sub-thinking, sub-text, parent-text-2
1739
+ // (sub-tool attaches to sub-text message)
1740
+ expect(status.messages).toHaveLength(4);
1741
+
1742
+ // Parent turn 1
1743
+ expect(status.messages[0].type).toBe(MessageType.MESSAGE_AI);
1744
+ expect(status.messages[0].content).toBe("I'll delegate this research task.");
1745
+ expect(status.messages[0].toolCalls).toHaveLength(0);
1746
+
1747
+ // Subagent thinking
1748
+ expect(status.messages[1].type).toBe(MessageType.MESSAGE_THINKING);
1749
+ expect(status.messages[1].content).toBe("Researching the topic...");
1750
+
1751
+ // Subagent text + tool call
1752
+ expect(status.messages[2].type).toBe(MessageType.MESSAGE_AI);
1753
+ expect(status.messages[2].content).toBe("I found the following results.");
1754
+ expect(status.messages[2].toolCalls).toHaveLength(1);
1755
+ expect(status.messages[2].toolCalls[0].name).toBe("web_search");
1756
+ expect(status.messages[2].toolCalls[0].status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
1757
+ expect(status.messages[2].toolCalls[0].result).toBe("3 results found");
1758
+
1759
+ // Parent turn 2
1760
+ expect(status.messages[3].type).toBe(MessageType.MESSAGE_AI);
1761
+ expect(status.messages[3].content).toBe("The research is complete.");
1762
+
1763
+ // Usage accumulated across ALL namespaces
1764
+ const usage = status.streamingUsage!;
1765
+ expect(usage.inputTokens).toBe(650n);
1766
+ expect(usage.outputTokens).toBe(50n);
1767
+ expect(usage.turnCount).toBe(3);
1768
+ });
1769
+ });
1770
+ });
1771
+ });