@machina.ai/cell-cli-core 1.10.0-rc1 → 1.13.0-rc1

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 (549) hide show
  1. package/dist/index.d.ts +5 -0
  2. package/dist/index.js +4 -0
  3. package/dist/index.js.map +1 -1
  4. package/dist/package.json +13 -5
  5. package/dist/src/agents/codebase-investigator.js +2 -5
  6. package/dist/src/agents/codebase-investigator.js.map +1 -1
  7. package/dist/src/agents/executor.d.ts +19 -0
  8. package/dist/src/agents/executor.js +234 -46
  9. package/dist/src/agents/executor.js.map +1 -1
  10. package/dist/src/agents/executor.test.js +371 -40
  11. package/dist/src/agents/executor.test.js.map +1 -1
  12. package/dist/src/agents/registry.js +4 -3
  13. package/dist/src/agents/registry.js.map +1 -1
  14. package/dist/src/agents/subagent-tool-wrapper.test.js +2 -4
  15. package/dist/src/agents/subagent-tool-wrapper.test.js.map +1 -1
  16. package/dist/src/agents/types.d.ts +2 -1
  17. package/dist/src/agents/types.js +1 -0
  18. package/dist/src/agents/types.js.map +1 -1
  19. package/dist/src/code_assist/converter.d.ts +1 -0
  20. package/dist/src/code_assist/converter.js +1 -0
  21. package/dist/src/code_assist/converter.js.map +1 -1
  22. package/dist/src/code_assist/converter.test.js +19 -0
  23. package/dist/src/code_assist/converter.test.js.map +1 -1
  24. package/dist/src/code_assist/experiments/client_metadata.d.ts +12 -0
  25. package/dist/src/code_assist/experiments/client_metadata.js +49 -0
  26. package/dist/src/code_assist/experiments/client_metadata.js.map +1 -0
  27. package/dist/src/code_assist/experiments/experiments.d.ts +17 -0
  28. package/dist/src/code_assist/experiments/experiments.js +36 -0
  29. package/dist/src/code_assist/experiments/experiments.js.map +1 -0
  30. package/dist/src/code_assist/experiments/types.d.ts +35 -0
  31. package/dist/src/code_assist/experiments/types.js +7 -0
  32. package/dist/src/code_assist/experiments/types.js.map +1 -0
  33. package/dist/src/code_assist/oauth-credential-storage.js +5 -4
  34. package/dist/src/code_assist/oauth-credential-storage.js.map +1 -1
  35. package/dist/src/code_assist/oauth-credential-storage.test.js +15 -3
  36. package/dist/src/code_assist/oauth-credential-storage.test.js.map +1 -1
  37. package/dist/src/code_assist/oauth2.d.ts +2 -2
  38. package/dist/src/code_assist/oauth2.js +64 -51
  39. package/dist/src/code_assist/oauth2.js.map +1 -1
  40. package/dist/src/code_assist/oauth2.test.js +65 -33
  41. package/dist/src/code_assist/oauth2.test.js.map +1 -1
  42. package/dist/src/code_assist/server.d.ts +6 -4
  43. package/dist/src/code_assist/server.js +11 -0
  44. package/dist/src/code_assist/server.js.map +1 -1
  45. package/dist/src/code_assist/server.test.js +17 -0
  46. package/dist/src/code_assist/server.test.js.map +1 -1
  47. package/dist/src/code_assist/setup.d.ts +2 -2
  48. package/dist/src/code_assist/setup.js.map +1 -1
  49. package/dist/src/code_assist/types.d.ts +1 -1
  50. package/dist/src/code_assist/types.js.map +1 -1
  51. package/dist/src/commands/extensions.d.ts +7 -0
  52. package/dist/src/commands/extensions.js +9 -0
  53. package/dist/src/commands/extensions.js.map +1 -0
  54. package/dist/src/commands/extensions.test.js +19 -0
  55. package/dist/src/commands/extensions.test.js.map +1 -0
  56. package/dist/src/config/config.d.ts +81 -32
  57. package/dist/src/config/config.js +193 -66
  58. package/dist/src/config/config.js.map +1 -1
  59. package/dist/src/config/config.test.js +115 -36
  60. package/dist/src/config/config.test.js.map +1 -1
  61. package/dist/src/config/models.d.ts +1 -1
  62. package/dist/src/config/models.js +2 -2
  63. package/dist/src/config/models.js.map +1 -1
  64. package/dist/src/config/storage.d.ts +3 -0
  65. package/dist/src/config/storage.js +20 -0
  66. package/dist/src/config/storage.js.map +1 -1
  67. package/dist/src/confirmation-bus/message-bus.d.ts +2 -1
  68. package/dist/src/confirmation-bus/message-bus.js +7 -1
  69. package/dist/src/confirmation-bus/message-bus.js.map +1 -1
  70. package/dist/src/confirmation-bus/types.d.ts +12 -2
  71. package/dist/src/confirmation-bus/types.js +1 -0
  72. package/dist/src/confirmation-bus/types.js.map +1 -1
  73. package/dist/src/core/apiKeyCredentialStorage.d.ts +17 -0
  74. package/dist/src/core/apiKeyCredentialStorage.js +64 -0
  75. package/dist/src/core/apiKeyCredentialStorage.js.map +1 -0
  76. package/dist/src/core/apiKeyCredentialStorage.test.d.ts +6 -0
  77. package/dist/src/core/apiKeyCredentialStorage.test.js +71 -0
  78. package/dist/src/core/apiKeyCredentialStorage.test.js.map +1 -0
  79. package/dist/src/core/client.d.ts +2 -11
  80. package/dist/src/core/client.js +31 -170
  81. package/dist/src/core/client.js.map +1 -1
  82. package/dist/src/core/client.test.js +107 -429
  83. package/dist/src/core/client.test.js.map +1 -1
  84. package/dist/src/core/contentGenerator.js +64 -59
  85. package/dist/src/core/contentGenerator.js.map +1 -1
  86. package/dist/src/core/contentGenerator.test.js +38 -4
  87. package/dist/src/core/contentGenerator.test.js.map +1 -1
  88. package/dist/src/core/coreToolScheduler.d.ts +8 -2
  89. package/dist/src/core/coreToolScheduler.js +337 -172
  90. package/dist/src/core/coreToolScheduler.js.map +1 -1
  91. package/dist/src/core/coreToolScheduler.test.js +363 -12
  92. package/dist/src/core/coreToolScheduler.test.js.map +1 -1
  93. package/dist/src/core/fakeContentGenerator.d.ts +33 -0
  94. package/dist/src/core/fakeContentGenerator.js +58 -0
  95. package/dist/src/core/fakeContentGenerator.js.map +1 -0
  96. package/dist/src/core/fakeContentGenerator.test.d.ts +6 -0
  97. package/dist/src/core/fakeContentGenerator.test.js +127 -0
  98. package/dist/src/core/fakeContentGenerator.test.js.map +1 -0
  99. package/dist/src/core/geminiChat.d.ts +2 -0
  100. package/dist/src/core/geminiChat.js +7 -2
  101. package/dist/src/core/geminiChat.js.map +1 -1
  102. package/dist/src/core/geminiChat.test.js +15 -3
  103. package/dist/src/core/geminiChat.test.js.map +1 -1
  104. package/dist/src/core/logger.js +21 -19
  105. package/dist/src/core/logger.js.map +1 -1
  106. package/dist/src/core/loggingContentGenerator.d.ts +1 -0
  107. package/dist/src/core/loggingContentGenerator.js +113 -33
  108. package/dist/src/core/loggingContentGenerator.js.map +1 -1
  109. package/dist/src/core/nonInteractiveToolExecutor.js +5 -4
  110. package/dist/src/core/nonInteractiveToolExecutor.js.map +1 -1
  111. package/dist/src/core/nonInteractiveToolExecutor.test.js +3 -0
  112. package/dist/src/core/nonInteractiveToolExecutor.test.js.map +1 -1
  113. package/dist/src/core/prompts.js +115 -72
  114. package/dist/src/core/prompts.js.map +1 -1
  115. package/dist/src/core/prompts.test.js +30 -108
  116. package/dist/src/core/prompts.test.js.map +1 -1
  117. package/dist/src/core/recordingContentGenerator.d.ts +18 -0
  118. package/dist/src/core/recordingContentGenerator.js +77 -0
  119. package/dist/src/core/recordingContentGenerator.js.map +1 -0
  120. package/dist/src/core/recordingContentGenerator.test.d.ts +6 -0
  121. package/dist/src/core/recordingContentGenerator.test.js +101 -0
  122. package/dist/src/core/recordingContentGenerator.test.js.map +1 -0
  123. package/dist/src/core/turn.d.ts +2 -0
  124. package/dist/src/core/turn.js +3 -1
  125. package/dist/src/core/turn.js.map +1 -1
  126. package/dist/src/core/turn.test.js +48 -0
  127. package/dist/src/core/turn.test.js.map +1 -1
  128. package/dist/src/fallback/handler.js +2 -0
  129. package/dist/src/fallback/handler.js.map +1 -1
  130. package/dist/src/generated/git-commit.d.ts +2 -2
  131. package/dist/src/generated/git-commit.js +2 -2
  132. package/dist/src/hooks/hookPlanner.d.ts +46 -0
  133. package/dist/src/hooks/hookPlanner.js +108 -0
  134. package/dist/src/hooks/hookPlanner.js.map +1 -0
  135. package/dist/src/hooks/hookPlanner.test.d.ts +6 -0
  136. package/dist/src/hooks/hookPlanner.test.js +255 -0
  137. package/dist/src/hooks/hookPlanner.test.js.map +1 -0
  138. package/dist/src/hooks/hookRegistry.d.ts +87 -0
  139. package/dist/src/hooks/hookRegistry.js +198 -0
  140. package/dist/src/hooks/hookRegistry.js.map +1 -0
  141. package/dist/src/hooks/hookRegistry.test.d.ts +6 -0
  142. package/dist/src/hooks/hookRegistry.test.js +341 -0
  143. package/dist/src/hooks/hookRegistry.test.js.map +1 -0
  144. package/dist/src/hooks/hookTranslator.d.ts +113 -0
  145. package/dist/src/hooks/hookTranslator.js +232 -0
  146. package/dist/src/hooks/hookTranslator.js.map +1 -0
  147. package/dist/src/hooks/hookTranslator.test.d.ts +6 -0
  148. package/dist/src/hooks/hookTranslator.test.js +192 -0
  149. package/dist/src/hooks/hookTranslator.test.js.map +1 -0
  150. package/dist/src/hooks/types.d.ts +384 -0
  151. package/dist/src/hooks/types.js +284 -0
  152. package/dist/src/hooks/types.js.map +1 -0
  153. package/dist/src/hooks/types.test.d.ts +6 -0
  154. package/dist/src/hooks/types.test.js +35 -0
  155. package/dist/src/hooks/types.test.js.map +1 -0
  156. package/dist/src/ide/ide-client.js +2 -1
  157. package/dist/src/ide/ide-client.js.map +1 -1
  158. package/dist/src/index.d.ts +15 -0
  159. package/dist/src/index.js +18 -0
  160. package/dist/src/index.js.map +1 -1
  161. package/dist/src/mcp/google-auth-provider.d.ts +2 -0
  162. package/dist/src/mcp/google-auth-provider.js +21 -3
  163. package/dist/src/mcp/google-auth-provider.js.map +1 -1
  164. package/dist/src/mcp/google-auth-provider.test.js +42 -9
  165. package/dist/src/mcp/google-auth-provider.test.js.map +1 -1
  166. package/dist/src/mcp/mcpLauncher.d.ts +26 -0
  167. package/dist/src/mcp/mcpLauncher.js +238 -0
  168. package/dist/src/mcp/mcpLauncher.js.map +1 -0
  169. package/dist/src/mcp/oauth-provider.d.ts +8 -5
  170. package/dist/src/mcp/oauth-provider.js +140 -55
  171. package/dist/src/mcp/oauth-provider.js.map +1 -1
  172. package/dist/src/mcp/oauth-provider.test.js +191 -2
  173. package/dist/src/mcp/oauth-provider.test.js.map +1 -1
  174. package/dist/src/mcp/oauth-token-storage.js +5 -4
  175. package/dist/src/mcp/oauth-token-storage.js.map +1 -1
  176. package/dist/src/mcp/oauth-token-storage.test.js +17 -11
  177. package/dist/src/mcp/oauth-token-storage.test.js.map +1 -1
  178. package/dist/src/mcp/oauth-utils.d.ts +7 -0
  179. package/dist/src/mcp/oauth-utils.js +28 -8
  180. package/dist/src/mcp/oauth-utils.js.map +1 -1
  181. package/dist/src/mcp/oauth-utils.test.js +45 -2
  182. package/dist/src/mcp/oauth-utils.test.js.map +1 -1
  183. package/dist/src/mcp/sa-impersonation-provider.d.ts +0 -6
  184. package/dist/src/mcp/sa-impersonation-provider.js +6 -23
  185. package/dist/src/mcp/sa-impersonation-provider.js.map +1 -1
  186. package/dist/src/mcp/token-storage/base-token-storage.test.js +75 -84
  187. package/dist/src/mcp/token-storage/base-token-storage.test.js.map +1 -1
  188. package/dist/src/mcp/token-storage/file-token-storage.js +1 -1
  189. package/dist/src/mcp/token-storage/file-token-storage.js.map +1 -1
  190. package/dist/src/mcp/token-storage/file-token-storage.test.js +7 -5
  191. package/dist/src/mcp/token-storage/file-token-storage.test.js.map +1 -1
  192. package/dist/src/mcp/token-storage/hybrid-token-storage.js +1 -1
  193. package/dist/src/mcp/token-storage/hybrid-token-storage.js.map +1 -1
  194. package/dist/src/mcp/token-storage/hybrid-token-storage.test.js +2 -2
  195. package/dist/src/mcp/token-storage/hybrid-token-storage.test.js.map +1 -1
  196. package/dist/src/mcp/token-storage/keychain-token-storage.d.ts +6 -2
  197. package/dist/src/mcp/token-storage/keychain-token-storage.js +63 -7
  198. package/dist/src/mcp/token-storage/keychain-token-storage.js.map +1 -1
  199. package/dist/src/mcp/token-storage/keychain-token-storage.test.js +54 -3
  200. package/dist/src/mcp/token-storage/keychain-token-storage.test.js.map +1 -1
  201. package/dist/src/mcp/token-storage/types.d.ts +6 -0
  202. package/dist/src/mcp/token-storage/types.js.map +1 -1
  203. package/dist/src/output/stream-json-formatter.d.ts +32 -0
  204. package/dist/src/output/stream-json-formatter.js +52 -0
  205. package/dist/src/output/stream-json-formatter.js.map +1 -0
  206. package/dist/src/output/stream-json-formatter.test.d.ts +6 -0
  207. package/dist/src/output/stream-json-formatter.test.js +479 -0
  208. package/dist/src/output/stream-json-formatter.test.js.map +1 -0
  209. package/dist/src/output/types.d.ts +63 -1
  210. package/dist/src/output/types.js +11 -0
  211. package/dist/src/output/types.js.map +1 -1
  212. package/dist/src/policy/config.d.ts +31 -0
  213. package/dist/src/policy/config.js +197 -0
  214. package/dist/src/policy/config.js.map +1 -0
  215. package/dist/src/policy/config.test.d.ts +6 -0
  216. package/dist/src/policy/config.test.js +404 -0
  217. package/dist/src/policy/config.test.js.map +1 -0
  218. package/dist/src/policy/index.d.ts +2 -0
  219. package/dist/src/policy/index.js +2 -0
  220. package/dist/src/policy/index.js.map +1 -1
  221. package/dist/src/policy/policies/read-only.toml +56 -0
  222. package/dist/src/policy/policies/write.toml +63 -0
  223. package/dist/src/policy/policies/yolo.toml +31 -0
  224. package/dist/src/policy/policy-engine.js +4 -0
  225. package/dist/src/policy/policy-engine.js.map +1 -1
  226. package/dist/src/policy/toml-loader.d.ts +46 -0
  227. package/dist/src/policy/toml-loader.js +314 -0
  228. package/dist/src/policy/toml-loader.js.map +1 -0
  229. package/dist/src/policy/toml-loader.test.d.ts +6 -0
  230. package/dist/src/policy/toml-loader.test.js +522 -0
  231. package/dist/src/policy/toml-loader.test.js.map +1 -0
  232. package/dist/src/policy/types.d.ts +18 -0
  233. package/dist/src/policy/types.js +6 -0
  234. package/dist/src/policy/types.js.map +1 -1
  235. package/dist/src/prompts/prompt-registry.js +2 -1
  236. package/dist/src/prompts/prompt-registry.js.map +1 -1
  237. package/dist/src/routing/strategies/classifierStrategy.js +3 -2
  238. package/dist/src/routing/strategies/classifierStrategy.js.map +1 -1
  239. package/dist/src/services/chatCompressionService.d.ts +32 -0
  240. package/dist/src/services/chatCompressionService.js +162 -0
  241. package/dist/src/services/chatCompressionService.js.map +1 -0
  242. package/dist/src/services/chatCompressionService.test.d.ts +6 -0
  243. package/dist/src/services/chatCompressionService.test.js +209 -0
  244. package/dist/src/services/chatCompressionService.test.js.map +1 -0
  245. package/dist/src/services/chatRecordingService.js +9 -8
  246. package/dist/src/services/chatRecordingService.js.map +1 -1
  247. package/dist/src/services/fileDiscoveryService.d.ts +2 -14
  248. package/dist/src/services/fileDiscoveryService.js +19 -55
  249. package/dist/src/services/fileDiscoveryService.js.map +1 -1
  250. package/dist/src/services/fileDiscoveryService.test.js +91 -11
  251. package/dist/src/services/fileDiscoveryService.test.js.map +1 -1
  252. package/dist/src/services/loopDetectionService.d.ts +1 -1
  253. package/dist/src/services/loopDetectionService.js +27 -13
  254. package/dist/src/services/loopDetectionService.js.map +1 -1
  255. package/dist/src/services/loopDetectionService.test.js +119 -11
  256. package/dist/src/services/loopDetectionService.test.js.map +1 -1
  257. package/dist/src/services/shellExecutionService.js +50 -23
  258. package/dist/src/services/shellExecutionService.js.map +1 -1
  259. package/dist/src/services/shellExecutionService.test.js +82 -15
  260. package/dist/src/services/shellExecutionService.test.js.map +1 -1
  261. package/dist/src/telemetry/activity-monitor.d.ts +116 -0
  262. package/dist/src/telemetry/activity-monitor.js +209 -0
  263. package/dist/src/telemetry/activity-monitor.js.map +1 -0
  264. package/dist/src/telemetry/activity-monitor.test.d.ts +6 -0
  265. package/dist/src/telemetry/activity-monitor.test.js +248 -0
  266. package/dist/src/telemetry/activity-monitor.test.js.map +1 -0
  267. package/dist/src/telemetry/clearcut-logger/clearcut-logger.d.ts +5 -1
  268. package/dist/src/telemetry/clearcut-logger/clearcut-logger.js +135 -57
  269. package/dist/src/telemetry/clearcut-logger/clearcut-logger.js.map +1 -1
  270. package/dist/src/telemetry/clearcut-logger/clearcut-logger.test.js +43 -75
  271. package/dist/src/telemetry/clearcut-logger/clearcut-logger.test.js.map +1 -1
  272. package/dist/src/telemetry/clearcut-logger/event-metadata-key.d.ts +7 -1
  273. package/dist/src/telemetry/clearcut-logger/event-metadata-key.js +13 -1
  274. package/dist/src/telemetry/clearcut-logger/event-metadata-key.js.map +1 -1
  275. package/dist/src/telemetry/gcp-exporters.js +0 -1
  276. package/dist/src/telemetry/gcp-exporters.js.map +1 -1
  277. package/dist/src/telemetry/gcp-exporters.test.js +1 -1
  278. package/dist/src/telemetry/gcp-exporters.test.js.map +1 -1
  279. package/dist/src/telemetry/index.d.ts +4 -2
  280. package/dist/src/telemetry/index.js +5 -3
  281. package/dist/src/telemetry/index.js.map +1 -1
  282. package/dist/src/telemetry/loggers.d.ts +2 -1
  283. package/dist/src/telemetry/loggers.js +37 -26
  284. package/dist/src/telemetry/loggers.js.map +1 -1
  285. package/dist/src/telemetry/loggers.test.js +215 -56
  286. package/dist/src/telemetry/loggers.test.js.map +1 -1
  287. package/dist/src/telemetry/metrics.d.ts +55 -6
  288. package/dist/src/telemetry/metrics.js +89 -1
  289. package/dist/src/telemetry/metrics.js.map +1 -1
  290. package/dist/src/telemetry/metrics.test.js +172 -213
  291. package/dist/src/telemetry/metrics.test.js.map +1 -1
  292. package/dist/src/telemetry/sdk.js +3 -2
  293. package/dist/src/telemetry/sdk.js.map +1 -1
  294. package/dist/src/telemetry/semantic.d.ts +82 -0
  295. package/dist/src/telemetry/semantic.js +269 -0
  296. package/dist/src/telemetry/semantic.js.map +1 -0
  297. package/dist/src/telemetry/semantic.test.d.ts +6 -0
  298. package/dist/src/telemetry/semantic.test.js +387 -0
  299. package/dist/src/telemetry/semantic.test.js.map +1 -0
  300. package/dist/src/telemetry/telemetry-utils.test.js +29 -28
  301. package/dist/src/telemetry/telemetry-utils.test.js.map +1 -1
  302. package/dist/src/telemetry/trace.d.ts +46 -0
  303. package/dist/src/telemetry/trace.js +121 -0
  304. package/dist/src/telemetry/trace.js.map +1 -0
  305. package/dist/src/telemetry/types.d.ts +79 -34
  306. package/dist/src/telemetry/types.js +191 -61
  307. package/dist/src/telemetry/types.js.map +1 -1
  308. package/dist/src/telemetry/uiTelemetry.js +6 -6
  309. package/dist/src/telemetry/uiTelemetry.js.map +1 -1
  310. package/dist/src/telemetry/uiTelemetry.test.js +88 -66
  311. package/dist/src/telemetry/uiTelemetry.test.js.map +1 -1
  312. package/dist/src/tools/edit.d.ts +3 -2
  313. package/dist/src/tools/edit.js +24 -19
  314. package/dist/src/tools/edit.js.map +1 -1
  315. package/dist/src/tools/edit.test.js +78 -2
  316. package/dist/src/tools/edit.test.js.map +1 -1
  317. package/dist/src/tools/glob.d.ts +3 -2
  318. package/dist/src/tools/glob.js +15 -19
  319. package/dist/src/tools/glob.js.map +1 -1
  320. package/dist/src/tools/glob.test.js +203 -199
  321. package/dist/src/tools/glob.test.js.map +1 -1
  322. package/dist/src/tools/grep.d.ts +3 -2
  323. package/dist/src/tools/grep.js +22 -16
  324. package/dist/src/tools/grep.js.map +1 -1
  325. package/dist/src/tools/ls.d.ts +3 -2
  326. package/dist/src/tools/ls.js +15 -20
  327. package/dist/src/tools/ls.js.map +1 -1
  328. package/dist/src/tools/ls.test.js +2 -9
  329. package/dist/src/tools/ls.test.js.map +1 -1
  330. package/dist/src/tools/mcp-client-manager.d.ts +49 -11
  331. package/dist/src/tools/mcp-client-manager.js +209 -31
  332. package/dist/src/tools/mcp-client-manager.js.map +1 -1
  333. package/dist/src/tools/mcp-client-manager.test.js +132 -25
  334. package/dist/src/tools/mcp-client-manager.test.js.map +1 -1
  335. package/dist/src/tools/mcp-client.d.ts +5 -1
  336. package/dist/src/tools/mcp-client.js +85 -104
  337. package/dist/src/tools/mcp-client.js.map +1 -1
  338. package/dist/src/tools/mcp-client.test.js +65 -6
  339. package/dist/src/tools/mcp-client.test.js.map +1 -1
  340. package/dist/src/tools/mcp-tool.d.ts +5 -2
  341. package/dist/src/tools/mcp-tool.js +16 -8
  342. package/dist/src/tools/mcp-tool.js.map +1 -1
  343. package/dist/src/tools/memoryTool.d.ts +6 -4
  344. package/dist/src/tools/memoryTool.js +13 -10
  345. package/dist/src/tools/memoryTool.js.map +1 -1
  346. package/dist/src/tools/message-bus-integration.test.js +14 -1
  347. package/dist/src/tools/message-bus-integration.test.js.map +1 -1
  348. package/dist/src/tools/modifiable-tool.js +3 -2
  349. package/dist/src/tools/modifiable-tool.js.map +1 -1
  350. package/dist/src/tools/read-file.d.ts +4 -3
  351. package/dist/src/tools/read-file.js +16 -11
  352. package/dist/src/tools/read-file.js.map +1 -1
  353. package/dist/src/tools/read-file.test.js +25 -2
  354. package/dist/src/tools/read-file.test.js.map +1 -1
  355. package/dist/src/tools/read-many-files.d.ts +4 -3
  356. package/dist/src/tools/read-many-files.js +19 -37
  357. package/dist/src/tools/read-many-files.js.map +1 -1
  358. package/dist/src/tools/read-many-files.test.js +0 -1
  359. package/dist/src/tools/read-many-files.test.js.map +1 -1
  360. package/dist/src/tools/ripGrep.d.ts +3 -2
  361. package/dist/src/tools/ripGrep.js +47 -17
  362. package/dist/src/tools/ripGrep.js.map +1 -1
  363. package/dist/src/tools/ripGrep.test.js +106 -60
  364. package/dist/src/tools/ripGrep.test.js.map +1 -1
  365. package/dist/src/tools/shell.d.ts +7 -5
  366. package/dist/src/tools/shell.js +39 -68
  367. package/dist/src/tools/shell.js.map +1 -1
  368. package/dist/src/tools/shell.test.js +69 -9
  369. package/dist/src/tools/shell.test.js.map +1 -1
  370. package/dist/src/tools/smart-edit.d.ts +3 -2
  371. package/dist/src/tools/smart-edit.js +30 -18
  372. package/dist/src/tools/smart-edit.js.map +1 -1
  373. package/dist/src/tools/smart-edit.test.js +62 -2
  374. package/dist/src/tools/smart-edit.test.js.map +1 -1
  375. package/dist/src/tools/tool-names.d.ts +8 -0
  376. package/dist/src/tools/tool-names.js +8 -5
  377. package/dist/src/tools/tool-names.js.map +1 -1
  378. package/dist/src/tools/tool-registry.d.ts +6 -19
  379. package/dist/src/tools/tool-registry.js +14 -49
  380. package/dist/src/tools/tool-registry.js.map +1 -1
  381. package/dist/src/tools/tool-registry.test.js +2 -24
  382. package/dist/src/tools/tool-registry.test.js.map +1 -1
  383. package/dist/src/tools/tools.d.ts +22 -8
  384. package/dist/src/tools/tools.js +65 -36
  385. package/dist/src/tools/tools.js.map +1 -1
  386. package/dist/src/tools/web-fetch.d.ts +4 -3
  387. package/dist/src/tools/web-fetch.js +37 -25
  388. package/dist/src/tools/web-fetch.js.map +1 -1
  389. package/dist/src/tools/web-fetch.test.js +262 -1
  390. package/dist/src/tools/web-fetch.test.js.map +1 -1
  391. package/dist/src/tools/web-search.d.ts +4 -3
  392. package/dist/src/tools/web-search.js +8 -6
  393. package/dist/src/tools/web-search.js.map +1 -1
  394. package/dist/src/tools/write-file.d.ts +3 -2
  395. package/dist/src/tools/write-file.js +8 -8
  396. package/dist/src/tools/write-file.js.map +1 -1
  397. package/dist/src/tools/write-file.test.js +1 -2
  398. package/dist/src/tools/write-file.test.js.map +1 -1
  399. package/dist/src/tools/write-todos.d.ts +4 -8
  400. package/dist/src/tools/write-todos.js +15 -6
  401. package/dist/src/tools/write-todos.js.map +1 -1
  402. package/dist/src/tools/write-todos.test.js +2 -2
  403. package/dist/src/tools/write-todos.test.js.map +1 -1
  404. package/dist/src/utils/bfsFileSearch.js +3 -2
  405. package/dist/src/utils/bfsFileSearch.js.map +1 -1
  406. package/dist/src/utils/channel.d.ts +19 -0
  407. package/dist/src/utils/channel.js +49 -0
  408. package/dist/src/utils/channel.js.map +1 -0
  409. package/dist/src/utils/channel.test.d.ts +6 -0
  410. package/dist/src/utils/channel.test.js +170 -0
  411. package/dist/src/utils/channel.test.js.map +1 -0
  412. package/dist/src/utils/debugLogger.d.ts +25 -0
  413. package/dist/src/utils/debugLogger.js +33 -0
  414. package/dist/src/utils/debugLogger.js.map +1 -0
  415. package/dist/src/utils/debugLogger.test.d.ts +6 -0
  416. package/dist/src/utils/debugLogger.test.js +67 -0
  417. package/dist/src/utils/debugLogger.test.js.map +1 -0
  418. package/dist/src/utils/delay.d.ts +16 -0
  419. package/dist/src/utils/delay.js +43 -0
  420. package/dist/src/utils/delay.js.map +1 -0
  421. package/dist/src/utils/delay.test.d.ts +6 -0
  422. package/dist/src/utils/delay.test.js +88 -0
  423. package/dist/src/utils/delay.test.js.map +1 -0
  424. package/dist/src/utils/editCorrector.js +5 -9
  425. package/dist/src/utils/editCorrector.js.map +1 -1
  426. package/dist/src/utils/editCorrector.test.js +3 -5
  427. package/dist/src/utils/editCorrector.test.js.map +1 -1
  428. package/dist/src/utils/editor.js +33 -38
  429. package/dist/src/utils/editor.js.map +1 -1
  430. package/dist/src/utils/environmentContext.d.ts +2 -1
  431. package/dist/src/utils/environmentContext.js +18 -33
  432. package/dist/src/utils/environmentContext.js.map +1 -1
  433. package/dist/src/utils/environmentContext.test.js +0 -34
  434. package/dist/src/utils/environmentContext.test.js.map +1 -1
  435. package/dist/src/utils/errorParsing.d.ts +1 -1
  436. package/dist/src/utils/errorParsing.js +5 -33
  437. package/dist/src/utils/errorParsing.js.map +1 -1
  438. package/dist/src/utils/errorParsing.test.js +0 -88
  439. package/dist/src/utils/errorParsing.test.js.map +1 -1
  440. package/dist/src/utils/errors.d.ts +3 -0
  441. package/dist/src/utils/errors.js +6 -0
  442. package/dist/src/utils/errors.js.map +1 -1
  443. package/dist/src/utils/events.d.ts +88 -0
  444. package/dist/src/utils/events.js +77 -0
  445. package/dist/src/utils/events.js.map +1 -0
  446. package/dist/src/utils/events.test.d.ts +6 -0
  447. package/dist/src/utils/events.test.js +131 -0
  448. package/dist/src/utils/events.test.js.map +1 -0
  449. package/dist/src/utils/extensionLoader.d.ts +78 -0
  450. package/dist/src/utils/extensionLoader.js +162 -0
  451. package/dist/src/utils/extensionLoader.js.map +1 -0
  452. package/dist/src/utils/extensionLoader.test.d.ts +6 -0
  453. package/dist/src/utils/extensionLoader.test.js +90 -0
  454. package/dist/src/utils/extensionLoader.test.js.map +1 -0
  455. package/dist/src/utils/fetch.d.ts +1 -0
  456. package/dist/src/utils/fetch.js +4 -0
  457. package/dist/src/utils/fetch.js.map +1 -1
  458. package/dist/src/utils/fileUtils.d.ts +4 -0
  459. package/dist/src/utils/fileUtils.js +34 -2
  460. package/dist/src/utils/fileUtils.js.map +1 -1
  461. package/dist/src/utils/fileUtils.test.js +12 -1
  462. package/dist/src/utils/fileUtils.test.js.map +1 -1
  463. package/dist/src/utils/flashFallback.test.js +26 -45
  464. package/dist/src/utils/flashFallback.test.js.map +1 -1
  465. package/dist/src/utils/getFolderStructure.js +9 -17
  466. package/dist/src/utils/getFolderStructure.js.map +1 -1
  467. package/dist/src/utils/gitIgnoreParser.d.ts +4 -1
  468. package/dist/src/utils/gitIgnoreParser.js +28 -10
  469. package/dist/src/utils/gitIgnoreParser.js.map +1 -1
  470. package/dist/src/utils/gitIgnoreParser.test.js +58 -0
  471. package/dist/src/utils/gitIgnoreParser.test.js.map +1 -1
  472. package/dist/src/utils/googleErrors.d.ts +104 -0
  473. package/dist/src/utils/googleErrors.js +152 -0
  474. package/dist/src/utils/googleErrors.js.map +1 -0
  475. package/dist/src/utils/googleErrors.test.d.ts +6 -0
  476. package/dist/src/utils/googleErrors.test.js +301 -0
  477. package/dist/src/utils/googleErrors.test.js.map +1 -0
  478. package/dist/src/utils/googleQuotaErrors.d.ts +36 -0
  479. package/dist/src/utils/googleQuotaErrors.js +149 -0
  480. package/dist/src/utils/googleQuotaErrors.js.map +1 -0
  481. package/dist/src/utils/googleQuotaErrors.test.d.ts +6 -0
  482. package/dist/src/utils/googleQuotaErrors.test.js +311 -0
  483. package/dist/src/utils/googleQuotaErrors.test.js.map +1 -0
  484. package/dist/src/utils/ignorePatterns.test.js +26 -30
  485. package/dist/src/utils/ignorePatterns.test.js.map +1 -1
  486. package/dist/src/utils/installationManager.js +2 -1
  487. package/dist/src/utils/installationManager.js.map +1 -1
  488. package/dist/src/utils/installationManager.test.js +3 -3
  489. package/dist/src/utils/installationManager.test.js.map +1 -1
  490. package/dist/src/utils/llm-edit-fixer.d.ts +1 -1
  491. package/dist/src/utils/llm-edit-fixer.js +29 -4
  492. package/dist/src/utils/llm-edit-fixer.js.map +1 -1
  493. package/dist/src/utils/llm-edit-fixer.test.js +21 -0
  494. package/dist/src/utils/llm-edit-fixer.test.js.map +1 -1
  495. package/dist/src/utils/memoryDiscovery.d.ts +11 -1
  496. package/dist/src/utils/memoryDiscovery.js +150 -11
  497. package/dist/src/utils/memoryDiscovery.js.map +1 -1
  498. package/dist/src/utils/memoryDiscovery.test.js +157 -19
  499. package/dist/src/utils/memoryDiscovery.test.js.map +1 -1
  500. package/dist/src/utils/memoryImportProcessor.js +3 -2
  501. package/dist/src/utils/memoryImportProcessor.js.map +1 -1
  502. package/dist/src/utils/nextSpeakerChecker.js +2 -1
  503. package/dist/src/utils/nextSpeakerChecker.js.map +1 -1
  504. package/dist/src/utils/package.d.ts +12 -0
  505. package/dist/src/utils/package.js +15 -0
  506. package/dist/src/utils/package.js.map +1 -0
  507. package/dist/src/utils/paths.js +126 -26
  508. package/dist/src/utils/paths.js.map +1 -1
  509. package/dist/src/utils/paths.test.js +200 -68
  510. package/dist/src/utils/paths.test.js.map +1 -1
  511. package/dist/src/utils/quotaErrorDetection.d.ts +0 -2
  512. package/dist/src/utils/quotaErrorDetection.js +0 -46
  513. package/dist/src/utils/quotaErrorDetection.js.map +1 -1
  514. package/dist/src/utils/retry.d.ts +1 -0
  515. package/dist/src/utils/retry.js +57 -158
  516. package/dist/src/utils/retry.js.map +1 -1
  517. package/dist/src/utils/retry.test.js +48 -109
  518. package/dist/src/utils/retry.test.js.map +1 -1
  519. package/dist/src/utils/safeJsonStringify.d.ts +4 -4
  520. package/dist/src/utils/safeJsonStringify.js +31 -7
  521. package/dist/src/utils/safeJsonStringify.js.map +1 -1
  522. package/dist/src/utils/shell-utils.d.ts +14 -2
  523. package/dist/src/utils/shell-utils.js +381 -136
  524. package/dist/src/utils/shell-utils.js.map +1 -1
  525. package/dist/src/utils/shell-utils.test.js +242 -60
  526. package/dist/src/utils/shell-utils.test.js.map +1 -1
  527. package/dist/src/utils/summarizer.js +2 -1
  528. package/dist/src/utils/summarizer.js.map +1 -1
  529. package/dist/src/utils/summarizer.test.js +0 -1
  530. package/dist/src/utils/summarizer.test.js.map +1 -1
  531. package/dist/src/utils/systemEncoding.js +5 -4
  532. package/dist/src/utils/systemEncoding.js.map +1 -1
  533. package/dist/src/utils/tool-utils.d.ts +2 -2
  534. package/dist/src/utils/tool-utils.js +14 -5
  535. package/dist/src/utils/tool-utils.js.map +1 -1
  536. package/dist/src/utils/userAccountManager.js +5 -4
  537. package/dist/src/utils/userAccountManager.js.map +1 -1
  538. package/dist/src/utils/workspaceContext.js +3 -2
  539. package/dist/src/utils/workspaceContext.js.map +1 -1
  540. package/dist/src/utils/workspaceContext.test.js +2 -2
  541. package/dist/src/utils/workspaceContext.test.js.map +1 -1
  542. package/dist/tsconfig.tsbuildinfo +1 -1
  543. package/package.json +13 -5
  544. package/dist/src/core/subagent.d.ts +0 -236
  545. package/dist/src/core/subagent.js +0 -482
  546. package/dist/src/core/subagent.js.map +0 -1
  547. package/dist/src/core/subagent.test.js +0 -556
  548. package/dist/src/core/subagent.test.js.map +0 -1
  549. /package/dist/src/{core/subagent.test.d.ts → commands/extensions.test.d.ts} +0 -0
@@ -0,0 +1,479 @@
1
+ /**
2
+ * @license
3
+ * Copyright 2025 Google LLC
4
+ * SPDX-License-Identifier: Apache-2.0
5
+ */
6
+ import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
7
+ import { StreamJsonFormatter } from './stream-json-formatter.js';
8
+ import { JsonStreamEventType } from './types.js';
9
+ import { ToolCallDecision } from '../telemetry/tool-call-decision.js';
10
+ describe('StreamJsonFormatter', () => {
11
+ let formatter;
12
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
13
+ let stdoutWriteSpy;
14
+ beforeEach(() => {
15
+ formatter = new StreamJsonFormatter();
16
+ stdoutWriteSpy = vi
17
+ .spyOn(process.stdout, 'write')
18
+ .mockImplementation(() => true);
19
+ });
20
+ afterEach(() => {
21
+ stdoutWriteSpy.mockRestore();
22
+ });
23
+ describe('formatEvent', () => {
24
+ it('should format init event as JSONL', () => {
25
+ const event = {
26
+ type: JsonStreamEventType.INIT,
27
+ timestamp: '2025-10-10T12:00:00.000Z',
28
+ session_id: 'test-session-123',
29
+ model: 'gemini-2.0-flash-exp',
30
+ };
31
+ const result = formatter.formatEvent(event);
32
+ expect(result).toBe(JSON.stringify(event) + '\n');
33
+ expect(JSON.parse(result.trim())).toEqual(event);
34
+ });
35
+ it('should format user message event', () => {
36
+ const event = {
37
+ type: JsonStreamEventType.MESSAGE,
38
+ timestamp: '2025-10-10T12:00:00.000Z',
39
+ role: 'user',
40
+ content: 'What is 2+2?',
41
+ };
42
+ const result = formatter.formatEvent(event);
43
+ expect(result).toBe(JSON.stringify(event) + '\n');
44
+ expect(JSON.parse(result.trim())).toEqual(event);
45
+ });
46
+ it('should format assistant message event with delta', () => {
47
+ const event = {
48
+ type: JsonStreamEventType.MESSAGE,
49
+ timestamp: '2025-10-10T12:00:00.000Z',
50
+ role: 'assistant',
51
+ content: '4',
52
+ delta: true,
53
+ };
54
+ const result = formatter.formatEvent(event);
55
+ expect(result).toBe(JSON.stringify(event) + '\n');
56
+ const parsed = JSON.parse(result.trim());
57
+ expect(parsed.delta).toBe(true);
58
+ });
59
+ it('should format tool_use event', () => {
60
+ const event = {
61
+ type: JsonStreamEventType.TOOL_USE,
62
+ timestamp: '2025-10-10T12:00:00.000Z',
63
+ tool_name: 'Read',
64
+ tool_id: 'read-123',
65
+ parameters: { file_path: '/path/to/file.txt' },
66
+ };
67
+ const result = formatter.formatEvent(event);
68
+ expect(result).toBe(JSON.stringify(event) + '\n');
69
+ expect(JSON.parse(result.trim())).toEqual(event);
70
+ });
71
+ it('should format tool_result event (success)', () => {
72
+ const event = {
73
+ type: JsonStreamEventType.TOOL_RESULT,
74
+ timestamp: '2025-10-10T12:00:00.000Z',
75
+ tool_id: 'read-123',
76
+ status: 'success',
77
+ output: 'File contents here',
78
+ };
79
+ const result = formatter.formatEvent(event);
80
+ expect(result).toBe(JSON.stringify(event) + '\n');
81
+ expect(JSON.parse(result.trim())).toEqual(event);
82
+ });
83
+ it('should format tool_result event (error)', () => {
84
+ const event = {
85
+ type: JsonStreamEventType.TOOL_RESULT,
86
+ timestamp: '2025-10-10T12:00:00.000Z',
87
+ tool_id: 'read-123',
88
+ status: 'error',
89
+ error: {
90
+ type: 'FILE_NOT_FOUND',
91
+ message: 'File not found',
92
+ },
93
+ };
94
+ const result = formatter.formatEvent(event);
95
+ expect(result).toBe(JSON.stringify(event) + '\n');
96
+ expect(JSON.parse(result.trim())).toEqual(event);
97
+ });
98
+ it('should format error event', () => {
99
+ const event = {
100
+ type: JsonStreamEventType.ERROR,
101
+ timestamp: '2025-10-10T12:00:00.000Z',
102
+ severity: 'warning',
103
+ message: 'Loop detected, stopping execution',
104
+ };
105
+ const result = formatter.formatEvent(event);
106
+ expect(result).toBe(JSON.stringify(event) + '\n');
107
+ expect(JSON.parse(result.trim())).toEqual(event);
108
+ });
109
+ it('should format result event with success status', () => {
110
+ const event = {
111
+ type: JsonStreamEventType.RESULT,
112
+ timestamp: '2025-10-10T12:00:00.000Z',
113
+ status: 'success',
114
+ stats: {
115
+ total_tokens: 100,
116
+ input_tokens: 50,
117
+ output_tokens: 50,
118
+ duration_ms: 1200,
119
+ tool_calls: 2,
120
+ },
121
+ };
122
+ const result = formatter.formatEvent(event);
123
+ expect(result).toBe(JSON.stringify(event) + '\n');
124
+ expect(JSON.parse(result.trim())).toEqual(event);
125
+ });
126
+ it('should format result event with error status', () => {
127
+ const event = {
128
+ type: JsonStreamEventType.RESULT,
129
+ timestamp: '2025-10-10T12:00:00.000Z',
130
+ status: 'error',
131
+ error: {
132
+ type: 'MaxSessionTurnsError',
133
+ message: 'Maximum session turns exceeded',
134
+ },
135
+ stats: {
136
+ total_tokens: 100,
137
+ input_tokens: 50,
138
+ output_tokens: 50,
139
+ duration_ms: 1200,
140
+ tool_calls: 0,
141
+ },
142
+ };
143
+ const result = formatter.formatEvent(event);
144
+ expect(result).toBe(JSON.stringify(event) + '\n');
145
+ expect(JSON.parse(result.trim())).toEqual(event);
146
+ });
147
+ it('should produce minified JSON without pretty-printing', () => {
148
+ const event = {
149
+ type: JsonStreamEventType.MESSAGE,
150
+ timestamp: '2025-10-10T12:00:00.000Z',
151
+ role: 'user',
152
+ content: 'Test',
153
+ };
154
+ const result = formatter.formatEvent(event);
155
+ // Should not contain multiple spaces or newlines (except trailing)
156
+ expect(result).not.toContain(' ');
157
+ expect(result.split('\n').length).toBe(2); // JSON + trailing newline
158
+ });
159
+ });
160
+ describe('emitEvent', () => {
161
+ it('should write formatted event to stdout', () => {
162
+ const event = {
163
+ type: JsonStreamEventType.INIT,
164
+ timestamp: '2025-10-10T12:00:00.000Z',
165
+ session_id: 'test-session',
166
+ model: 'gemini-2.0-flash-exp',
167
+ };
168
+ formatter.emitEvent(event);
169
+ expect(stdoutWriteSpy).toHaveBeenCalledTimes(1);
170
+ expect(stdoutWriteSpy).toHaveBeenCalledWith(JSON.stringify(event) + '\n');
171
+ });
172
+ it('should emit multiple events sequentially', () => {
173
+ const event1 = {
174
+ type: JsonStreamEventType.INIT,
175
+ timestamp: '2025-10-10T12:00:00.000Z',
176
+ session_id: 'test-session',
177
+ model: 'gemini-2.0-flash-exp',
178
+ };
179
+ const event2 = {
180
+ type: JsonStreamEventType.MESSAGE,
181
+ timestamp: '2025-10-10T12:00:01.000Z',
182
+ role: 'user',
183
+ content: 'Hello',
184
+ };
185
+ formatter.emitEvent(event1);
186
+ formatter.emitEvent(event2);
187
+ expect(stdoutWriteSpy).toHaveBeenCalledTimes(2);
188
+ expect(stdoutWriteSpy).toHaveBeenNthCalledWith(1, JSON.stringify(event1) + '\n');
189
+ expect(stdoutWriteSpy).toHaveBeenNthCalledWith(2, JSON.stringify(event2) + '\n');
190
+ });
191
+ });
192
+ describe('convertToStreamStats', () => {
193
+ it('should aggregate token counts from single model', () => {
194
+ const metrics = {
195
+ models: {
196
+ 'gemini-2.0-flash': {
197
+ api: {
198
+ totalRequests: 1,
199
+ totalErrors: 0,
200
+ totalLatencyMs: 1000,
201
+ },
202
+ tokens: {
203
+ prompt: 50,
204
+ candidates: 30,
205
+ total: 80,
206
+ cached: 0,
207
+ thoughts: 0,
208
+ tool: 0,
209
+ },
210
+ },
211
+ },
212
+ tools: {
213
+ totalCalls: 2,
214
+ totalSuccess: 2,
215
+ totalFail: 0,
216
+ totalDurationMs: 500,
217
+ totalDecisions: {
218
+ [ToolCallDecision.ACCEPT]: 0,
219
+ [ToolCallDecision.REJECT]: 0,
220
+ [ToolCallDecision.MODIFY]: 0,
221
+ [ToolCallDecision.AUTO_ACCEPT]: 2,
222
+ },
223
+ byName: {},
224
+ },
225
+ files: {
226
+ totalLinesAdded: 0,
227
+ totalLinesRemoved: 0,
228
+ },
229
+ };
230
+ const result = formatter.convertToStreamStats(metrics, 1200);
231
+ expect(result).toEqual({
232
+ total_tokens: 80,
233
+ input_tokens: 50,
234
+ output_tokens: 30,
235
+ duration_ms: 1200,
236
+ tool_calls: 2,
237
+ });
238
+ });
239
+ it('should aggregate token counts from multiple models', () => {
240
+ const metrics = {
241
+ models: {
242
+ 'gemini-2.0-flash': {
243
+ api: {
244
+ totalRequests: 1,
245
+ totalErrors: 0,
246
+ totalLatencyMs: 1000,
247
+ },
248
+ tokens: {
249
+ prompt: 50,
250
+ candidates: 30,
251
+ total: 80,
252
+ cached: 0,
253
+ thoughts: 0,
254
+ tool: 0,
255
+ },
256
+ },
257
+ 'gemini-1.5-pro': {
258
+ api: {
259
+ totalRequests: 1,
260
+ totalErrors: 0,
261
+ totalLatencyMs: 2000,
262
+ },
263
+ tokens: {
264
+ prompt: 100,
265
+ candidates: 70,
266
+ total: 170,
267
+ cached: 0,
268
+ thoughts: 0,
269
+ tool: 0,
270
+ },
271
+ },
272
+ },
273
+ tools: {
274
+ totalCalls: 5,
275
+ totalSuccess: 5,
276
+ totalFail: 0,
277
+ totalDurationMs: 1000,
278
+ totalDecisions: {
279
+ [ToolCallDecision.ACCEPT]: 0,
280
+ [ToolCallDecision.REJECT]: 0,
281
+ [ToolCallDecision.MODIFY]: 0,
282
+ [ToolCallDecision.AUTO_ACCEPT]: 5,
283
+ },
284
+ byName: {},
285
+ },
286
+ files: {
287
+ totalLinesAdded: 0,
288
+ totalLinesRemoved: 0,
289
+ },
290
+ };
291
+ const result = formatter.convertToStreamStats(metrics, 3000);
292
+ expect(result).toEqual({
293
+ total_tokens: 250, // 80 + 170
294
+ input_tokens: 150, // 50 + 100
295
+ output_tokens: 100, // 30 + 70
296
+ duration_ms: 3000,
297
+ tool_calls: 5,
298
+ });
299
+ });
300
+ it('should handle empty metrics', () => {
301
+ const metrics = {
302
+ models: {},
303
+ tools: {
304
+ totalCalls: 0,
305
+ totalSuccess: 0,
306
+ totalFail: 0,
307
+ totalDurationMs: 0,
308
+ totalDecisions: {
309
+ [ToolCallDecision.ACCEPT]: 0,
310
+ [ToolCallDecision.REJECT]: 0,
311
+ [ToolCallDecision.MODIFY]: 0,
312
+ [ToolCallDecision.AUTO_ACCEPT]: 0,
313
+ },
314
+ byName: {},
315
+ },
316
+ files: {
317
+ totalLinesAdded: 0,
318
+ totalLinesRemoved: 0,
319
+ },
320
+ };
321
+ const result = formatter.convertToStreamStats(metrics, 100);
322
+ expect(result).toEqual({
323
+ total_tokens: 0,
324
+ input_tokens: 0,
325
+ output_tokens: 0,
326
+ duration_ms: 100,
327
+ tool_calls: 0,
328
+ });
329
+ });
330
+ it('should use session-level tool calls count', () => {
331
+ const metrics = {
332
+ models: {},
333
+ tools: {
334
+ totalCalls: 3,
335
+ totalSuccess: 2,
336
+ totalFail: 1,
337
+ totalDurationMs: 500,
338
+ totalDecisions: {
339
+ [ToolCallDecision.ACCEPT]: 0,
340
+ [ToolCallDecision.REJECT]: 0,
341
+ [ToolCallDecision.MODIFY]: 0,
342
+ [ToolCallDecision.AUTO_ACCEPT]: 3,
343
+ },
344
+ byName: {
345
+ Read: {
346
+ count: 2,
347
+ success: 2,
348
+ fail: 0,
349
+ durationMs: 300,
350
+ decisions: {
351
+ [ToolCallDecision.ACCEPT]: 0,
352
+ [ToolCallDecision.REJECT]: 0,
353
+ [ToolCallDecision.MODIFY]: 0,
354
+ [ToolCallDecision.AUTO_ACCEPT]: 2,
355
+ },
356
+ },
357
+ Glob: {
358
+ count: 1,
359
+ success: 0,
360
+ fail: 1,
361
+ durationMs: 200,
362
+ decisions: {
363
+ [ToolCallDecision.ACCEPT]: 0,
364
+ [ToolCallDecision.REJECT]: 0,
365
+ [ToolCallDecision.MODIFY]: 0,
366
+ [ToolCallDecision.AUTO_ACCEPT]: 1,
367
+ },
368
+ },
369
+ },
370
+ },
371
+ files: {
372
+ totalLinesAdded: 0,
373
+ totalLinesRemoved: 0,
374
+ },
375
+ };
376
+ const result = formatter.convertToStreamStats(metrics, 1000);
377
+ expect(result.tool_calls).toBe(3);
378
+ });
379
+ it('should pass through duration unchanged', () => {
380
+ const metrics = {
381
+ models: {},
382
+ tools: {
383
+ totalCalls: 0,
384
+ totalSuccess: 0,
385
+ totalFail: 0,
386
+ totalDurationMs: 0,
387
+ totalDecisions: {
388
+ [ToolCallDecision.ACCEPT]: 0,
389
+ [ToolCallDecision.REJECT]: 0,
390
+ [ToolCallDecision.MODIFY]: 0,
391
+ [ToolCallDecision.AUTO_ACCEPT]: 0,
392
+ },
393
+ byName: {},
394
+ },
395
+ files: {
396
+ totalLinesAdded: 0,
397
+ totalLinesRemoved: 0,
398
+ },
399
+ };
400
+ const result = formatter.convertToStreamStats(metrics, 5000);
401
+ expect(result.duration_ms).toBe(5000);
402
+ });
403
+ });
404
+ describe('JSON validity', () => {
405
+ it('should produce valid JSON for all event types', () => {
406
+ const events = [
407
+ {
408
+ type: JsonStreamEventType.INIT,
409
+ timestamp: '2025-10-10T12:00:00.000Z',
410
+ session_id: 'test',
411
+ model: 'gemini-2.0-flash',
412
+ },
413
+ {
414
+ type: JsonStreamEventType.MESSAGE,
415
+ timestamp: '2025-10-10T12:00:00.000Z',
416
+ role: 'user',
417
+ content: 'Test',
418
+ },
419
+ {
420
+ type: JsonStreamEventType.TOOL_USE,
421
+ timestamp: '2025-10-10T12:00:00.000Z',
422
+ tool_name: 'Read',
423
+ tool_id: 'read-1',
424
+ parameters: {},
425
+ },
426
+ {
427
+ type: JsonStreamEventType.TOOL_RESULT,
428
+ timestamp: '2025-10-10T12:00:00.000Z',
429
+ tool_id: 'read-1',
430
+ status: 'success',
431
+ },
432
+ {
433
+ type: JsonStreamEventType.ERROR,
434
+ timestamp: '2025-10-10T12:00:00.000Z',
435
+ severity: 'error',
436
+ message: 'Test error',
437
+ },
438
+ {
439
+ type: JsonStreamEventType.RESULT,
440
+ timestamp: '2025-10-10T12:00:00.000Z',
441
+ status: 'success',
442
+ stats: {
443
+ total_tokens: 0,
444
+ input_tokens: 0,
445
+ output_tokens: 0,
446
+ duration_ms: 0,
447
+ tool_calls: 0,
448
+ },
449
+ },
450
+ ];
451
+ events.forEach((event) => {
452
+ const formatted = formatter.formatEvent(event);
453
+ expect(() => JSON.parse(formatted)).not.toThrow();
454
+ });
455
+ });
456
+ it('should preserve field types', () => {
457
+ const event = {
458
+ type: JsonStreamEventType.RESULT,
459
+ timestamp: '2025-10-10T12:00:00.000Z',
460
+ status: 'success',
461
+ stats: {
462
+ total_tokens: 100,
463
+ input_tokens: 50,
464
+ output_tokens: 50,
465
+ duration_ms: 1200,
466
+ tool_calls: 2,
467
+ },
468
+ };
469
+ const formatted = formatter.formatEvent(event);
470
+ const parsed = JSON.parse(formatted.trim());
471
+ expect(typeof parsed.stats.total_tokens).toBe('number');
472
+ expect(typeof parsed.stats.input_tokens).toBe('number');
473
+ expect(typeof parsed.stats.output_tokens).toBe('number');
474
+ expect(typeof parsed.stats.duration_ms).toBe('number');
475
+ expect(typeof parsed.stats.tool_calls).toBe('number');
476
+ });
477
+ });
478
+ });
479
+ //# sourceMappingURL=stream-json-formatter.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"stream-json-formatter.test.js","sourceRoot":"","sources":["../../../src/output/stream-json-formatter.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAUjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AAEtE,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,SAA8B,CAAC;IACnC,8DAA8D;IAC9D,IAAI,cAAmB,CAAC;IAExB,UAAU,CAAC,GAAG,EAAE;QACd,SAAS,GAAG,IAAI,mBAAmB,EAAE,CAAC;QACtC,cAAc,GAAG,EAAE;aAChB,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC;aAC9B,kBAAkB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,cAAc,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,KAAK,GAAc;gBACvB,IAAI,EAAE,mBAAmB,CAAC,IAAI;gBAC9B,SAAS,EAAE,0BAA0B;gBACrC,UAAU,EAAE,kBAAkB;gBAC9B,KAAK,EAAE,sBAAsB;aAC9B,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,KAAK,GAAiB;gBAC1B,IAAI,EAAE,mBAAmB,CAAC,OAAO;gBACjC,SAAS,EAAE,0BAA0B;gBACrC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,cAAc;aACxB,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,KAAK,GAAiB;gBAC1B,IAAI,EAAE,mBAAmB,CAAC,OAAO;gBACjC,SAAS,EAAE,0BAA0B;gBACrC,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,GAAG;gBACZ,KAAK,EAAE,IAAI;aACZ,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,KAAK,GAAiB;gBAC1B,IAAI,EAAE,mBAAmB,CAAC,QAAQ;gBAClC,SAAS,EAAE,0BAA0B;gBACrC,SAAS,EAAE,MAAM;gBACjB,OAAO,EAAE,UAAU;gBACnB,UAAU,EAAE,EAAE,SAAS,EAAE,mBAAmB,EAAE;aAC/C,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAoB;gBAC7B,IAAI,EAAE,mBAAmB,CAAC,WAAW;gBACrC,SAAS,EAAE,0BAA0B;gBACrC,OAAO,EAAE,UAAU;gBACnB,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,oBAAoB;aAC7B,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,KAAK,GAAoB;gBAC7B,IAAI,EAAE,mBAAmB,CAAC,WAAW;gBACrC,SAAS,EAAE,0BAA0B;gBACrC,OAAO,EAAE,UAAU;gBACnB,MAAM,EAAE,OAAO;gBACf,KAAK,EAAE;oBACL,IAAI,EAAE,gBAAgB;oBACtB,OAAO,EAAE,gBAAgB;iBAC1B;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,KAAK,GAAe;gBACxB,IAAI,EAAE,mBAAmB,CAAC,KAAK;gBAC/B,SAAS,EAAE,0BAA0B;gBACrC,QAAQ,EAAE,SAAS;gBACnB,OAAO,EAAE,mCAAmC;aAC7C,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,KAAK,GAAgB;gBACzB,IAAI,EAAE,mBAAmB,CAAC,MAAM;gBAChC,SAAS,EAAE,0BAA0B;gBACrC,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE;oBACL,YAAY,EAAE,GAAG;oBACjB,YAAY,EAAE,EAAE;oBAChB,aAAa,EAAE,EAAE;oBACjB,WAAW,EAAE,IAAI;oBACjB,UAAU,EAAE,CAAC;iBACd;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,KAAK,GAAgB;gBACzB,IAAI,EAAE,mBAAmB,CAAC,MAAM;gBAChC,SAAS,EAAE,0BAA0B;gBACrC,MAAM,EAAE,OAAO;gBACf,KAAK,EAAE;oBACL,IAAI,EAAE,sBAAsB;oBAC5B,OAAO,EAAE,gCAAgC;iBAC1C;gBACD,KAAK,EAAE;oBACL,YAAY,EAAE,GAAG;oBACjB,YAAY,EAAE,EAAE;oBAChB,aAAa,EAAE,EAAE;oBACjB,WAAW,EAAE,IAAI;oBACjB,UAAU,EAAE,CAAC;iBACd;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,KAAK,GAAiB;gBAC1B,IAAI,EAAE,mBAAmB,CAAC,OAAO;gBACjC,SAAS,EAAE,0BAA0B;gBACrC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,MAAM;aAChB,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAE5C,mEAAmE;YACnE,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,0BAA0B;QACvE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,KAAK,GAAc;gBACvB,IAAI,EAAE,mBAAmB,CAAC,IAAI;gBAC9B,SAAS,EAAE,0BAA0B;gBACrC,UAAU,EAAE,cAAc;gBAC1B,KAAK,EAAE,sBAAsB;aAC9B,CAAC;YAEF,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAE3B,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,MAAM,GAAc;gBACxB,IAAI,EAAE,mBAAmB,CAAC,IAAI;gBAC9B,SAAS,EAAE,0BAA0B;gBACrC,UAAU,EAAE,cAAc;gBAC1B,KAAK,EAAE,sBAAsB;aAC9B,CAAC;YAEF,MAAM,MAAM,GAAiB;gBAC3B,IAAI,EAAE,mBAAmB,CAAC,OAAO;gBACjC,SAAS,EAAE,0BAA0B;gBACrC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO;aACjB,CAAC;YAEF,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC5B,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAE5B,MAAM,CAAC,cAAc,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,cAAc,CAAC,CAAC,uBAAuB,CAC5C,CAAC,EACD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,IAAI,CAC9B,CAAC;YACF,MAAM,CAAC,cAAc,CAAC,CAAC,uBAAuB,CAC5C,CAAC,EACD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,IAAI,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,OAAO,GAAmB;gBAC9B,MAAM,EAAE;oBACN,kBAAkB,EAAE;wBAClB,GAAG,EAAE;4BACH,aAAa,EAAE,CAAC;4BAChB,WAAW,EAAE,CAAC;4BACd,cAAc,EAAE,IAAI;yBACrB;wBACD,MAAM,EAAE;4BACN,MAAM,EAAE,EAAE;4BACV,UAAU,EAAE,EAAE;4BACd,KAAK,EAAE,EAAE;4BACT,MAAM,EAAE,CAAC;4BACT,QAAQ,EAAE,CAAC;4BACX,IAAI,EAAE,CAAC;yBACR;qBACF;iBACF;gBACD,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,CAAC;oBACf,SAAS,EAAE,CAAC;oBACZ,eAAe,EAAE,GAAG;oBACpB,cAAc,EAAE;wBACd,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC;qBAClC;oBACD,MAAM,EAAE,EAAE;iBACX;gBACD,KAAK,EAAE;oBACL,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,CAAC;iBACrB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,oBAAoB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAE7D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,YAAY,EAAE,EAAE;gBAChB,YAAY,EAAE,EAAE;gBAChB,aAAa,EAAE,EAAE;gBACjB,WAAW,EAAE,IAAI;gBACjB,UAAU,EAAE,CAAC;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,OAAO,GAAmB;gBAC9B,MAAM,EAAE;oBACN,kBAAkB,EAAE;wBAClB,GAAG,EAAE;4BACH,aAAa,EAAE,CAAC;4BAChB,WAAW,EAAE,CAAC;4BACd,cAAc,EAAE,IAAI;yBACrB;wBACD,MAAM,EAAE;4BACN,MAAM,EAAE,EAAE;4BACV,UAAU,EAAE,EAAE;4BACd,KAAK,EAAE,EAAE;4BACT,MAAM,EAAE,CAAC;4BACT,QAAQ,EAAE,CAAC;4BACX,IAAI,EAAE,CAAC;yBACR;qBACF;oBACD,gBAAgB,EAAE;wBAChB,GAAG,EAAE;4BACH,aAAa,EAAE,CAAC;4BAChB,WAAW,EAAE,CAAC;4BACd,cAAc,EAAE,IAAI;yBACrB;wBACD,MAAM,EAAE;4BACN,MAAM,EAAE,GAAG;4BACX,UAAU,EAAE,EAAE;4BACd,KAAK,EAAE,GAAG;4BACV,MAAM,EAAE,CAAC;4BACT,QAAQ,EAAE,CAAC;4BACX,IAAI,EAAE,CAAC;yBACR;qBACF;iBACF;gBACD,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,CAAC;oBACf,SAAS,EAAE,CAAC;oBACZ,eAAe,EAAE,IAAI;oBACrB,cAAc,EAAE;wBACd,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC;qBAClC;oBACD,MAAM,EAAE,EAAE;iBACX;gBACD,KAAK,EAAE;oBACL,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,CAAC;iBACrB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,oBAAoB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAE7D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,YAAY,EAAE,GAAG,EAAE,WAAW;gBAC9B,YAAY,EAAE,GAAG,EAAE,WAAW;gBAC9B,aAAa,EAAE,GAAG,EAAE,UAAU;gBAC9B,WAAW,EAAE,IAAI;gBACjB,UAAU,EAAE,CAAC;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,OAAO,GAAmB;gBAC9B,MAAM,EAAE,EAAE;gBACV,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,CAAC;oBACf,SAAS,EAAE,CAAC;oBACZ,eAAe,EAAE,CAAC;oBAClB,cAAc,EAAE;wBACd,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC;qBAClC;oBACD,MAAM,EAAE,EAAE;iBACX;gBACD,KAAK,EAAE;oBACL,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,CAAC;iBACrB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,oBAAoB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,YAAY,EAAE,CAAC;gBACf,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,CAAC;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,OAAO,GAAmB;gBAC9B,MAAM,EAAE,EAAE;gBACV,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,CAAC;oBACf,SAAS,EAAE,CAAC;oBACZ,eAAe,EAAE,GAAG;oBACpB,cAAc,EAAE;wBACd,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC;qBAClC;oBACD,MAAM,EAAE;wBACN,IAAI,EAAE;4BACJ,KAAK,EAAE,CAAC;4BACR,OAAO,EAAE,CAAC;4BACV,IAAI,EAAE,CAAC;4BACP,UAAU,EAAE,GAAG;4BACf,SAAS,EAAE;gCACT,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;gCAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;gCAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;gCAC5B,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC;6BAClC;yBACF;wBACD,IAAI,EAAE;4BACJ,KAAK,EAAE,CAAC;4BACR,OAAO,EAAE,CAAC;4BACV,IAAI,EAAE,CAAC;4BACP,UAAU,EAAE,GAAG;4BACf,SAAS,EAAE;gCACT,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;gCAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;gCAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;gCAC5B,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC;6BAClC;yBACF;qBACF;iBACF;gBACD,KAAK,EAAE;oBACL,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,CAAC;iBACrB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,oBAAoB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAE7D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,OAAO,GAAmB;gBAC9B,MAAM,EAAE,EAAE;gBACV,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,CAAC;oBACf,SAAS,EAAE,CAAC;oBACZ,eAAe,EAAE,CAAC;oBAClB,cAAc,EAAE;wBACd,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5B,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC;qBAClC;oBACD,MAAM,EAAE,EAAE;iBACX;gBACD,KAAK,EAAE;oBACL,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,CAAC;iBACrB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,SAAS,CAAC,oBAAoB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAE7D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,MAAM,GAAG;gBACb;oBACE,IAAI,EAAE,mBAAmB,CAAC,IAAI;oBAC9B,SAAS,EAAE,0BAA0B;oBACrC,UAAU,EAAE,MAAM;oBAClB,KAAK,EAAE,kBAAkB;iBACb;gBACd;oBACE,IAAI,EAAE,mBAAmB,CAAC,OAAO;oBACjC,SAAS,EAAE,0BAA0B;oBACrC,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,MAAM;iBACA;gBACjB;oBACE,IAAI,EAAE,mBAAmB,CAAC,QAAQ;oBAClC,SAAS,EAAE,0BAA0B;oBACrC,SAAS,EAAE,MAAM;oBACjB,OAAO,EAAE,QAAQ;oBACjB,UAAU,EAAE,EAAE;iBACC;gBACjB;oBACE,IAAI,EAAE,mBAAmB,CAAC,WAAW;oBACrC,SAAS,EAAE,0BAA0B;oBACrC,OAAO,EAAE,QAAQ;oBACjB,MAAM,EAAE,SAAS;iBACC;gBACpB;oBACE,IAAI,EAAE,mBAAmB,CAAC,KAAK;oBAC/B,SAAS,EAAE,0BAA0B;oBACrC,QAAQ,EAAE,OAAO;oBACjB,OAAO,EAAE,YAAY;iBACR;gBACf;oBACE,IAAI,EAAE,mBAAmB,CAAC,MAAM;oBAChC,SAAS,EAAE,0BAA0B;oBACrC,MAAM,EAAE,SAAS;oBACjB,KAAK,EAAE;wBACL,YAAY,EAAE,CAAC;wBACf,YAAY,EAAE,CAAC;wBACf,aAAa,EAAE,CAAC;wBAChB,WAAW,EAAE,CAAC;wBACd,UAAU,EAAE,CAAC;qBACd;iBACa;aACjB,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACvB,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBAC/C,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,KAAK,GAAgB;gBACzB,IAAI,EAAE,mBAAmB,CAAC,MAAM;gBAChC,SAAS,EAAE,0BAA0B;gBACrC,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE;oBACL,YAAY,EAAE,GAAG;oBACjB,YAAY,EAAE,EAAE;oBAChB,aAAa,EAAE,EAAE;oBACjB,WAAW,EAAE,IAAI;oBACjB,UAAU,EAAE,CAAC;iBACd;aACF,CAAC;YAEF,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;YAE5C,MAAM,CAAC,OAAO,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACxD,MAAM,CAAC,OAAO,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACxD,MAAM,CAAC,OAAO,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzD,MAAM,CAAC,OAAO,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -6,7 +6,8 @@
6
6
  import type { SessionMetrics } from '../telemetry/uiTelemetry.js';
7
7
  export declare enum OutputFormat {
8
8
  TEXT = "text",
9
- JSON = "json"
9
+ JSON = "json",
10
+ STREAM_JSON = "stream-json"
10
11
  }
11
12
  export interface JsonError {
12
13
  type: string;
@@ -18,3 +19,64 @@ export interface JsonOutput {
18
19
  stats?: SessionMetrics;
19
20
  error?: JsonError;
20
21
  }
22
+ export declare enum JsonStreamEventType {
23
+ INIT = "init",
24
+ MESSAGE = "message",
25
+ TOOL_USE = "tool_use",
26
+ TOOL_RESULT = "tool_result",
27
+ ERROR = "error",
28
+ RESULT = "result"
29
+ }
30
+ export interface BaseJsonStreamEvent {
31
+ type: JsonStreamEventType;
32
+ timestamp: string;
33
+ }
34
+ export interface InitEvent extends BaseJsonStreamEvent {
35
+ type: JsonStreamEventType.INIT;
36
+ session_id: string;
37
+ model: string;
38
+ }
39
+ export interface MessageEvent extends BaseJsonStreamEvent {
40
+ type: JsonStreamEventType.MESSAGE;
41
+ role: 'user' | 'assistant';
42
+ content: string;
43
+ delta?: boolean;
44
+ }
45
+ export interface ToolUseEvent extends BaseJsonStreamEvent {
46
+ type: JsonStreamEventType.TOOL_USE;
47
+ tool_name: string;
48
+ tool_id: string;
49
+ parameters: Record<string, unknown>;
50
+ }
51
+ export interface ToolResultEvent extends BaseJsonStreamEvent {
52
+ type: JsonStreamEventType.TOOL_RESULT;
53
+ tool_id: string;
54
+ status: 'success' | 'error';
55
+ output?: string;
56
+ error?: {
57
+ type: string;
58
+ message: string;
59
+ };
60
+ }
61
+ export interface ErrorEvent extends BaseJsonStreamEvent {
62
+ type: JsonStreamEventType.ERROR;
63
+ severity: 'warning' | 'error';
64
+ message: string;
65
+ }
66
+ export interface StreamStats {
67
+ total_tokens: number;
68
+ input_tokens: number;
69
+ output_tokens: number;
70
+ duration_ms: number;
71
+ tool_calls: number;
72
+ }
73
+ export interface ResultEvent extends BaseJsonStreamEvent {
74
+ type: JsonStreamEventType.RESULT;
75
+ status: 'success' | 'error';
76
+ error?: {
77
+ type: string;
78
+ message: string;
79
+ };
80
+ stats?: StreamStats;
81
+ }
82
+ export type JsonStreamEvent = InitEvent | MessageEvent | ToolUseEvent | ToolResultEvent | ErrorEvent | ResultEvent;
@@ -7,5 +7,16 @@ export var OutputFormat;
7
7
  (function (OutputFormat) {
8
8
  OutputFormat["TEXT"] = "text";
9
9
  OutputFormat["JSON"] = "json";
10
+ OutputFormat["STREAM_JSON"] = "stream-json";
10
11
  })(OutputFormat || (OutputFormat = {}));
12
+ // Streaming JSON event types
13
+ export var JsonStreamEventType;
14
+ (function (JsonStreamEventType) {
15
+ JsonStreamEventType["INIT"] = "init";
16
+ JsonStreamEventType["MESSAGE"] = "message";
17
+ JsonStreamEventType["TOOL_USE"] = "tool_use";
18
+ JsonStreamEventType["TOOL_RESULT"] = "tool_result";
19
+ JsonStreamEventType["ERROR"] = "error";
20
+ JsonStreamEventType["RESULT"] = "result";
21
+ })(JsonStreamEventType || (JsonStreamEventType = {}));
11
22
  //# sourceMappingURL=types.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/output/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,CAAN,IAAY,YAGX;AAHD,WAAY,YAAY;IACtB,6BAAa,CAAA;IACb,6BAAa,CAAA;AACf,CAAC,EAHW,YAAY,KAAZ,YAAY,QAGvB"}
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/output/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,CAAN,IAAY,YAIX;AAJD,WAAY,YAAY;IACtB,6BAAa,CAAA;IACb,6BAAa,CAAA;IACb,2CAA2B,CAAA;AAC7B,CAAC,EAJW,YAAY,KAAZ,YAAY,QAIvB;AAcD,6BAA6B;AAC7B,MAAM,CAAN,IAAY,mBAOX;AAPD,WAAY,mBAAmB;IAC7B,oCAAa,CAAA;IACb,0CAAmB,CAAA;IACnB,4CAAqB,CAAA;IACrB,kDAA2B,CAAA;IAC3B,sCAAe,CAAA;IACf,wCAAiB,CAAA;AACnB,CAAC,EAPW,mBAAmB,KAAnB,mBAAmB,QAO9B"}
@@ -0,0 +1,31 @@
1
+ /**
2
+ * @license
3
+ * Copyright 2025 Google LLC
4
+ * SPDX-License-Identifier: Apache-2.0
5
+ */
6
+ import { type PolicyEngineConfig, type ApprovalMode, type PolicySettings } from './types.js';
7
+ import type { PolicyEngine } from './policy-engine.js';
8
+ import { type PolicyFileError } from './toml-loader.js';
9
+ import { type MessageBus } from '../confirmation-bus/message-bus.js';
10
+ export declare const DEFAULT_CORE_POLICIES_DIR: string;
11
+ export declare const DEFAULT_POLICY_TIER = 1;
12
+ export declare const USER_POLICY_TIER = 2;
13
+ export declare const ADMIN_POLICY_TIER = 3;
14
+ /**
15
+ * Gets the list of directories to search for policy files, in order of increasing priority
16
+ * (Default -> User -> Admin).
17
+ *
18
+ * @param defaultPoliciesDir Optional path to a directory containing default policies.
19
+ */
20
+ export declare function getPolicyDirectories(defaultPoliciesDir?: string): string[];
21
+ /**
22
+ * Determines the policy tier (1=default, 2=user, 3=admin) for a given directory.
23
+ * This is used by the TOML loader to assign priority bands.
24
+ */
25
+ export declare function getPolicyTier(dir: string, defaultPoliciesDir?: string): number;
26
+ /**
27
+ * Formats a policy file error for console logging.
28
+ */
29
+ export declare function formatPolicyError(error: PolicyFileError): string;
30
+ export declare function createPolicyEngineConfig(settings: PolicySettings, approvalMode: ApprovalMode, defaultPoliciesDir?: string): Promise<PolicyEngineConfig>;
31
+ export declare function createPolicyUpdater(policyEngine: PolicyEngine, messageBus: MessageBus): void;