@sprinterai/runtime 0.3.0 → 0.5.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (385) hide show
  1. package/LICENSE +21 -0
  2. package/dist/adapters/a2a-adapter.d.ts +37 -9
  3. package/dist/adapters/a2a-adapter.d.ts.map +1 -1
  4. package/dist/adapters/a2a-adapter.js +132 -9
  5. package/dist/adapters/a2a-adapter.js.map +1 -1
  6. package/dist/adapters/a2a-adapter.test.d.ts +2 -0
  7. package/dist/adapters/a2a-adapter.test.d.ts.map +1 -0
  8. package/dist/adapters/a2a-adapter.test.js +295 -0
  9. package/dist/adapters/a2a-adapter.test.js.map +1 -0
  10. package/dist/adapters/http-agent-adapter.d.ts +19 -9
  11. package/dist/adapters/http-agent-adapter.d.ts.map +1 -1
  12. package/dist/adapters/http-agent-adapter.js +55 -9
  13. package/dist/adapters/http-agent-adapter.js.map +1 -1
  14. package/dist/adapters/http-agent-adapter.test.d.ts +2 -0
  15. package/dist/adapters/http-agent-adapter.test.d.ts.map +1 -0
  16. package/dist/adapters/http-agent-adapter.test.js +164 -0
  17. package/dist/adapters/http-agent-adapter.test.js.map +1 -0
  18. package/dist/adapters/index.d.ts +8 -8
  19. package/dist/adapters/index.d.ts.map +1 -1
  20. package/dist/adapters/index.js +4 -4
  21. package/dist/adapters/index.js.map +1 -1
  22. package/dist/adapters/mcp-adapter.d.ts +29 -9
  23. package/dist/adapters/mcp-adapter.d.ts.map +1 -1
  24. package/dist/adapters/mcp-adapter.js +43 -8
  25. package/dist/adapters/mcp-adapter.js.map +1 -1
  26. package/dist/adapters/mcp-adapter.test.d.ts +2 -0
  27. package/dist/adapters/mcp-adapter.test.d.ts.map +1 -0
  28. package/dist/adapters/mcp-adapter.test.js +118 -0
  29. package/dist/adapters/mcp-adapter.test.js.map +1 -0
  30. package/dist/adapters/openclaw-adapter.d.ts +34 -8
  31. package/dist/adapters/openclaw-adapter.d.ts.map +1 -1
  32. package/dist/adapters/openclaw-adapter.js +38 -8
  33. package/dist/adapters/openclaw-adapter.js.map +1 -1
  34. package/dist/adapters/openclaw-adapter.test.d.ts +2 -0
  35. package/dist/adapters/openclaw-adapter.test.d.ts.map +1 -0
  36. package/dist/adapters/openclaw-adapter.test.js +77 -0
  37. package/dist/adapters/openclaw-adapter.test.js.map +1 -0
  38. package/dist/agent/agent-registry.test.js +1 -1
  39. package/dist/agent/agent-resolver.d.ts +8 -2
  40. package/dist/agent/agent-resolver.d.ts.map +1 -1
  41. package/dist/agent/agent-resolver.js +7 -1
  42. package/dist/agent/agent-resolver.js.map +1 -1
  43. package/dist/agent/agent-resolver.test.js +21 -3
  44. package/dist/agent/agent-resolver.test.js.map +1 -1
  45. package/dist/agent/delegate.test.js +1 -1
  46. package/dist/agent/execute-agent.d.ts +32 -8
  47. package/dist/agent/execute-agent.d.ts.map +1 -1
  48. package/dist/agent/execute-agent.js +40 -3
  49. package/dist/agent/execute-agent.js.map +1 -1
  50. package/dist/agent/index.d.ts +9 -9
  51. package/dist/agent/index.d.ts.map +1 -1
  52. package/dist/agent/index.js +5 -5
  53. package/dist/agent/index.js.map +1 -1
  54. package/dist/agent/prompt-builder.test.js +1 -1
  55. package/dist/approval/approval-manager.d.ts +19 -0
  56. package/dist/approval/approval-manager.d.ts.map +1 -0
  57. package/dist/approval/approval-manager.js +36 -0
  58. package/dist/approval/approval-manager.js.map +1 -0
  59. package/dist/approval/approval-manager.test.d.ts +2 -0
  60. package/dist/approval/approval-manager.test.d.ts.map +1 -0
  61. package/dist/approval/approval-manager.test.js +239 -0
  62. package/dist/approval/approval-manager.test.js.map +1 -0
  63. package/dist/approval/index.d.ts +3 -0
  64. package/dist/approval/index.d.ts.map +1 -0
  65. package/dist/approval/index.js +2 -0
  66. package/dist/approval/index.js.map +1 -0
  67. package/dist/chat/chat-handler.d.ts +14 -7
  68. package/dist/chat/chat-handler.d.ts.map +1 -1
  69. package/dist/chat/chat-handler.js +56 -11
  70. package/dist/chat/chat-handler.js.map +1 -1
  71. package/dist/chat/chat-handler.test.js +100 -11
  72. package/dist/chat/chat-handler.test.js.map +1 -1
  73. package/dist/chat/index.d.ts +3 -3
  74. package/dist/chat/index.js +2 -2
  75. package/dist/chat/message-utils.d.ts +15 -7
  76. package/dist/chat/message-utils.d.ts.map +1 -1
  77. package/dist/chat/message-utils.js +94 -22
  78. package/dist/chat/message-utils.js.map +1 -1
  79. package/dist/chat/message-utils.test.js +71 -1
  80. package/dist/chat/message-utils.test.js.map +1 -1
  81. package/dist/document/chunk-generator.d.ts +6 -0
  82. package/dist/document/chunk-generator.d.ts.map +1 -0
  83. package/dist/document/chunk-generator.js +107 -0
  84. package/dist/document/chunk-generator.js.map +1 -0
  85. package/dist/document/chunk-generator.test.d.ts +2 -0
  86. package/dist/document/chunk-generator.test.d.ts.map +1 -0
  87. package/dist/document/chunk-generator.test.js +166 -0
  88. package/dist/document/chunk-generator.test.js.map +1 -0
  89. package/dist/document/document-processor.d.ts +27 -0
  90. package/dist/document/document-processor.d.ts.map +1 -0
  91. package/dist/document/document-processor.js +44 -0
  92. package/dist/document/document-processor.js.map +1 -0
  93. package/dist/document/document-processor.test.d.ts +2 -0
  94. package/dist/document/document-processor.test.d.ts.map +1 -0
  95. package/dist/document/document-processor.test.js +197 -0
  96. package/dist/document/document-processor.test.js.map +1 -0
  97. package/dist/document/index.d.ts +5 -0
  98. package/dist/document/index.d.ts.map +1 -0
  99. package/dist/document/index.js +4 -0
  100. package/dist/document/index.js.map +1 -0
  101. package/dist/document/parsers/index.d.ts +2 -0
  102. package/dist/document/parsers/index.d.ts.map +1 -0
  103. package/dist/document/parsers/index.js +2 -0
  104. package/dist/document/parsers/index.js.map +1 -0
  105. package/dist/document/parsers/text-parser.d.ts +4 -0
  106. package/dist/document/parsers/text-parser.d.ts.map +1 -0
  107. package/dist/document/parsers/text-parser.js +23 -0
  108. package/dist/document/parsers/text-parser.js.map +1 -0
  109. package/dist/document/parsers/text-parser.test.d.ts +2 -0
  110. package/dist/document/parsers/text-parser.test.d.ts.map +1 -0
  111. package/dist/document/parsers/text-parser.test.js +64 -0
  112. package/dist/document/parsers/text-parser.test.js.map +1 -0
  113. package/dist/eval/eval-runner.test.js +2 -2
  114. package/dist/eval/index.d.ts +3 -3
  115. package/dist/eval/index.js +2 -2
  116. package/dist/eval/scorers.test.js +1 -1
  117. package/dist/events/event-bus.d.ts +12 -0
  118. package/dist/events/event-bus.d.ts.map +1 -0
  119. package/dist/events/event-bus.js +77 -0
  120. package/dist/events/event-bus.js.map +1 -0
  121. package/dist/events/event-bus.test.d.ts +2 -0
  122. package/dist/events/event-bus.test.d.ts.map +1 -0
  123. package/dist/events/event-bus.test.js +155 -0
  124. package/dist/events/event-bus.test.js.map +1 -0
  125. package/dist/events/index.d.ts +2 -0
  126. package/dist/events/index.d.ts.map +1 -0
  127. package/dist/events/index.js +2 -0
  128. package/dist/events/index.js.map +1 -0
  129. package/dist/guardrail/guardrail-pipeline.test.js +1 -1
  130. package/dist/guardrail/index.d.ts +2 -2
  131. package/dist/guardrail/index.js +1 -1
  132. package/dist/index.d.ts +81 -45
  133. package/dist/index.d.ts.map +1 -1
  134. package/dist/index.js +55 -27
  135. package/dist/index.js.map +1 -1
  136. package/dist/jobs/in-memory-job-queue.d.ts +20 -0
  137. package/dist/jobs/in-memory-job-queue.d.ts.map +1 -0
  138. package/dist/jobs/in-memory-job-queue.js +120 -0
  139. package/dist/jobs/in-memory-job-queue.js.map +1 -0
  140. package/dist/jobs/in-memory-job-queue.test.d.ts +2 -0
  141. package/dist/jobs/in-memory-job-queue.test.d.ts.map +1 -0
  142. package/dist/jobs/in-memory-job-queue.test.js +146 -0
  143. package/dist/jobs/in-memory-job-queue.test.js.map +1 -0
  144. package/dist/jobs/index.d.ts +4 -0
  145. package/dist/jobs/index.d.ts.map +1 -0
  146. package/dist/jobs/index.js +3 -0
  147. package/dist/jobs/index.js.map +1 -0
  148. package/dist/jobs/job-runner.d.ts +42 -0
  149. package/dist/jobs/job-runner.d.ts.map +1 -0
  150. package/dist/jobs/job-runner.js +119 -0
  151. package/dist/jobs/job-runner.js.map +1 -0
  152. package/dist/jobs/job-runner.test.d.ts +2 -0
  153. package/dist/jobs/job-runner.test.d.ts.map +1 -0
  154. package/dist/jobs/job-runner.test.js +190 -0
  155. package/dist/jobs/job-runner.test.js.map +1 -0
  156. package/dist/memory/index.d.ts +3 -3
  157. package/dist/memory/index.js +2 -2
  158. package/dist/memory/memory-prompt.d.ts +1 -1
  159. package/dist/module/create-runtime.d.ts +21 -6
  160. package/dist/module/create-runtime.d.ts.map +1 -1
  161. package/dist/module/create-runtime.js +7 -7
  162. package/dist/module/create-runtime.js.map +1 -1
  163. package/dist/module/create-runtime.test.js +8 -8
  164. package/dist/module/index.d.ts +5 -4
  165. package/dist/module/index.d.ts.map +1 -1
  166. package/dist/module/index.js +3 -2
  167. package/dist/module/index.js.map +1 -1
  168. package/dist/module/map-supabase-stores.d.ts +33 -0
  169. package/dist/module/map-supabase-stores.d.ts.map +1 -0
  170. package/dist/module/map-supabase-stores.js +28 -0
  171. package/dist/module/map-supabase-stores.js.map +1 -0
  172. package/dist/module/module-loader.d.ts +9 -1
  173. package/dist/module/module-loader.d.ts.map +1 -1
  174. package/dist/module/module-loader.js +40 -1
  175. package/dist/module/module-loader.js.map +1 -1
  176. package/dist/module/module-loader.test.js +38 -1
  177. package/dist/module/module-loader.test.js.map +1 -1
  178. package/dist/notification/digest-scheduler.d.ts +18 -0
  179. package/dist/notification/digest-scheduler.d.ts.map +1 -0
  180. package/dist/notification/digest-scheduler.js +44 -0
  181. package/dist/notification/digest-scheduler.js.map +1 -0
  182. package/dist/notification/digest-scheduler.test.d.ts +2 -0
  183. package/dist/notification/digest-scheduler.test.d.ts.map +1 -0
  184. package/dist/notification/digest-scheduler.test.js +306 -0
  185. package/dist/notification/digest-scheduler.test.js.map +1 -0
  186. package/dist/notification/index.d.ts +5 -0
  187. package/dist/notification/index.d.ts.map +1 -0
  188. package/dist/notification/index.js +3 -0
  189. package/dist/notification/index.js.map +1 -0
  190. package/dist/notification/notification-engine.d.ts +20 -0
  191. package/dist/notification/notification-engine.d.ts.map +1 -0
  192. package/dist/notification/notification-engine.js +78 -0
  193. package/dist/notification/notification-engine.js.map +1 -0
  194. package/dist/notification/notification-engine.test.d.ts +2 -0
  195. package/dist/notification/notification-engine.test.d.ts.map +1 -0
  196. package/dist/notification/notification-engine.test.js +364 -0
  197. package/dist/notification/notification-engine.test.js.map +1 -0
  198. package/dist/providers/index.d.ts +5 -0
  199. package/dist/providers/index.d.ts.map +1 -0
  200. package/dist/providers/index.js +3 -0
  201. package/dist/providers/index.js.map +1 -0
  202. package/dist/providers/model-resolver.d.ts +93 -0
  203. package/dist/providers/model-resolver.d.ts.map +1 -0
  204. package/dist/providers/model-resolver.js +152 -0
  205. package/dist/providers/model-resolver.js.map +1 -0
  206. package/dist/providers/model-resolver.test.d.ts +2 -0
  207. package/dist/providers/model-resolver.test.d.ts.map +1 -0
  208. package/dist/providers/model-resolver.test.js +199 -0
  209. package/dist/providers/model-resolver.test.js.map +1 -0
  210. package/dist/providers/noop-providers.d.ts +6 -0
  211. package/dist/providers/noop-providers.d.ts.map +1 -0
  212. package/dist/providers/noop-providers.js +15 -0
  213. package/dist/providers/noop-providers.js.map +1 -0
  214. package/dist/providers/noop-providers.test.d.ts +2 -0
  215. package/dist/providers/noop-providers.test.d.ts.map +1 -0
  216. package/dist/providers/noop-providers.test.js +63 -0
  217. package/dist/providers/noop-providers.test.js.map +1 -0
  218. package/dist/providers/provider-interfaces.d.ts +65 -0
  219. package/dist/providers/provider-interfaces.d.ts.map +1 -0
  220. package/dist/providers/provider-interfaces.js +2 -0
  221. package/dist/providers/provider-interfaces.js.map +1 -0
  222. package/dist/realtime/index.d.ts +3 -0
  223. package/dist/realtime/index.d.ts.map +1 -0
  224. package/dist/realtime/index.js +2 -0
  225. package/dist/realtime/index.js.map +1 -0
  226. package/dist/realtime/realtime-manager.d.ts +20 -0
  227. package/dist/realtime/realtime-manager.d.ts.map +1 -0
  228. package/dist/realtime/realtime-manager.js +110 -0
  229. package/dist/realtime/realtime-manager.js.map +1 -0
  230. package/dist/realtime/realtime-manager.test.d.ts +2 -0
  231. package/dist/realtime/realtime-manager.test.d.ts.map +1 -0
  232. package/dist/realtime/realtime-manager.test.js +273 -0
  233. package/dist/realtime/realtime-manager.test.js.map +1 -0
  234. package/dist/scoring/compute-score.test.js +1 -1
  235. package/dist/scoring/index.d.ts +2 -2
  236. package/dist/scoring/index.js +1 -1
  237. package/dist/search/cosine-similarity.d.ts +8 -0
  238. package/dist/search/cosine-similarity.d.ts.map +1 -0
  239. package/dist/search/cosine-similarity.js +28 -0
  240. package/dist/search/cosine-similarity.js.map +1 -0
  241. package/dist/search/cosine-similarity.test.d.ts +2 -0
  242. package/dist/search/cosine-similarity.test.d.ts.map +1 -0
  243. package/dist/search/cosine-similarity.test.js +49 -0
  244. package/dist/search/cosine-similarity.test.js.map +1 -0
  245. package/dist/search/hybrid-search.d.ts +47 -0
  246. package/dist/search/hybrid-search.d.ts.map +1 -0
  247. package/dist/search/hybrid-search.js +111 -0
  248. package/dist/search/hybrid-search.js.map +1 -0
  249. package/dist/search/hybrid-search.test.d.ts +2 -0
  250. package/dist/search/hybrid-search.test.d.ts.map +1 -0
  251. package/dist/search/hybrid-search.test.js +238 -0
  252. package/dist/search/hybrid-search.test.js.map +1 -0
  253. package/dist/search/in-memory-search.d.ts +17 -0
  254. package/dist/search/in-memory-search.d.ts.map +1 -0
  255. package/dist/search/in-memory-search.js +59 -0
  256. package/dist/search/in-memory-search.js.map +1 -0
  257. package/dist/search/in-memory-search.test.d.ts +2 -0
  258. package/dist/search/in-memory-search.test.d.ts.map +1 -0
  259. package/dist/search/in-memory-search.test.js +169 -0
  260. package/dist/search/in-memory-search.test.js.map +1 -0
  261. package/dist/search/index.d.ts +6 -0
  262. package/dist/search/index.d.ts.map +1 -0
  263. package/dist/search/index.js +4 -0
  264. package/dist/search/index.js.map +1 -0
  265. package/dist/testing/in-memory-agent-store.d.ts +5 -1
  266. package/dist/testing/in-memory-agent-store.d.ts.map +1 -1
  267. package/dist/testing/in-memory-agent-store.js +19 -0
  268. package/dist/testing/in-memory-agent-store.js.map +1 -1
  269. package/dist/testing/in-memory-agent-store.test.js +1 -1
  270. package/dist/testing/in-memory-chat-store.d.ts.map +1 -1
  271. package/dist/testing/in-memory-chat-store.js +2 -1
  272. package/dist/testing/in-memory-chat-store.js.map +1 -1
  273. package/dist/testing/in-memory-entity-store.d.ts +5 -1
  274. package/dist/testing/in-memory-entity-store.d.ts.map +1 -1
  275. package/dist/testing/in-memory-entity-store.js +69 -4
  276. package/dist/testing/in-memory-entity-store.js.map +1 -1
  277. package/dist/testing/in-memory-entity-store.test.js +1 -1
  278. package/dist/testing/in-memory-memory-store.d.ts +3 -1
  279. package/dist/testing/in-memory-memory-store.d.ts.map +1 -1
  280. package/dist/testing/in-memory-memory-store.js +20 -0
  281. package/dist/testing/in-memory-memory-store.js.map +1 -1
  282. package/dist/testing/in-memory-tool-store.d.ts +13 -2
  283. package/dist/testing/in-memory-tool-store.d.ts.map +1 -1
  284. package/dist/testing/in-memory-tool-store.js +51 -0
  285. package/dist/testing/in-memory-tool-store.js.map +1 -1
  286. package/dist/testing/index.d.ts +7 -7
  287. package/dist/testing/index.js +7 -7
  288. package/dist/tool/ai-bridge.d.ts +1 -0
  289. package/dist/tool/ai-bridge.d.ts.map +1 -1
  290. package/dist/tool/ai-bridge.js +1 -0
  291. package/dist/tool/ai-bridge.js.map +1 -1
  292. package/dist/tool/ai-bridge.test.js +1 -1
  293. package/dist/tool/catalog.d.ts +19 -0
  294. package/dist/tool/catalog.d.ts.map +1 -0
  295. package/dist/tool/catalog.js +88 -0
  296. package/dist/tool/catalog.js.map +1 -0
  297. package/dist/tool/catalog.test.d.ts +2 -0
  298. package/dist/tool/catalog.test.d.ts.map +1 -0
  299. package/dist/tool/catalog.test.js +129 -0
  300. package/dist/tool/catalog.test.js.map +1 -0
  301. package/dist/tool/entity-tools-factory.test.js +2 -2
  302. package/dist/tool/execute-registered-tool.d.ts +17 -0
  303. package/dist/tool/execute-registered-tool.d.ts.map +1 -0
  304. package/dist/tool/execute-registered-tool.js +24 -0
  305. package/dist/tool/execute-registered-tool.js.map +1 -0
  306. package/dist/tool/execute-registered-tool.test.d.ts +2 -0
  307. package/dist/tool/execute-registered-tool.test.d.ts.map +1 -0
  308. package/dist/tool/execute-registered-tool.test.js +73 -0
  309. package/dist/tool/execute-registered-tool.test.js.map +1 -0
  310. package/dist/tool/index.d.ts +11 -7
  311. package/dist/tool/index.d.ts.map +1 -1
  312. package/dist/tool/index.js +7 -5
  313. package/dist/tool/index.js.map +1 -1
  314. package/dist/tool/resolve-tools.d.ts +3 -1
  315. package/dist/tool/resolve-tools.d.ts.map +1 -1
  316. package/dist/tool/resolve-tools.js +1 -1
  317. package/dist/tool/resolve-tools.js.map +1 -1
  318. package/dist/tool/resolve-tools.test.js +1 -1
  319. package/dist/tool/tool-executor.d.ts +3 -2
  320. package/dist/tool/tool-executor.d.ts.map +1 -1
  321. package/dist/tool/tool-executor.js +4 -2
  322. package/dist/tool/tool-executor.js.map +1 -1
  323. package/dist/tool/tool-executor.test.js +2 -2
  324. package/dist/tool/tool-registry.test.js +1 -1
  325. package/dist/tool/zod-to-json-schema.d.ts +7 -0
  326. package/dist/tool/zod-to-json-schema.d.ts.map +1 -0
  327. package/dist/tool/zod-to-json-schema.js +12 -0
  328. package/dist/tool/zod-to-json-schema.js.map +1 -0
  329. package/dist/tool/zod-to-json-schema.test.d.ts +2 -0
  330. package/dist/tool/zod-to-json-schema.test.d.ts.map +1 -0
  331. package/dist/tool/zod-to-json-schema.test.js +127 -0
  332. package/dist/tool/zod-to-json-schema.test.js.map +1 -0
  333. package/dist/webhook/index.d.ts +4 -0
  334. package/dist/webhook/index.d.ts.map +1 -0
  335. package/dist/webhook/index.js +3 -0
  336. package/dist/webhook/index.js.map +1 -0
  337. package/dist/webhook/webhook-delivery.d.ts +12 -0
  338. package/dist/webhook/webhook-delivery.d.ts.map +1 -0
  339. package/dist/webhook/webhook-delivery.js +102 -0
  340. package/dist/webhook/webhook-delivery.js.map +1 -0
  341. package/dist/webhook/webhook-delivery.test.d.ts +2 -0
  342. package/dist/webhook/webhook-delivery.test.d.ts.map +1 -0
  343. package/dist/webhook/webhook-delivery.test.js +284 -0
  344. package/dist/webhook/webhook-delivery.test.js.map +1 -0
  345. package/dist/webhook/webhook-signer.d.ts +14 -0
  346. package/dist/webhook/webhook-signer.d.ts.map +1 -0
  347. package/dist/webhook/webhook-signer.js +31 -0
  348. package/dist/webhook/webhook-signer.js.map +1 -0
  349. package/dist/webhook/webhook-signer.test.d.ts +2 -0
  350. package/dist/webhook/webhook-signer.test.d.ts.map +1 -0
  351. package/dist/webhook/webhook-signer.test.js +74 -0
  352. package/dist/webhook/webhook-signer.test.js.map +1 -0
  353. package/dist/workflow/compile.js +4 -1
  354. package/dist/workflow/compile.js.map +1 -1
  355. package/dist/workflow/compile.test.js +1 -1
  356. package/dist/workflow/evaluate-nodes.d.ts +24 -0
  357. package/dist/workflow/evaluate-nodes.d.ts.map +1 -0
  358. package/dist/workflow/evaluate-nodes.js +126 -0
  359. package/dist/workflow/evaluate-nodes.js.map +1 -0
  360. package/dist/workflow/evaluate-nodes.test.d.ts +2 -0
  361. package/dist/workflow/evaluate-nodes.test.d.ts.map +1 -0
  362. package/dist/workflow/evaluate-nodes.test.js +363 -0
  363. package/dist/workflow/evaluate-nodes.test.js.map +1 -0
  364. package/dist/workflow/index.d.ts +8 -3
  365. package/dist/workflow/index.d.ts.map +1 -1
  366. package/dist/workflow/index.js +4 -2
  367. package/dist/workflow/index.js.map +1 -1
  368. package/dist/workflow/node-executor.d.ts +40 -0
  369. package/dist/workflow/node-executor.d.ts.map +1 -0
  370. package/dist/workflow/node-executor.js +2 -0
  371. package/dist/workflow/node-executor.js.map +1 -0
  372. package/dist/workflow/node-executor.test.d.ts +2 -0
  373. package/dist/workflow/node-executor.test.d.ts.map +1 -0
  374. package/dist/workflow/node-executor.test.js +91 -0
  375. package/dist/workflow/node-executor.test.js.map +1 -0
  376. package/dist/workflow/status.test.js +1 -1
  377. package/dist/workflow/workflow-runner.d.ts +57 -0
  378. package/dist/workflow/workflow-runner.d.ts.map +1 -0
  379. package/dist/workflow/workflow-runner.js +263 -0
  380. package/dist/workflow/workflow-runner.js.map +1 -0
  381. package/dist/workflow/workflow-runner.test.d.ts +2 -0
  382. package/dist/workflow/workflow-runner.test.d.ts.map +1 -0
  383. package/dist/workflow/workflow-runner.test.js +657 -0
  384. package/dist/workflow/workflow-runner.test.js.map +1 -0
  385. package/package.json +19 -14
@@ -0,0 +1,239 @@
1
+ import { describe, it, expect, vi, beforeEach } from 'vitest';
2
+ import { createApprovalManager } from './approval-manager.js';
3
+ // ─── Helpers ───────────────────────────────────────────────
4
+ function makeApprovalRequest(overrides = {}) {
5
+ return {
6
+ id: 'apr-1',
7
+ tenant_id: 'tenant-1',
8
+ entity_id: 'ent-1',
9
+ field_name: 'revenue',
10
+ proposed_value: 50000,
11
+ current_value: 40000,
12
+ requested_by: 'agent-extractor',
13
+ assigned_to: null,
14
+ status: 'pending',
15
+ review_comment: null,
16
+ reviewed_by: null,
17
+ reviewed_at: null,
18
+ workflow_node_run_id: null,
19
+ extraction_result_id: null,
20
+ created_at: '2026-01-01T00:00:00Z',
21
+ ...overrides,
22
+ };
23
+ }
24
+ function createMockStore() {
25
+ return {
26
+ createApproval: vi.fn(async (input) => makeApprovalRequest({
27
+ tenant_id: input.tenant_id,
28
+ entity_id: input.entity_id,
29
+ field_name: input.field_name,
30
+ proposed_value: input.proposed_value,
31
+ current_value: input.current_value ?? null,
32
+ requested_by: input.requested_by,
33
+ assigned_to: input.assigned_to ?? null,
34
+ workflow_node_run_id: input.workflow_node_run_id ?? null,
35
+ extraction_result_id: input.extraction_result_id ?? null,
36
+ })),
37
+ getApproval: vi.fn(async () => null),
38
+ listPendingApprovals: vi.fn(async () => []),
39
+ reviewApproval: vi.fn(async (id, review) => makeApprovalRequest({
40
+ id,
41
+ status: review.approved ? 'approved' : 'rejected',
42
+ reviewed_by: review.reviewed_by,
43
+ review_comment: review.review_comment ?? null,
44
+ reviewed_at: '2026-01-02T00:00:00Z',
45
+ })),
46
+ expireApprovals: vi.fn(async () => 0),
47
+ };
48
+ }
49
+ function createMockEventBus() {
50
+ return {
51
+ emit: vi.fn(async () => undefined),
52
+ on: vi.fn(() => () => undefined),
53
+ onAny: vi.fn(() => () => undefined),
54
+ removeAllListeners: vi.fn(),
55
+ };
56
+ }
57
+ // ─── Tests ─────────────────────────────────────────────────
58
+ describe('createApprovalManager', () => {
59
+ let store;
60
+ let eventBus;
61
+ beforeEach(() => {
62
+ store = createMockStore();
63
+ eventBus = createMockEventBus();
64
+ });
65
+ describe('requestApproval', () => {
66
+ it('should create an approval request via the store', async () => {
67
+ const manager = createApprovalManager({ store, eventBus });
68
+ const input = {
69
+ tenant_id: 'tenant-1',
70
+ entity_id: 'ent-1',
71
+ field_name: 'revenue',
72
+ proposed_value: 50000,
73
+ current_value: 40000,
74
+ requested_by: 'agent-extractor',
75
+ };
76
+ const result = await manager.requestApproval(input);
77
+ expect(store.createApproval).toHaveBeenCalledWith(input);
78
+ expect(result.entity_id).toBe('ent-1');
79
+ expect(result.field_name).toBe('revenue');
80
+ expect(result.proposed_value).toBe(50000);
81
+ expect(result.status).toBe('pending');
82
+ });
83
+ it('should emit approval.requested event', async () => {
84
+ const manager = createApprovalManager({ store, eventBus });
85
+ await manager.requestApproval({
86
+ tenant_id: 'tenant-1',
87
+ entity_id: 'ent-1',
88
+ field_name: 'revenue',
89
+ proposed_value: 50000,
90
+ requested_by: 'agent-extractor',
91
+ });
92
+ expect(eventBus.emit).toHaveBeenCalledWith('approval.requested', {
93
+ entityId: 'ent-1',
94
+ tenantId: 'tenant-1',
95
+ fieldName: 'revenue',
96
+ proposedValue: 50000,
97
+ requestedBy: 'agent-extractor',
98
+ });
99
+ });
100
+ it('should work without event bus', async () => {
101
+ const manager = createApprovalManager({ store });
102
+ const result = await manager.requestApproval({
103
+ tenant_id: 'tenant-1',
104
+ entity_id: 'ent-1',
105
+ field_name: 'revenue',
106
+ proposed_value: 50000,
107
+ requested_by: 'agent-extractor',
108
+ });
109
+ expect(result.entity_id).toBe('ent-1');
110
+ expect(store.createApproval).toHaveBeenCalledOnce();
111
+ // No error means event bus was safely skipped
112
+ });
113
+ it('should pass through optional fields to the store', async () => {
114
+ const manager = createApprovalManager({ store, eventBus });
115
+ await manager.requestApproval({
116
+ tenant_id: 'tenant-1',
117
+ entity_id: 'ent-1',
118
+ field_name: 'revenue',
119
+ proposed_value: 50000,
120
+ requested_by: 'agent-1',
121
+ assigned_to: 'admin-1',
122
+ workflow_node_run_id: 'wfnr-1',
123
+ extraction_result_id: 'er-1',
124
+ });
125
+ expect(store.createApproval).toHaveBeenCalledWith(expect.objectContaining({
126
+ assigned_to: 'admin-1',
127
+ workflow_node_run_id: 'wfnr-1',
128
+ extraction_result_id: 'er-1',
129
+ }));
130
+ });
131
+ });
132
+ describe('review', () => {
133
+ it('should approve a request via the store', async () => {
134
+ const manager = createApprovalManager({ store, eventBus });
135
+ const result = await manager.review('apr-1', {
136
+ approved: true,
137
+ reviewed_by: 'admin-1',
138
+ });
139
+ expect(store.reviewApproval).toHaveBeenCalledWith('apr-1', {
140
+ approved: true,
141
+ reviewed_by: 'admin-1',
142
+ });
143
+ expect(result.status).toBe('approved');
144
+ expect(result.reviewed_by).toBe('admin-1');
145
+ });
146
+ it('should reject a request via the store', async () => {
147
+ const manager = createApprovalManager({ store, eventBus });
148
+ const result = await manager.review('apr-1', {
149
+ approved: false,
150
+ reviewed_by: 'admin-1',
151
+ review_comment: 'Value seems incorrect',
152
+ });
153
+ expect(store.reviewApproval).toHaveBeenCalledWith('apr-1', {
154
+ approved: false,
155
+ reviewed_by: 'admin-1',
156
+ review_comment: 'Value seems incorrect',
157
+ });
158
+ expect(result.status).toBe('rejected');
159
+ expect(result.review_comment).toBe('Value seems incorrect');
160
+ });
161
+ it('should emit approval.completed event with approved status', async () => {
162
+ const manager = createApprovalManager({ store, eventBus });
163
+ await manager.review('apr-1', {
164
+ approved: true,
165
+ reviewed_by: 'admin-1',
166
+ });
167
+ expect(eventBus.emit).toHaveBeenCalledWith('approval.completed', {
168
+ entityId: 'ent-1',
169
+ tenantId: 'tenant-1',
170
+ fieldName: 'revenue',
171
+ proposedValue: 50000,
172
+ approved: true,
173
+ reviewedBy: 'admin-1',
174
+ });
175
+ });
176
+ it('should emit approval.completed event with rejected status', async () => {
177
+ const manager = createApprovalManager({ store, eventBus });
178
+ await manager.review('apr-1', {
179
+ approved: false,
180
+ reviewed_by: 'admin-1',
181
+ });
182
+ expect(eventBus.emit).toHaveBeenCalledWith('approval.completed', {
183
+ entityId: 'ent-1',
184
+ tenantId: 'tenant-1',
185
+ fieldName: 'revenue',
186
+ proposedValue: 50000,
187
+ approved: false,
188
+ reviewedBy: 'admin-1',
189
+ });
190
+ });
191
+ it('should work without event bus', async () => {
192
+ const manager = createApprovalManager({ store });
193
+ const result = await manager.review('apr-1', {
194
+ approved: true,
195
+ reviewed_by: 'admin-1',
196
+ });
197
+ expect(result.status).toBe('approved');
198
+ expect(store.reviewApproval).toHaveBeenCalledOnce();
199
+ });
200
+ });
201
+ describe('listPending', () => {
202
+ it('should pass through to store with tenant ID', async () => {
203
+ const pending = [
204
+ makeApprovalRequest({ id: 'apr-1' }),
205
+ makeApprovalRequest({ id: 'apr-2', field_name: 'title' }),
206
+ ];
207
+ store.listPendingApprovals.mockResolvedValueOnce(pending);
208
+ const manager = createApprovalManager({ store, eventBus });
209
+ const results = await manager.listPending('tenant-1');
210
+ expect(store.listPendingApprovals).toHaveBeenCalledWith('tenant-1', undefined);
211
+ expect(results).toHaveLength(2);
212
+ expect(results[0].id).toBe('apr-1');
213
+ expect(results[1].id).toBe('apr-2');
214
+ });
215
+ it('should pass through entityId filter option', async () => {
216
+ store.listPendingApprovals.mockResolvedValueOnce([]);
217
+ const manager = createApprovalManager({ store, eventBus });
218
+ await manager.listPending('tenant-1', { entityId: 'ent-1' });
219
+ expect(store.listPendingApprovals).toHaveBeenCalledWith('tenant-1', {
220
+ entityId: 'ent-1',
221
+ });
222
+ });
223
+ it('should pass through assignedTo filter option', async () => {
224
+ store.listPendingApprovals.mockResolvedValueOnce([]);
225
+ const manager = createApprovalManager({ store, eventBus });
226
+ await manager.listPending('tenant-1', { assignedTo: 'admin-1' });
227
+ expect(store.listPendingApprovals).toHaveBeenCalledWith('tenant-1', {
228
+ assignedTo: 'admin-1',
229
+ });
230
+ });
231
+ it('should return empty array when no pending approvals', async () => {
232
+ store.listPendingApprovals.mockResolvedValueOnce([]);
233
+ const manager = createApprovalManager({ store });
234
+ const results = await manager.listPending('tenant-1');
235
+ expect(results).toEqual([]);
236
+ });
237
+ });
238
+ });
239
+ //# sourceMappingURL=approval-manager.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"approval-manager.test.js","sourceRoot":"","sources":["../../src/approval/approval-manager.test.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAE9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,8DAA8D;AAE9D,SAAS,mBAAmB,CAAC,YAAsC,EAAE;IACnE,OAAO;QACL,EAAE,EAAE,OAAO;QACX,SAAS,EAAE,UAAU;QACrB,SAAS,EAAE,OAAO;QAClB,UAAU,EAAE,SAAS;QACrB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,KAAK;QACpB,YAAY,EAAE,iBAAiB;QAC/B,WAAW,EAAE,IAAI;QACjB,MAAM,EAAE,SAAS;QACjB,cAAc,EAAE,IAAI;QACpB,WAAW,EAAE,IAAI;QACjB,WAAW,EAAE,IAAI;QACjB,oBAAoB,EAAE,IAAI;QAC1B,oBAAoB,EAAE,IAAI;QAC1B,UAAU,EAAE,sBAAsB;QAClC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,eAAe;IAOtB,OAAO;QACL,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,KAA0B,EAAE,EAAE,CACzD,mBAAmB,CAAC;YAClB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,IAAI;YAC1C,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,IAAI;YACtC,oBAAoB,EAAE,KAAK,CAAC,oBAAoB,IAAI,IAAI;YACxD,oBAAoB,EAAE,KAAK,CAAC,oBAAoB,IAAI,IAAI;SACzD,CAAC,CACH;QACD,WAAW,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC;QACpC,oBAAoB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC;QAC3C,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,EAAU,EAAE,MAA2B,EAAE,EAAE,CACtE,mBAAmB,CAAC;YAClB,EAAE;YACF,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU;YACjD,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,IAAI;YAC7C,WAAW,EAAE,sBAAsB;SACpC,CAAC,CACH;QACD,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;KACtC,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB;IAMzB,OAAO;QACL,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC;QAClC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAChC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QACnC,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;KAC5B,CAAC;AACJ,CAAC;AAED,8DAA8D;AAE9D,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,IAAI,KAAyC,CAAC;IAC9C,IAAI,QAA+C,CAAC;IAEpD,UAAU,CAAC,GAAG,EAAE;QACd,KAAK,GAAG,eAAe,EAAE,CAAC;QAC1B,QAAQ,GAAG,kBAAkB,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3D,MAAM,KAAK,GAAwB;gBACjC,SAAS,EAAE,UAAU;gBACrB,SAAS,EAAE,OAAO;gBAClB,UAAU,EAAE,SAAS;gBACrB,cAAc,EAAE,KAAK;gBACrB,aAAa,EAAE,KAAK;gBACpB,YAAY,EAAE,iBAAiB;aAChC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAEpD,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE3D,MAAM,OAAO,CAAC,eAAe,CAAC;gBAC5B,SAAS,EAAE,UAAU;gBACrB,SAAS,EAAE,OAAO;gBAClB,UAAU,EAAE,SAAS;gBACrB,cAAc,EAAE,KAAK;gBACrB,YAAY,EAAE,iBAAiB;aAChC,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,EAAE;gBAC/D,QAAQ,EAAE,OAAO;gBACjB,QAAQ,EAAE,UAAU;gBACpB,SAAS,EAAE,SAAS;gBACpB,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,iBAAiB;aAC/B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;YAEjD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC;gBAC3C,SAAS,EAAE,UAAU;gBACrB,SAAS,EAAE,OAAO;gBAClB,UAAU,EAAE,SAAS;gBACrB,cAAc,EAAE,KAAK;gBACrB,YAAY,EAAE,iBAAiB;aAChC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,oBAAoB,EAAE,CAAC;YACpD,8CAA8C;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE3D,MAAM,OAAO,CAAC,eAAe,CAAC;gBAC5B,SAAS,EAAE,UAAU;gBACrB,SAAS,EAAE,OAAO;gBAClB,UAAU,EAAE,SAAS;gBACrB,cAAc,EAAE,KAAK;gBACrB,YAAY,EAAE,SAAS;gBACvB,WAAW,EAAE,SAAS;gBACtB,oBAAoB,EAAE,QAAQ;gBAC9B,oBAAoB,EAAE,MAAM;aAC7B,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,WAAW,EAAE,SAAS;gBACtB,oBAAoB,EAAE,QAAQ;gBAC9B,oBAAoB,EAAE,MAAM;aAC7B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE;gBAC3C,QAAQ,EAAE,IAAI;gBACd,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE;gBACzD,QAAQ,EAAE,IAAI;gBACd,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE;gBAC3C,QAAQ,EAAE,KAAK;gBACf,WAAW,EAAE,SAAS;gBACtB,cAAc,EAAE,uBAAuB;aACxC,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE;gBACzD,QAAQ,EAAE,KAAK;gBACf,WAAW,EAAE,SAAS;gBACtB,cAAc,EAAE,uBAAuB;aACxC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE3D,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE;gBAC5B,QAAQ,EAAE,IAAI;gBACd,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,EAAE;gBAC/D,QAAQ,EAAE,OAAO;gBACjB,QAAQ,EAAE,UAAU;gBACpB,SAAS,EAAE,SAAS;gBACpB,aAAa,EAAE,KAAK;gBACpB,QAAQ,EAAE,IAAI;gBACd,UAAU,EAAE,SAAS;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE3D,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE;gBAC5B,QAAQ,EAAE,KAAK;gBACf,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,EAAE;gBAC/D,QAAQ,EAAE,OAAO;gBACjB,QAAQ,EAAE,UAAU;gBACpB,SAAS,EAAE,SAAS;gBACpB,aAAa,EAAE,KAAK;gBACpB,QAAQ,EAAE,KAAK;gBACf,UAAU,EAAE,SAAS;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;YAEjD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE;gBAC3C,QAAQ,EAAE,IAAI;gBACd,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,OAAO,GAAG;gBACd,mBAAmB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC;gBACpC,mBAAmB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;aAC1D,CAAC;YACF,KAAK,CAAC,oBAAoB,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAE1D,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3D,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YAEtD,MAAM,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;YAC/E,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,KAAK,CAAC,oBAAoB,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;YAErD,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3D,MAAM,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;YAE7D,MAAM,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE;gBAClE,QAAQ,EAAE,OAAO;aAClB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,KAAK,CAAC,oBAAoB,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;YAErD,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3D,MAAM,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;YAEjE,MAAM,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE;gBAClE,UAAU,EAAE,SAAS;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,KAAK,CAAC,oBAAoB,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;YAErD,MAAM,OAAO,GAAG,qBAAqB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YAEtD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,3 @@
1
+ export { createApprovalManager } from './approval-manager.js';
2
+ export type { ApprovalManagerOptions, ApprovalManager } from './approval-manager.js';
3
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/approval/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,YAAY,EAAE,sBAAsB,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC"}
@@ -0,0 +1,2 @@
1
+ export { createApprovalManager } from './approval-manager.js';
2
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/approval/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC"}
@@ -1,9 +1,10 @@
1
- import type { AgentStore, EntityStore, ChatStore, TenantStore, MemoryStore, ToolDefinition } from '@sprinterai/core';
1
+ import type { AgentStore, EntityStore, ChatStore, TenantStore, MemoryStore, ToolDefinition, GuardrailConfig, PermissionSubject } from '@sprinterai/core';
2
2
  import type { LanguageModel } from 'ai';
3
- import type { AgentRegistry } from '../agent/agent-registry';
4
- import type { ResolvedAgent } from '../agent/agent-resolver';
5
- import { executeAgentStream } from '../agent/execute-agent';
6
- import type { PromptSection } from '../agent/prompt-builder';
3
+ import type { AgentRegistry } from '../agent/agent-registry.js';
4
+ import type { ResolvedAgent } from '../agent/agent-resolver.js';
5
+ import { executeAgentStream } from '../agent/execute-agent.js';
6
+ import type { AgentFinishEvent } from '../agent/execute-agent.js';
7
+ import type { PromptSection } from '../agent/prompt-builder.js';
7
8
  /** Context passed to handler hooks. */
8
9
  export interface ChatContext {
9
10
  agent: ResolvedAgent;
@@ -29,11 +30,15 @@ export interface ChatHandlerConfig {
29
30
  toolGroupMap?: Record<string, string[]>;
30
31
  /** Permissions for tool filtering. */
31
32
  permissions?: string[];
33
+ /** Permission subject for dynamic resolution. Takes precedence over static permissions. */
34
+ subject?: PermissionSubject;
35
+ /** Optional guardrail config — input guardrails run before the model, output guardrails on finish. */
36
+ guardrails?: GuardrailConfig;
32
37
  hooks?: {
33
38
  buildExtraSections?: (ctx: ChatContext) => Promise<PromptSection[]>;
34
39
  buildExtraTools?: (ctx: ChatContext) => Promise<ToolDefinition[]>;
35
40
  onToolResult?: (name: string, result: unknown, ctx: ChatContext) => Promise<void>;
36
- onStreamFinish?: (text: string, ctx: ChatContext) => Promise<void>;
41
+ onStreamFinish?: (event: AgentFinishEvent, ctx: ChatContext) => Promise<void>;
37
42
  };
38
43
  }
39
44
  /** Request shape for the chat handler. */
@@ -49,6 +54,8 @@ export interface ChatRequest {
49
54
  userId?: string;
50
55
  tenantId?: string;
51
56
  permissions?: string[];
57
+ /** Permission subject for this request. Takes precedence over config.subject and static permissions. */
58
+ subject?: PermissionSubject;
52
59
  }
53
60
  /** Chat handler returned by the factory. */
54
61
  export interface ChatHandler {
@@ -64,7 +71,7 @@ export interface ChatHandler {
64
71
  * ```ts
65
72
  * const handler = createChatRouteHandler(config);
66
73
  * const { stream, agent } = await handler.handle(request);
67
- * return stream.toDataStreamResponse();
74
+ * return stream.toUIMessageStreamResponse();
68
75
  * ```
69
76
  */
70
77
  export declare function createChatRouteHandler(config: ChatHandlerConfig): ChatHandler;
@@ -1 +1 @@
1
- {"version":3,"file":"chat-handler.d.ts","sourceRoot":"","sources":["../../src/chat/chat-handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,SAAS,EACT,WAAW,EACX,WAAW,EACX,cAAc,EACf,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAExC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,KAAK,EAAE,aAAa,EAAuB,MAAM,yBAAyB,CAAC;AAClF,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAM7D,uCAAuC;AACvC,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,aAAa,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,uDAAuD;AACvD,MAAM,WAAW,iBAAiB;IAChC,MAAM,EAAE;QACN,KAAK,EAAE,UAAU,CAAC;QAClB,MAAM,EAAE,WAAW,CAAC;QACpB,IAAI,EAAE,SAAS,CAAC;QAChB,MAAM,EAAE,WAAW,CAAC;QACpB,MAAM,EAAE,WAAW,CAAC;KACrB,CAAC;IACF,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,aAAa,CAAC;IACjD,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,mEAAmE;IACnE,QAAQ,CAAC,EAAE,cAAc,EAAE,CAAC;IAC5B,kCAAkC;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IACxC,sCAAsC;IACtC,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE;QACN,kBAAkB,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;QACpE,eAAe,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QAClE,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QAClF,cAAc,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACpE,CAAC;CACH;AAED,0CAA0C;AAC1C,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,KAAK,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC;KACnB,CAAC,CAAC;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;CACxB;AAED,4CAA4C;AAC5C,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC;QACpC,MAAM,EAAE,UAAU,CAAC,OAAO,kBAAkB,CAAC,CAAC;QAC9C,KAAK,EAAE,aAAa,CAAC;KACtB,CAAC,CAAC;CACJ;AAED;;;;;;;;;GASG;AACH,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,iBAAiB,GAAG,WAAW,CAkM7E"}
1
+ {"version":3,"file":"chat-handler.d.ts","sourceRoot":"","sources":["../../src/chat/chat-handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EACX,SAAS,EACT,WAAW,EACX,WAAW,EACX,cAAc,EACd,eAAe,EACf,iBAAiB,EAClB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAExC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,KAAK,EAAE,aAAa,EAAuB,MAAM,yBAAyB,CAAC;AAClF,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC/D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAO7D,uCAAuC;AACvC,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,aAAa,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,uDAAuD;AACvD,MAAM,WAAW,iBAAiB;IAChC,MAAM,EAAE;QACN,KAAK,EAAE,UAAU,CAAC;QAClB,MAAM,EAAE,WAAW,CAAC;QACpB,IAAI,EAAE,SAAS,CAAC;QAChB,MAAM,EAAE,WAAW,CAAC;QACpB,MAAM,EAAE,WAAW,CAAC;KACrB,CAAC;IACF,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,aAAa,CAAC;IACjD,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,mEAAmE;IACnE,QAAQ,CAAC,EAAE,cAAc,EAAE,CAAC;IAC5B,kCAAkC;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IACxC,sCAAsC;IACtC,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,2FAA2F;IAC3F,OAAO,CAAC,EAAE,iBAAiB,CAAC;IAC5B,sGAAsG;IACtG,UAAU,CAAC,EAAE,eAAe,CAAC;IAC7B,KAAK,CAAC,EAAE;QACN,kBAAkB,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;QACpE,eAAe,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QAClE,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QAClF,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,gBAAgB,EAAE,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KAC/E,CAAC;CACH;AAED,0CAA0C;AAC1C,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,KAAK,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC;KACnB,CAAC,CAAC;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,wGAAwG;IACxG,OAAO,CAAC,EAAE,iBAAiB,CAAC;CAC7B;AAED,4CAA4C;AAC5C,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC;QACpC,MAAM,EAAE,UAAU,CAAC,OAAO,kBAAkB,CAAC,CAAC;QAC9C,KAAK,EAAE,aAAa,CAAC;KACtB,CAAC,CAAC;CACJ;AAED;;;;;;;;;GASG;AACH,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,iBAAiB,GAAG,WAAW,CAoP7E"}
@@ -1,8 +1,9 @@
1
- import { resolveAgent } from '../agent/agent-resolver';
2
- import { executeAgentStream } from '../agent/execute-agent';
3
- import { loadMemories } from '../memory/memory-manager';
4
- import { buildMemoryPrompt } from '../memory/memory-prompt';
5
- import { toAIMessages, validateMessage } from './message-utils';
1
+ import { resolveAgent } from '../agent/agent-resolver.js';
2
+ import { executeAgentStream } from '../agent/execute-agent.js';
3
+ import { runInputGuardrails } from '../guardrail/guardrail-pipeline.js';
4
+ import { loadMemories } from '../memory/memory-manager.js';
5
+ import { buildMemoryPrompt } from '../memory/memory-prompt.js';
6
+ import { toUIMessages, validateMessage } from './message-utils.js';
6
7
  /**
7
8
  * Factory: create a chat route handler with all dependencies injected.
8
9
  *
@@ -10,7 +11,7 @@ import { toAIMessages, validateMessage } from './message-utils';
10
11
  * ```ts
11
12
  * const handler = createChatRouteHandler(config);
12
13
  * const { stream, agent } = await handler.handle(request);
13
- * return stream.toDataStreamResponse();
14
+ * return stream.toUIMessageStreamResponse();
14
15
  * ```
15
16
  */
16
17
  export function createChatRouteHandler(config) {
@@ -147,6 +148,14 @@ export function createChatRouteHandler(config) {
147
148
  const extraTools = await config.hooks.buildExtraTools(ctx);
148
149
  tools.push(...extraTools);
149
150
  }
151
+ // Run input guardrails before calling the model
152
+ if (config.guardrails?.inputGuardrails && config.guardrails.inputGuardrails.length > 0) {
153
+ const inputContent = lastMsg.content ?? '';
154
+ const inputResult = await runInputGuardrails(config.guardrails.inputGuardrails, inputContent);
155
+ if (!inputResult.passed) {
156
+ throw new Error(`Input guardrail failed: ${inputResult.failReason ?? 'blocked'}`);
157
+ }
158
+ }
150
159
  // Resolve model
151
160
  const modelId = agent.model ?? 'default';
152
161
  const model = config.resolveModel(modelId);
@@ -154,19 +163,44 @@ export function createChatRouteHandler(config) {
154
163
  const allMessages = historicalMessages.length > 0
155
164
  ? [...historicalMessages, ...request.messages]
156
165
  : request.messages;
157
- const aiMessages = toAIMessages(allMessages);
166
+ const aiMessages = toUIMessages(allMessages);
158
167
  // Execute
159
168
  const stream = executeAgentStream({
160
169
  model,
161
170
  systemSections,
162
171
  tools: tools.length > 0 ? tools : undefined,
163
172
  messages: aiMessages,
164
- onFinish: async (result) => {
165
- // Save assistant message
173
+ onFinish: async (event) => {
174
+ // Build parts array from tool calls for persistence
175
+ const parts = [];
176
+ if (event.text) {
177
+ parts.push({ type: 'text', text: event.text });
178
+ }
179
+ for (const tc of event.toolCalls) {
180
+ parts.push({
181
+ type: 'tool-invocation',
182
+ toolInvocationId: `${tc.toolName}-${Date.now()}`,
183
+ toolName: tc.toolName,
184
+ args: tc.args,
185
+ state: 'call',
186
+ });
187
+ }
188
+ for (const tr of event.toolResults) {
189
+ parts.push({
190
+ type: 'tool-invocation',
191
+ toolInvocationId: `${tr.toolName}-${Date.now()}`,
192
+ toolName: tr.toolName,
193
+ args: tr.args,
194
+ result: tr.result,
195
+ state: 'result',
196
+ });
197
+ }
198
+ // Save assistant message with parts
166
199
  if (request.chatId) {
167
200
  try {
168
201
  await config.stores.chat.saveMessage(request.chatId, {
169
- content: result.text ?? '',
202
+ content: event.text ?? '',
203
+ parts: parts.length > 0 ? parts : undefined,
170
204
  senderType: 'agent',
171
205
  senderId: agentSlug,
172
206
  });
@@ -175,8 +209,19 @@ export function createChatRouteHandler(config) {
175
209
  // Non-critical
176
210
  }
177
211
  }
212
+ // Fire onToolResult hooks for each tool result
213
+ if (config.hooks?.onToolResult) {
214
+ for (const tr of event.toolResults) {
215
+ try {
216
+ await config.hooks.onToolResult(tr.toolName, tr.result, ctx);
217
+ }
218
+ catch {
219
+ // Non-critical
220
+ }
221
+ }
222
+ }
178
223
  if (config.hooks?.onStreamFinish) {
179
- await config.hooks.onStreamFinish(result.text ?? '', ctx);
224
+ await config.hooks.onStreamFinish(event, ctx);
180
225
  }
181
226
  },
182
227
  });
@@ -1 +1 @@
1
- {"version":3,"file":"chat-handler.js","sourceRoot":"","sources":["../../src/chat/chat-handler.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAEvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAE5D,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AA2DhE;;;;;;;;;GASG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAyB;IAC9D,OAAO;QACL,KAAK,CAAC,MAAM,CAAC,OAAoB;YAC/B,oBAAoB;YACpB,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvD,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC9D,MAAM,UAAU,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,oBAAoB,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;YAC1D,CAAC;YAED,gBAAgB;YAChB,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,gBAAgB,CAAC;YAC/D,MAAM,cAAc,GAAwB;gBAC1C,aAAa,EAAE,MAAM,CAAC,aAAa;gBACnC,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK;aAChC,CAAC;YACF,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YAE5D,MAAM,GAAG,GAAgB;gBACvB,KAAK;gBACL,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;aAC3B,CAAC;YAEF,uCAAuC;YACvC,IAAI,kBAAkB,GAKjB,EAAE,CAAC;YACR,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;oBACrE,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAClC,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;4BACvC,IAAI,EAAE,CAAC,CAAC,WAAW,KAAK,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW;4BAC7D,OAAO,EAAE,CAAC,CAAC,OAAO;4BAClB,WAAW,EAAE,CAAC,CAAC,WAAW;4BAC1B,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,SAAS;yBAC5B,CAAC,CAAC,CAAC;oBACN,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,kDAAkD;gBACpD,CAAC;YACH,CAAC;YAED,6BAA6B;YAC7B,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACrC,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE;wBACnD,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;wBAC9B,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,UAAU,EAAE,MAAM;wBAClB,QAAQ,EAAE,OAAO,CAAC,MAAM;qBACzB,CAAC,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBACP,6CAA6C;gBAC/C,CAAC;YACH,CAAC;YAED,+BAA+B;YAC/B,MAAM,cAAc,GAAoB;gBACtC;oBACE,KAAK,EAAE,UAAU;oBACjB,OAAO,EAAE,KAAK,CAAC,YAAY;oBAC3B,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,sBAAsB;YACtB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;oBAC3C,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE;wBACxD,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,YAAY;qBACb,CAAC,CAAC;oBACH,MAAM,aAAa,GAAG,iBAAiB,CAAC,QAAQ,CAAC,YAAY,EAAE,QAAQ,CAAC,gBAAgB,CAAC,CAAC;oBAC1F,IAAI,aAAa,EAAE,CAAC;wBAClB,cAAc,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;oBACrC,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,4CAA4C;gBAC9C,CAAC;YACH,CAAC;YAED,+BAA+B;YAC/B,IAAI,MAAM,CAAC,KAAK,EAAE,kBAAkB,EAAE,CAAC;gBACrC,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;gBACzD,cAAc,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YAChC,CAAC;YAED,8BAA8B;YAC9B,MAAM,KAAK,GAAqB,EAAE,CAAC;YAEnC,mDAAmD;YACnD,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClD,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC;gBACjC,IAAI,WAAW,EAAE,UAAU,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;oBACnD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;oBACvC,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,UAAU,EAAE,CAAC;wBAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;wBAC/C,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC5C,CAAC;oBACD,2BAA2B;oBAC3B,IAAI,WAAW,CAAC,WAAW,EAAE,CAAC;wBAC5B,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC9D,CAAC;oBACD,MAAM,oBAAoB,GAAG,OAAO,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;oBACvE,KAAK,CAAC,IAAI,CACR,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC9B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;4BAAE,OAAO,KAAK,CAAC;wBAC5C,oCAAoC;wBACpC,IACE,oBAAoB;4BACpB,CAAC,CAAC,kBAAkB;4BACpB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,EACpD,CAAC;4BACD,OAAO,KAAK,CAAC;wBACf,CAAC;wBACD,OAAO,IAAI,CAAC;oBACd,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,+EAA+E;oBAC/E,MAAM,oBAAoB,GAAG,OAAO,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;oBACvE,KAAK,CAAC,IAAI,CACR,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC9B,IACE,oBAAoB;4BACpB,CAAC,CAAC,kBAAkB;4BACpB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,EACpD,CAAC;4BACD,OAAO,KAAK,CAAC;wBACf,CAAC;wBACD,OAAO,IAAI,CAAC;oBACd,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,gCAAgC;YAChC,IAAI,MAAM,CAAC,KAAK,EAAE,eAAe,EAAE,CAAC;gBAClC,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;gBAC3D,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;YAC5B,CAAC;YAED,gBAAgB;YAChB,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,IAAI,SAAS,CAAC;YACzC,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAE3C,mDAAmD;YACnD,MAAM,WAAW,GACf,kBAAkB,CAAC,MAAM,GAAG,CAAC;gBAC3B,CAAC,CAAC,CAAC,GAAG,kBAAkB,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC;gBAC9C,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;YAEvB,MAAM,UAAU,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;YAE7C,UAAU;YACV,MAAM,MAAM,GAAG,kBAAkB,CAAC;gBAChC,KAAK;gBACL,cAAc;gBACd,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;gBAC3C,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;oBACzB,yBAAyB;oBACzB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;wBACnB,IAAI,CAAC;4BACH,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE;gCACnD,OAAO,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;gCAC1B,UAAU,EAAE,OAAO;gCACnB,QAAQ,EAAE,SAAS;6BACpB,CAAC,CAAC;wBACL,CAAC;wBAAC,MAAM,CAAC;4BACP,eAAe;wBACjB,CAAC;oBACH,CAAC;oBACD,IAAI,MAAM,CAAC,KAAK,EAAE,cAAc,EAAE,CAAC;wBACjC,MAAM,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC;oBAC5D,CAAC;gBACH,CAAC;aACF,CAAC,CAAC;YAEH,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QAC3B,CAAC;KACF,CAAC;AACJ,CAAC"}
1
+ {"version":3,"file":"chat-handler.js","sourceRoot":"","sources":["../../src/chat/chat-handler.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAEvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAG5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AACrE,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE5D,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAiEhE;;;;;;;;;GASG;AACH,MAAM,UAAU,sBAAsB,CAAC,MAAyB;IAC9D,OAAO;QACL,KAAK,CAAC,MAAM,CAAC,OAAoB;YAC/B,oBAAoB;YACpB,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvD,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC9D,MAAM,UAAU,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,oBAAoB,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;YAC1D,CAAC;YAED,gBAAgB;YAChB,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,MAAM,CAAC,gBAAgB,CAAC;YAC/D,MAAM,cAAc,GAAwB;gBAC1C,aAAa,EAAE,MAAM,CAAC,aAAa;gBACnC,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,KAAK;aAChC,CAAC;YACF,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YAE5D,MAAM,GAAG,GAAgB;gBACvB,KAAK;gBACL,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;aAC3B,CAAC;YAEF,uCAAuC;YACvC,IAAI,kBAAkB,GAKjB,EAAE,CAAC;YACR,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;oBACrE,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAClC,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;4BACvC,IAAI,EAAE,CAAC,CAAC,WAAW,KAAK,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW;4BAC7D,OAAO,EAAE,CAAC,CAAC,OAAO;4BAClB,WAAW,EAAE,CAAC,CAAC,WAAW;4BAC1B,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,SAAS;yBAC5B,CAAC,CAAC,CAAC;oBACN,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,kDAAkD;gBACpD,CAAC;YACH,CAAC;YAED,6BAA6B;YAC7B,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACrC,IAAI,CAAC;oBACH,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE;wBACnD,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;wBAC9B,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,UAAU,EAAE,MAAM;wBAClB,QAAQ,EAAE,OAAO,CAAC,MAAM;qBACzB,CAAC,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC;oBACP,6CAA6C;gBAC/C,CAAC;YACH,CAAC;YAED,+BAA+B;YAC/B,MAAM,cAAc,GAAoB;gBACtC;oBACE,KAAK,EAAE,UAAU;oBACjB,OAAO,EAAE,KAAK,CAAC,YAAY;oBAC3B,QAAQ,EAAE,GAAG;iBACd;aACF,CAAC;YAEF,sBAAsB;YACtB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;oBAC3C,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE;wBACxD,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,YAAY;qBACb,CAAC,CAAC;oBACH,MAAM,aAAa,GAAG,iBAAiB,CAAC,QAAQ,CAAC,YAAY,EAAE,QAAQ,CAAC,gBAAgB,CAAC,CAAC;oBAC1F,IAAI,aAAa,EAAE,CAAC;wBAClB,cAAc,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;oBACrC,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,4CAA4C;gBAC9C,CAAC;YACH,CAAC;YAED,+BAA+B;YAC/B,IAAI,MAAM,CAAC,KAAK,EAAE,kBAAkB,EAAE,CAAC;gBACrC,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC;gBACzD,cAAc,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YAChC,CAAC;YAED,8BAA8B;YAC9B,MAAM,KAAK,GAAqB,EAAE,CAAC;YAEnC,mDAAmD;YACnD,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClD,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC;gBACjC,IAAI,WAAW,EAAE,UAAU,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;oBACnD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;oBACvC,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,UAAU,EAAE,CAAC;wBAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;wBAC/C,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC5C,CAAC;oBACD,2BAA2B;oBAC3B,IAAI,WAAW,CAAC,WAAW,EAAE,CAAC;wBAC5B,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC9D,CAAC;oBACD,MAAM,oBAAoB,GAAG,OAAO,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;oBACvE,KAAK,CAAC,IAAI,CACR,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC9B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;4BAAE,OAAO,KAAK,CAAC;wBAC5C,oCAAoC;wBACpC,IACE,oBAAoB;4BACpB,CAAC,CAAC,kBAAkB;4BACpB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,EACpD,CAAC;4BACD,OAAO,KAAK,CAAC;wBACf,CAAC;wBACD,OAAO,IAAI,CAAC;oBACd,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,+EAA+E;oBAC/E,MAAM,oBAAoB,GAAG,OAAO,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;oBACvE,KAAK,CAAC,IAAI,CACR,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC9B,IACE,oBAAoB;4BACpB,CAAC,CAAC,kBAAkB;4BACpB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,EACpD,CAAC;4BACD,OAAO,KAAK,CAAC;wBACf,CAAC;wBACD,OAAO,IAAI,CAAC;oBACd,CAAC,CAAC,CACH,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,gCAAgC;YAChC,IAAI,MAAM,CAAC,KAAK,EAAE,eAAe,EAAE,CAAC;gBAClC,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;gBAC3D,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;YAC5B,CAAC;YAED,gDAAgD;YAChD,IAAI,MAAM,CAAC,UAAU,EAAE,eAAe,IAAI,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvF,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;gBAC3C,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAC1C,MAAM,CAAC,UAAU,CAAC,eAAe,EACjC,YAAY,CACb,CAAC;gBACF,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;oBACxB,MAAM,IAAI,KAAK,CAAC,2BAA2B,WAAW,CAAC,UAAU,IAAI,SAAS,EAAE,CAAC,CAAC;gBACpF,CAAC;YACH,CAAC;YAED,gBAAgB;YAChB,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,IAAI,SAAS,CAAC;YACzC,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAE3C,mDAAmD;YACnD,MAAM,WAAW,GACf,kBAAkB,CAAC,MAAM,GAAG,CAAC;gBAC3B,CAAC,CAAC,CAAC,GAAG,kBAAkB,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC;gBAC9C,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;YAEvB,MAAM,UAAU,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;YAE7C,UAAU;YACV,MAAM,MAAM,GAAG,kBAAkB,CAAC;gBAChC,KAAK;gBACL,cAAc;gBACd,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;gBAC3C,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;oBACxB,oDAAoD;oBACpD,MAAM,KAAK,GAAc,EAAE,CAAC;oBAC5B,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBACf,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;oBACjD,CAAC;oBACD,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;wBACjC,KAAK,CAAC,IAAI,CAAC;4BACT,IAAI,EAAE,iBAAiB;4BACvB,gBAAgB,EAAE,GAAG,EAAE,CAAC,QAAQ,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE;4BAChD,QAAQ,EAAE,EAAE,CAAC,QAAQ;4BACrB,IAAI,EAAE,EAAE,CAAC,IAAI;4BACb,KAAK,EAAE,MAAM;yBACd,CAAC,CAAC;oBACL,CAAC;oBACD,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;wBACnC,KAAK,CAAC,IAAI,CAAC;4BACT,IAAI,EAAE,iBAAiB;4BACvB,gBAAgB,EAAE,GAAG,EAAE,CAAC,QAAQ,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE;4BAChD,QAAQ,EAAE,EAAE,CAAC,QAAQ;4BACrB,IAAI,EAAE,EAAE,CAAC,IAAI;4BACb,MAAM,EAAE,EAAE,CAAC,MAAM;4BACjB,KAAK,EAAE,QAAQ;yBAChB,CAAC,CAAC;oBACL,CAAC;oBAED,oCAAoC;oBACpC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;wBACnB,IAAI,CAAC;4BACH,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE;gCACnD,OAAO,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE;gCACzB,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;gCAC3C,UAAU,EAAE,OAAO;gCACnB,QAAQ,EAAE,SAAS;6BACpB,CAAC,CAAC;wBACL,CAAC;wBAAC,MAAM,CAAC;4BACP,eAAe;wBACjB,CAAC;oBACH,CAAC;oBAED,+CAA+C;oBAC/C,IAAI,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC;wBAC/B,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;4BACnC,IAAI,CAAC;gCACH,MAAM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;4BAC/D,CAAC;4BAAC,MAAM,CAAC;gCACP,eAAe;4BACjB,CAAC;wBACH,CAAC;oBACH,CAAC;oBAED,IAAI,MAAM,CAAC,KAAK,EAAE,cAAc,EAAE,CAAC;wBACjC,MAAM,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;oBAChD,CAAC;gBACH,CAAC;aACF,CAAC,CAAC;YAEH,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QAC3B,CAAC;KACF,CAAC;AACJ,CAAC"}
@@ -1,26 +1,32 @@
1
1
  import { describe, it, expect, vi, beforeEach } from 'vitest';
2
2
  import { z } from 'zod';
3
- import { AgentRegistry } from '../agent/agent-registry';
4
- import { InMemoryAgentStore } from '../testing/in-memory-agent-store';
5
- import { InMemoryChatStore } from '../testing/in-memory-chat-store';
6
- import { InMemoryEntityStore } from '../testing/in-memory-entity-store';
7
- import { InMemoryMemoryStore } from '../testing/in-memory-memory-store';
8
- import { InMemoryTenantStore } from '../testing/in-memory-tenant-store';
9
- import { createChatRouteHandler } from './chat-handler';
3
+ import { AgentRegistry } from '../agent/agent-registry.js';
4
+ import { InMemoryAgentStore } from '../testing/in-memory-agent-store.js';
5
+ import { InMemoryChatStore } from '../testing/in-memory-chat-store.js';
6
+ import { InMemoryEntityStore } from '../testing/in-memory-entity-store.js';
7
+ import { InMemoryMemoryStore } from '../testing/in-memory-memory-store.js';
8
+ import { InMemoryTenantStore } from '../testing/in-memory-tenant-store.js';
9
+ import { createChatRouteHandler } from './chat-handler.js';
10
10
  // Mock AI SDK to avoid actual model calls
11
11
  vi.mock('ai', () => ({
12
12
  streamText: vi.fn((opts) => {
13
- // Call onFinish if provided (simulates stream completion)
13
+ // Call onFinish if provided (simulates stream completion with full event)
14
14
  if (typeof opts.onFinish === 'function') {
15
15
  setTimeout(() => {
16
- opts.onFinish({ text: 'Hello' });
16
+ opts.onFinish({
17
+ text: 'Hello',
18
+ finishReason: 'stop',
19
+ totalUsage: { inputTokens: 10, outputTokens: 5 },
20
+ steps: [],
21
+ });
17
22
  }, 0);
18
23
  }
19
24
  return {
20
25
  textStream: (async function* () {
21
26
  yield 'Hello';
22
27
  })(),
23
- toDataStreamResponse: () => new Response('mocked'),
28
+ toUIMessageStreamResponse: () => new Response('mocked'),
29
+ toTextStreamResponse: () => new Response('mocked-text'),
24
30
  };
25
31
  }),
26
32
  generateText: vi.fn(),
@@ -303,7 +309,7 @@ describe('createChatRouteHandler', () => {
303
309
  });
304
310
  // onFinish is called asynchronously by the mock — wait for it
305
311
  await new Promise((r) => setTimeout(r, 50));
306
- expect(onStreamFinish).toHaveBeenCalledWith('Hello', expect.objectContaining({
312
+ expect(onStreamFinish).toHaveBeenCalledWith(expect.objectContaining({ text: 'Hello' }), expect.objectContaining({
307
313
  agent: expect.objectContaining({ id: 'analyst' }),
308
314
  }));
309
315
  });
@@ -352,5 +358,88 @@ describe('createChatRouteHandler', () => {
352
358
  });
353
359
  expect(agent.id).toBe('analyst');
354
360
  });
361
+ it('blocks message when input guardrail fails', async () => {
362
+ const handler = makeHandler({
363
+ guardrails: {
364
+ inputGuardrails: [
365
+ {
366
+ name: 'no-profanity',
367
+ check: async () => ({ passed: false, reason: 'Contains profanity' }),
368
+ },
369
+ ],
370
+ },
371
+ });
372
+ await expect(handler.handle({
373
+ messages: [{ role: 'user', content: 'bad words' }],
374
+ })).rejects.toThrow('Input guardrail failed: Contains profanity');
375
+ });
376
+ it('allows message when input guardrail passes', async () => {
377
+ const handler = makeHandler({
378
+ guardrails: {
379
+ inputGuardrails: [
380
+ {
381
+ name: 'always-pass',
382
+ check: async () => ({ passed: true }),
383
+ },
384
+ ],
385
+ },
386
+ });
387
+ const { agent } = await handler.handle({
388
+ messages: [{ role: 'user', content: 'hello' }],
389
+ });
390
+ expect(agent.id).toBe('analyst');
391
+ });
392
+ it('saves assistant message with parts including tool calls', async () => {
393
+ const { streamText: mockStreamText } = await import('ai');
394
+ // Override the mock to simulate tool calls in the finish event
395
+ vi.mocked(mockStreamText).mockImplementationOnce((opts) => {
396
+ if (typeof opts.onFinish === 'function') {
397
+ setTimeout(() => {
398
+ opts.onFinish({
399
+ text: 'I found 3 results.',
400
+ finishReason: 'stop',
401
+ totalUsage: { inputTokens: 50, outputTokens: 20 },
402
+ steps: [
403
+ {
404
+ toolCalls: [{ toolName: 'searchEntities', args: { query: 'test' } }],
405
+ toolResults: [
406
+ {
407
+ toolName: 'searchEntities',
408
+ args: { query: 'test' },
409
+ result: { count: 3 },
410
+ },
411
+ ],
412
+ },
413
+ ],
414
+ });
415
+ }, 0);
416
+ }
417
+ return {
418
+ textStream: (async function* () {
419
+ yield 'I found 3 results.';
420
+ })(),
421
+ toUIMessageStreamResponse: () => new Response('mocked'),
422
+ toTextStreamResponse: () => new Response('mocked-text'),
423
+ };
424
+ });
425
+ const chat = await chatStore.createChat({ chatType: 'ai' });
426
+ const handler = makeHandler();
427
+ await handler.handle({
428
+ messages: [{ role: 'user', content: 'search for test' }],
429
+ chatId: chat.id,
430
+ userId: 'user-1',
431
+ });
432
+ // Wait for async onFinish
433
+ await new Promise((r) => setTimeout(r, 50));
434
+ const msgs = await chatStore.getMessages(chat.id);
435
+ // 1 user message + 1 assistant message
436
+ const assistantMsg = msgs.find((m) => m.sender_type === 'agent');
437
+ expect(assistantMsg).toBeDefined();
438
+ expect(assistantMsg?.content).toBe('I found 3 results.');
439
+ expect(assistantMsg?.parts).toBeDefined();
440
+ expect(Array.isArray(assistantMsg?.parts)).toBe(true);
441
+ // Should have text part + tool invocation parts
442
+ expect((assistantMsg?.parts).length).toBeGreaterThanOrEqual(1);
443
+ });
355
444
  });
356
445
  //# sourceMappingURL=chat-handler.test.js.map