@strands-agents/sdk 0.4.0 → 0.6.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 (306) hide show
  1. package/dist/src/__fixtures__/agent-helpers.d.ts +10 -1
  2. package/dist/src/__fixtures__/agent-helpers.d.ts.map +1 -1
  3. package/dist/src/__fixtures__/agent-helpers.js +13 -2
  4. package/dist/src/__fixtures__/agent-helpers.js.map +1 -1
  5. package/dist/src/__fixtures__/metrics-helpers.d.ts +25 -0
  6. package/dist/src/__fixtures__/metrics-helpers.d.ts.map +1 -0
  7. package/dist/src/__fixtures__/metrics-helpers.js +34 -0
  8. package/dist/src/__fixtures__/metrics-helpers.js.map +1 -0
  9. package/dist/src/__fixtures__/mock-message-model.d.ts.map +1 -1
  10. package/dist/src/__fixtures__/mock-message-model.js +12 -0
  11. package/dist/src/__fixtures__/mock-message-model.js.map +1 -1
  12. package/dist/src/__fixtures__/mock-span.d.ts +78 -0
  13. package/dist/src/__fixtures__/mock-span.d.ts.map +1 -0
  14. package/dist/src/__fixtures__/mock-span.js +93 -0
  15. package/dist/src/__fixtures__/mock-span.js.map +1 -0
  16. package/dist/src/__fixtures__/mock-storage-provider.d.ts +5 -0
  17. package/dist/src/__fixtures__/mock-storage-provider.d.ts.map +1 -1
  18. package/dist/src/__fixtures__/mock-storage-provider.js +23 -6
  19. package/dist/src/__fixtures__/mock-storage-provider.js.map +1 -1
  20. package/dist/src/__fixtures__/slim-types.d.ts +2 -1
  21. package/dist/src/__fixtures__/slim-types.d.ts.map +1 -1
  22. package/dist/src/__fixtures__/tool-helpers.d.ts +2 -2
  23. package/dist/src/__fixtures__/tool-helpers.d.ts.map +1 -1
  24. package/dist/src/__fixtures__/tool-helpers.js +4 -4
  25. package/dist/src/__fixtures__/tool-helpers.js.map +1 -1
  26. package/dist/src/__tests__/app-state.test.d.ts +2 -0
  27. package/dist/src/__tests__/app-state.test.d.ts.map +1 -0
  28. package/dist/src/{agent/__tests__/state.test.js → __tests__/app-state.test.js} +46 -46
  29. package/dist/src/__tests__/app-state.test.js.map +1 -0
  30. package/dist/src/__tests__/mcp.test.js +129 -18
  31. package/dist/src/__tests__/mcp.test.js.map +1 -1
  32. package/dist/src/agent/__tests__/agent.test.js +119 -11
  33. package/dist/src/agent/__tests__/agent.test.js.map +1 -1
  34. package/dist/src/agent/__tests__/agent.tracer.test.d.ts +2 -0
  35. package/dist/src/agent/__tests__/agent.tracer.test.d.ts.map +1 -0
  36. package/dist/src/agent/__tests__/agent.tracer.test.js +470 -0
  37. package/dist/src/agent/__tests__/agent.tracer.test.js.map +1 -0
  38. package/dist/src/agent/agent.d.ts +79 -9
  39. package/dist/src/agent/agent.d.ts.map +1 -1
  40. package/dist/src/agent/agent.js +253 -68
  41. package/dist/src/agent/agent.js.map +1 -1
  42. package/dist/src/{agent/state.d.ts → app-state.d.ts} +7 -7
  43. package/dist/src/app-state.d.ts.map +1 -0
  44. package/dist/src/{agent/state.js → app-state.js} +6 -6
  45. package/dist/src/app-state.js.map +1 -0
  46. package/dist/src/conversation-manager/sliding-window-conversation-manager.d.ts +1 -1
  47. package/dist/src/conversation-manager/sliding-window-conversation-manager.js +1 -1
  48. package/dist/src/errors.d.ts +6 -0
  49. package/dist/src/errors.d.ts.map +1 -1
  50. package/dist/src/errors.js +9 -0
  51. package/dist/src/errors.js.map +1 -1
  52. package/dist/src/hooks/__tests__/events.test.js +2 -0
  53. package/dist/src/hooks/__tests__/events.test.js.map +1 -1
  54. package/dist/src/hooks/events.d.ts +16 -0
  55. package/dist/src/hooks/events.d.ts.map +1 -1
  56. package/dist/src/hooks/events.js.map +1 -1
  57. package/dist/src/hooks/index.d.ts +1 -1
  58. package/dist/src/hooks/index.d.ts.map +1 -1
  59. package/dist/src/index.d.ts +23 -7
  60. package/dist/src/index.d.ts.map +1 -1
  61. package/dist/src/index.js +19 -2
  62. package/dist/src/index.js.map +1 -1
  63. package/dist/src/mcp.d.ts +41 -0
  64. package/dist/src/mcp.d.ts.map +1 -1
  65. package/dist/src/mcp.js +60 -6
  66. package/dist/src/mcp.js.map +1 -1
  67. package/dist/src/models/__tests__/bedrock.test.js +688 -0
  68. package/dist/src/models/__tests__/bedrock.test.js.map +1 -1
  69. package/dist/src/models/__tests__/model.test.js +113 -0
  70. package/dist/src/models/__tests__/model.test.js.map +1 -1
  71. package/dist/src/models/__tests__/streaming.test.d.ts +2 -0
  72. package/dist/src/models/__tests__/streaming.test.d.ts.map +1 -0
  73. package/dist/src/models/__tests__/streaming.test.js +50 -0
  74. package/dist/src/models/__tests__/streaming.test.js.map +1 -0
  75. package/dist/src/models/bedrock.d.ts +80 -1
  76. package/dist/src/models/bedrock.d.ts.map +1 -1
  77. package/dist/src/models/bedrock.js +214 -5
  78. package/dist/src/models/bedrock.js.map +1 -1
  79. package/dist/src/models/model.d.ts +10 -0
  80. package/dist/src/models/model.d.ts.map +1 -1
  81. package/dist/src/models/model.js +58 -2
  82. package/dist/src/models/model.js.map +1 -1
  83. package/dist/src/models/streaming.d.ts +84 -3
  84. package/dist/src/models/streaming.d.ts.map +1 -1
  85. package/dist/src/models/streaming.js +26 -0
  86. package/dist/src/models/streaming.js.map +1 -1
  87. package/dist/src/multiagent/__tests__/events.test.d.ts +2 -0
  88. package/dist/src/multiagent/__tests__/events.test.d.ts.map +1 -0
  89. package/dist/src/multiagent/__tests__/events.test.js +205 -0
  90. package/dist/src/multiagent/__tests__/events.test.js.map +1 -0
  91. package/dist/src/multiagent/__tests__/graph.test.d.ts +2 -0
  92. package/dist/src/multiagent/__tests__/graph.test.d.ts.map +1 -0
  93. package/dist/src/multiagent/__tests__/graph.test.js +438 -0
  94. package/dist/src/multiagent/__tests__/graph.test.js.map +1 -0
  95. package/dist/src/multiagent/__tests__/nodes.test.js +105 -9
  96. package/dist/src/multiagent/__tests__/nodes.test.js.map +1 -1
  97. package/dist/src/multiagent/__tests__/queue.test.d.ts +2 -0
  98. package/dist/src/multiagent/__tests__/queue.test.d.ts.map +1 -0
  99. package/dist/src/multiagent/__tests__/queue.test.js +118 -0
  100. package/dist/src/multiagent/__tests__/queue.test.js.map +1 -0
  101. package/dist/src/multiagent/__tests__/swarm.test.d.ts +2 -0
  102. package/dist/src/multiagent/__tests__/swarm.test.d.ts.map +1 -0
  103. package/dist/src/multiagent/__tests__/swarm.test.js +241 -0
  104. package/dist/src/multiagent/__tests__/swarm.test.js.map +1 -0
  105. package/dist/src/multiagent/base.d.ts +25 -0
  106. package/dist/src/multiagent/base.d.ts.map +1 -0
  107. package/dist/src/multiagent/base.js +2 -0
  108. package/dist/src/multiagent/base.js.map +1 -0
  109. package/dist/src/multiagent/edge.d.ts +36 -0
  110. package/dist/src/multiagent/edge.d.ts.map +1 -0
  111. package/dist/src/multiagent/edge.js +15 -0
  112. package/dist/src/multiagent/edge.js.map +1 -0
  113. package/dist/src/multiagent/events.d.ts +129 -4
  114. package/dist/src/multiagent/events.d.ts.map +1 -1
  115. package/dist/src/multiagent/events.js +140 -2
  116. package/dist/src/multiagent/events.js.map +1 -1
  117. package/dist/src/multiagent/graph.d.ts +125 -0
  118. package/dist/src/multiagent/graph.d.ts.map +1 -0
  119. package/dist/src/multiagent/graph.js +394 -0
  120. package/dist/src/multiagent/graph.js.map +1 -0
  121. package/dist/src/multiagent/index.d.ts +12 -6
  122. package/dist/src/multiagent/index.d.ts.map +1 -1
  123. package/dist/src/multiagent/index.js +6 -3
  124. package/dist/src/multiagent/index.js.map +1 -1
  125. package/dist/src/multiagent/nodes.d.ts +66 -16
  126. package/dist/src/multiagent/nodes.d.ts.map +1 -1
  127. package/dist/src/multiagent/nodes.js +82 -24
  128. package/dist/src/multiagent/nodes.js.map +1 -1
  129. package/dist/src/multiagent/queue.d.ts +73 -0
  130. package/dist/src/multiagent/queue.d.ts.map +1 -0
  131. package/dist/src/multiagent/queue.js +72 -0
  132. package/dist/src/multiagent/queue.js.map +1 -0
  133. package/dist/src/multiagent/state.d.ts +79 -10
  134. package/dist/src/multiagent/state.d.ts.map +1 -1
  135. package/dist/src/multiagent/state.js +101 -9
  136. package/dist/src/multiagent/state.js.map +1 -1
  137. package/dist/src/multiagent/swarm.d.ts +102 -0
  138. package/dist/src/multiagent/swarm.d.ts.map +1 -0
  139. package/dist/src/multiagent/swarm.js +238 -0
  140. package/dist/src/multiagent/swarm.js.map +1 -0
  141. package/dist/src/registry/__tests__/tool-registry.test.d.ts +2 -0
  142. package/dist/src/registry/__tests__/tool-registry.test.d.ts.map +1 -0
  143. package/dist/src/registry/__tests__/tool-registry.test.js +124 -0
  144. package/dist/src/registry/__tests__/tool-registry.test.js.map +1 -0
  145. package/dist/src/registry/tool-registry.d.ts +32 -20
  146. package/dist/src/registry/tool-registry.d.ts.map +1 -1
  147. package/dist/src/registry/tool-registry.js +60 -158
  148. package/dist/src/registry/tool-registry.js.map +1 -1
  149. package/dist/src/session/__tests__/file-storage.test.node.js +75 -15
  150. package/dist/src/session/__tests__/file-storage.test.node.js.map +1 -1
  151. package/dist/src/session/__tests__/s3-storage.test.d.ts +2 -0
  152. package/dist/src/session/__tests__/s3-storage.test.d.ts.map +1 -0
  153. package/dist/src/session/__tests__/{s3-storage.test.node.js → s3-storage.test.js} +161 -75
  154. package/dist/src/session/__tests__/s3-storage.test.js.map +1 -0
  155. package/dist/src/session/__tests__/session-manager.test.d.ts +2 -0
  156. package/dist/src/session/__tests__/session-manager.test.d.ts.map +1 -0
  157. package/dist/src/session/__tests__/session-manager.test.js +461 -0
  158. package/dist/src/session/__tests__/session-manager.test.js.map +1 -0
  159. package/dist/src/session/__tests__/validation.test.js +28 -1
  160. package/dist/src/session/__tests__/validation.test.js.map +1 -1
  161. package/dist/src/session/file-storage.d.ts +53 -27
  162. package/dist/src/session/file-storage.d.ts.map +1 -1
  163. package/dist/src/session/file-storage.js +103 -52
  164. package/dist/src/session/file-storage.js.map +1 -1
  165. package/dist/src/session/index.d.ts +6 -13
  166. package/dist/src/session/index.d.ts.map +1 -1
  167. package/dist/src/session/index.js +4 -12
  168. package/dist/src/session/index.js.map +1 -1
  169. package/dist/src/session/s3-storage.d.ts +49 -20
  170. package/dist/src/session/s3-storage.d.ts.map +1 -1
  171. package/dist/src/session/s3-storage.js +120 -35
  172. package/dist/src/session/s3-storage.js.map +1 -1
  173. package/dist/src/session/session-manager.d.ts +83 -0
  174. package/dist/src/session/session-manager.d.ts.map +1 -0
  175. package/dist/src/session/session-manager.js +122 -0
  176. package/dist/src/session/session-manager.js.map +1 -0
  177. package/dist/src/session/storage.d.ts +19 -11
  178. package/dist/src/session/storage.d.ts.map +1 -1
  179. package/dist/src/session/types.d.ts +6 -18
  180. package/dist/src/session/types.d.ts.map +1 -1
  181. package/dist/src/session/validation.d.ts +7 -0
  182. package/dist/src/session/validation.d.ts.map +1 -1
  183. package/dist/src/session/validation.js +12 -0
  184. package/dist/src/session/validation.js.map +1 -1
  185. package/dist/src/structured-output/__tests__/context.test.js +13 -13
  186. package/dist/src/structured-output/__tests__/context.test.js.map +1 -1
  187. package/dist/src/structured-output/context.js +1 -1
  188. package/dist/src/structured-output/context.js.map +1 -1
  189. package/dist/src/telemetry/__tests__/config.test.d.ts +2 -0
  190. package/dist/src/telemetry/__tests__/config.test.d.ts.map +1 -0
  191. package/dist/src/telemetry/__tests__/config.test.js +40 -0
  192. package/dist/src/telemetry/__tests__/config.test.js.map +1 -0
  193. package/dist/src/telemetry/__tests__/config.test.node.d.ts +2 -0
  194. package/dist/src/telemetry/__tests__/config.test.node.d.ts.map +1 -0
  195. package/dist/src/telemetry/__tests__/config.test.node.js +103 -0
  196. package/dist/src/telemetry/__tests__/config.test.node.js.map +1 -0
  197. package/dist/src/telemetry/__tests__/json.test.d.ts +2 -0
  198. package/dist/src/telemetry/__tests__/json.test.d.ts.map +1 -0
  199. package/dist/src/telemetry/__tests__/json.test.js +89 -0
  200. package/dist/src/telemetry/__tests__/json.test.js.map +1 -0
  201. package/dist/src/telemetry/__tests__/meter.test.d.ts +2 -0
  202. package/dist/src/telemetry/__tests__/meter.test.d.ts.map +1 -0
  203. package/dist/src/telemetry/__tests__/meter.test.js +457 -0
  204. package/dist/src/telemetry/__tests__/meter.test.js.map +1 -0
  205. package/dist/src/telemetry/__tests__/tracer.test.node.d.ts +2 -0
  206. package/dist/src/telemetry/__tests__/tracer.test.node.d.ts.map +1 -0
  207. package/dist/src/telemetry/__tests__/tracer.test.node.js +611 -0
  208. package/dist/src/telemetry/__tests__/tracer.test.node.js.map +1 -0
  209. package/dist/src/telemetry/config.d.ts +82 -0
  210. package/dist/src/telemetry/config.d.ts.map +1 -0
  211. package/dist/src/telemetry/config.js +133 -0
  212. package/dist/src/telemetry/config.js.map +1 -0
  213. package/dist/src/telemetry/index.d.ts +34 -0
  214. package/dist/src/telemetry/index.d.ts.map +1 -0
  215. package/dist/src/telemetry/index.js +33 -0
  216. package/dist/src/telemetry/index.js.map +1 -0
  217. package/dist/src/telemetry/json.d.ts +11 -0
  218. package/dist/src/telemetry/json.d.ts.map +1 -0
  219. package/dist/src/telemetry/json.js +25 -0
  220. package/dist/src/telemetry/json.js.map +1 -0
  221. package/dist/src/telemetry/meter.d.ts +277 -0
  222. package/dist/src/telemetry/meter.d.ts.map +1 -0
  223. package/dist/src/telemetry/meter.js +292 -0
  224. package/dist/src/telemetry/meter.js.map +1 -0
  225. package/dist/src/telemetry/tracer.d.ts +219 -0
  226. package/dist/src/telemetry/tracer.d.ts.map +1 -0
  227. package/dist/src/telemetry/tracer.js +610 -0
  228. package/dist/src/telemetry/tracer.js.map +1 -0
  229. package/dist/src/telemetry/types.d.ts +101 -0
  230. package/dist/src/telemetry/types.d.ts.map +1 -0
  231. package/dist/src/telemetry/types.js +5 -0
  232. package/dist/src/{multiagent → telemetry}/types.js.map +1 -1
  233. package/dist/src/tools/__tests__/tool-factory.test.d.ts +2 -0
  234. package/dist/src/tools/__tests__/tool-factory.test.d.ts.map +1 -0
  235. package/dist/src/tools/__tests__/tool-factory.test.js +98 -0
  236. package/dist/src/tools/__tests__/tool-factory.test.js.map +1 -0
  237. package/dist/src/tools/__tests__/zod-tool.test-d.js +1 -1
  238. package/dist/src/tools/__tests__/zod-tool.test-d.js.map +1 -1
  239. package/dist/src/tools/__tests__/zod-tool.test.js +3 -4
  240. package/dist/src/tools/__tests__/zod-tool.test.js.map +1 -1
  241. package/dist/src/tools/function-tool.d.ts +15 -2
  242. package/dist/src/tools/function-tool.d.ts.map +1 -1
  243. package/dist/src/tools/function-tool.js +24 -0
  244. package/dist/src/tools/function-tool.js.map +1 -1
  245. package/dist/src/tools/tool-factory.d.ts +22 -0
  246. package/dist/src/tools/tool-factory.d.ts.map +1 -0
  247. package/dist/src/tools/tool-factory.js +55 -0
  248. package/dist/src/tools/tool-factory.js.map +1 -0
  249. package/dist/src/tools/tool.d.ts +1 -1
  250. package/dist/src/tools/tool.js +1 -1
  251. package/dist/src/tools/zod-tool.d.ts +55 -52
  252. package/dist/src/tools/zod-tool.d.ts.map +1 -1
  253. package/dist/src/tools/zod-tool.js +7 -61
  254. package/dist/src/tools/zod-tool.js.map +1 -1
  255. package/dist/src/tsconfig.tsbuildinfo +1 -1
  256. package/dist/src/types/__tests__/agent.test.js +11 -0
  257. package/dist/src/types/__tests__/agent.test.js.map +1 -1
  258. package/dist/src/types/__tests__/citations.test.d.ts +2 -0
  259. package/dist/src/types/__tests__/citations.test.d.ts.map +1 -0
  260. package/dist/src/types/__tests__/citations.test.js +104 -0
  261. package/dist/src/types/__tests__/citations.test.js.map +1 -0
  262. package/dist/src/types/__tests__/messages.test.js +26 -0
  263. package/dist/src/types/__tests__/messages.test.js.map +1 -1
  264. package/dist/src/types/agent.d.ts +10 -3
  265. package/dist/src/types/agent.d.ts.map +1 -1
  266. package/dist/src/types/agent.js +9 -0
  267. package/dist/src/types/agent.js.map +1 -1
  268. package/dist/src/types/citations.d.ts +180 -0
  269. package/dist/src/types/citations.d.ts.map +1 -0
  270. package/dist/src/types/citations.js +45 -0
  271. package/dist/src/types/citations.js.map +1 -0
  272. package/dist/src/types/messages.d.ts +9 -5
  273. package/dist/src/types/messages.d.ts.map +1 -1
  274. package/dist/src/types/messages.js +5 -1
  275. package/dist/src/types/messages.js.map +1 -1
  276. package/dist/src/vended-tools/bash/__tests__/bash.test.node.js +2 -2
  277. package/dist/src/vended-tools/bash/__tests__/bash.test.node.js.map +1 -1
  278. package/dist/src/vended-tools/bash/bash.d.ts.map +1 -1
  279. package/dist/src/vended-tools/bash/bash.js +1 -2
  280. package/dist/src/vended-tools/bash/bash.js.map +1 -1
  281. package/dist/src/vended-tools/file_editor/__tests__/file-editor.test.node.js +4 -4
  282. package/dist/src/vended-tools/file_editor/__tests__/file-editor.test.node.js.map +1 -1
  283. package/dist/src/vended-tools/file_editor/file-editor.js +1 -1
  284. package/dist/src/vended-tools/file_editor/file-editor.js.map +1 -1
  285. package/dist/src/vended-tools/http_request/http-request.d.ts.map +1 -1
  286. package/dist/src/vended-tools/http_request/http-request.js +1 -2
  287. package/dist/src/vended-tools/http_request/http-request.js.map +1 -1
  288. package/dist/src/vended-tools/notebook/__tests__/notebook.test.js +2 -2
  289. package/dist/src/vended-tools/notebook/__tests__/notebook.test.js.map +1 -1
  290. package/dist/src/vended-tools/notebook/notebook.d.ts +1 -1
  291. package/package.json +25 -4
  292. package/dist/src/agent/__tests__/state.test.d.ts +0 -2
  293. package/dist/src/agent/__tests__/state.test.d.ts.map +0 -1
  294. package/dist/src/agent/__tests__/state.test.js.map +0 -1
  295. package/dist/src/agent/state.d.ts.map +0 -1
  296. package/dist/src/agent/state.js.map +0 -1
  297. package/dist/src/multiagent/types.d.ts +0 -5
  298. package/dist/src/multiagent/types.d.ts.map +0 -1
  299. package/dist/src/multiagent/types.js +0 -2
  300. package/dist/src/registry/registry.d.ts +0 -117
  301. package/dist/src/registry/registry.d.ts.map +0 -1
  302. package/dist/src/registry/registry.js +0 -298
  303. package/dist/src/registry/registry.js.map +0 -1
  304. package/dist/src/session/__tests__/s3-storage.test.node.d.ts +0 -2
  305. package/dist/src/session/__tests__/s3-storage.test.node.d.ts.map +0 -1
  306. package/dist/src/session/__tests__/s3-storage.test.node.js.map +0 -1
@@ -1,6 +1,22 @@
1
1
  import { Message, ReasoningBlock, TextBlock, ToolUseBlock, } from '../types/messages.js';
2
- import { ModelContentBlockDeltaEvent, ModelContentBlockStartEvent, ModelContentBlockStopEvent, ModelMessageStartEvent, ModelMessageStopEvent, ModelMetadataEvent, } from './streaming.js';
2
+ import { CitationsBlock } from '../types/citations.js';
3
+ import { ModelContentBlockDeltaEvent, ModelContentBlockStartEvent, ModelContentBlockStopEvent, ModelMessageStartEvent, ModelMessageStopEvent, ModelMetadataEvent, ModelRedactionEvent, } from './streaming.js';
3
4
  import { MaxTokensError, ModelError, normalizeError } from '../errors.js';
5
+ class CitationAccumulator {
6
+ citations = [];
7
+ content = [];
8
+ push(citations, content) {
9
+ this.citations.push(...citations);
10
+ this.content.push(...content);
11
+ }
12
+ hasData() {
13
+ return this.citations.length > 0;
14
+ }
15
+ reset() {
16
+ this.citations = [];
17
+ this.content = [];
18
+ }
19
+ }
4
20
  /**
5
21
  * Base abstract class for model providers.
6
22
  * Defines the contract that all model provider implementations must follow.
@@ -11,6 +27,12 @@ import { MaxTokensError, ModelError, normalizeError } from '../errors.js';
11
27
  * @typeParam T - Model configuration type extending BaseModelConfig
12
28
  */
13
29
  export class Model {
30
+ /**
31
+ * The model ID from the current configuration, if configured.
32
+ */
33
+ get modelId() {
34
+ return this.getConfig().modelId;
35
+ }
14
36
  /**
15
37
  * Converts event data to event class representation
16
38
  *
@@ -31,6 +53,8 @@ export class Model {
31
53
  return new ModelMessageStopEvent(event_data);
32
54
  case 'modelMetadataEvent':
33
55
  return new ModelMetadataEvent(event_data);
56
+ case 'modelRedactionEvent':
57
+ return new ModelRedactionEvent(event_data);
34
58
  default:
35
59
  throw new Error(`Unsupported event type: ${event_data}`);
36
60
  }
@@ -73,10 +97,12 @@ export class Model {
73
97
  let toolUseId = '';
74
98
  let toolReasoningSignature = '';
75
99
  let accumulatedReasoning = {};
100
+ const accumulatedCitations = new CitationAccumulator();
76
101
  let errorToThrow = undefined;
77
102
  let stoppedMessage = null;
78
103
  let finalStopReason = null;
79
104
  let metadata = undefined;
105
+ let redactionMessage = undefined;
80
106
  for await (const event_data of this.stream(messages, options)) {
81
107
  const event = this._convert_to_class_event(event_data);
82
108
  yield event; // Pass through immediately
@@ -95,8 +121,9 @@ export class Model {
95
121
  accumulatedToolInput = '';
96
122
  accumulatedText = '';
97
123
  accumulatedReasoning = {};
124
+ accumulatedCitations.reset();
98
125
  break;
99
- case 'modelContentBlockDeltaEvent':
126
+ case 'modelContentBlockDeltaEvent': {
100
127
  switch (event.delta.type) {
101
128
  case 'textDelta':
102
129
  accumulatedText += event.delta.text;
@@ -112,8 +139,12 @@ export class Model {
112
139
  if (event.delta.redactedContent)
113
140
  accumulatedReasoning.redactedContent = event.delta.redactedContent;
114
141
  break;
142
+ case 'citationsDelta':
143
+ accumulatedCitations.push(event.delta.citations, event.delta.content);
144
+ break;
115
145
  }
116
146
  break;
147
+ }
117
148
  case 'modelContentBlockStopEvent': {
118
149
  // Finalize and emit complete ContentBlock
119
150
  let block;
@@ -135,6 +166,13 @@ export class Model {
135
166
  });
136
167
  accumulatedReasoning = {}; // Reset after creating reasoning block
137
168
  }
169
+ else if (accumulatedCitations.hasData()) {
170
+ block = new CitationsBlock({
171
+ citations: accumulatedCitations.citations,
172
+ content: accumulatedCitations.content,
173
+ });
174
+ accumulatedCitations.reset();
175
+ }
138
176
  else {
139
177
  block = new TextBlock(accumulatedText);
140
178
  }
@@ -163,6 +201,21 @@ export class Model {
163
201
  // Store metadata, keeping the last one if multiple events occur
164
202
  metadata = event;
165
203
  break;
204
+ case 'modelRedactionEvent':
205
+ // Handle content redaction from guardrails
206
+ if (event.inputRedaction) {
207
+ // Store redaction message for agent to handle input message redaction
208
+ redactionMessage = event.inputRedaction.replaceContent;
209
+ }
210
+ if (event.outputRedaction) {
211
+ // Update output message directly with redacted content
212
+ // Redaction event comes after modelMessageStopEvent, so we overwrite stoppedMessage
213
+ stoppedMessage = new Message({
214
+ role: 'assistant',
215
+ content: [new TextBlock(event.outputRedaction.replaceContent)],
216
+ });
217
+ }
218
+ break;
166
219
  default:
167
220
  break;
168
221
  }
@@ -187,6 +240,9 @@ export class Model {
187
240
  if (metadata !== undefined) {
188
241
  result.metadata = metadata;
189
242
  }
243
+ if (redactionMessage !== undefined) {
244
+ result.redaction = { userMessage: redactionMessage };
245
+ }
190
246
  return result;
191
247
  }
192
248
  catch (error) {
@@ -1 +1 @@
1
- {"version":3,"file":"model.js","sourceRoot":"","sources":["../../../src/models/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,OAAO,EACP,cAAc,EAId,SAAS,EACT,YAAY,GACb,MAAM,sBAAsB,CAAA;AAE7B,OAAO,EACL,2BAA2B,EAC3B,2BAA2B,EAC3B,0BAA0B,EAC1B,sBAAsB,EACtB,qBAAqB,EACrB,kBAAkB,GAEnB,MAAM,gBAAgB,CAAA;AACvB,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,cAAc,CAAA;AAgFzE;;;;;;;;GAQG;AACH,MAAM,OAAgB,KAAK;IA0BzB;;;;;OAKG;IACK,uBAAuB,CAAC,UAA4B;QAC1D,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;YACxB,KAAK,wBAAwB;gBAC3B,OAAO,IAAI,sBAAsB,CAAC,UAAU,CAAC,CAAA;YAC/C,KAAK,6BAA6B;gBAChC,OAAO,IAAI,2BAA2B,CAAC,UAAU,CAAC,CAAA;YACpD,KAAK,6BAA6B;gBAChC,OAAO,IAAI,2BAA2B,CAAC,UAAU,CAAC,CAAA;YACpD,KAAK,4BAA4B;gBAC/B,OAAO,IAAI,0BAA0B,CAAC,UAAU,CAAC,CAAA;YACnD,KAAK,uBAAuB;gBAC1B,OAAO,IAAI,qBAAqB,CAAC,UAAU,CAAC,CAAA;YAC9C,KAAK,oBAAoB;gBACvB,OAAO,IAAI,kBAAkB,CAAC,UAAU,CAAC,CAAA;YAC3C;gBACE,MAAM,IAAI,KAAK,CAAC,2BAA2B,UAAU,EAAE,CAAC,CAAA;QAC5D,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,KAAK,CAAC,CAAC,gBAAgB,CACrB,QAAmB,EACnB,OAAuB;QAEvB,IAAI,CAAC;YACH,8BAA8B;YAC9B,IAAI,WAAW,GAAgB,IAAI,CAAA;YACnC,MAAM,aAAa,GAAmB,EAAE,CAAA;YACxC,IAAI,eAAe,GAAG,EAAE,CAAA;YACxB,IAAI,oBAAoB,GAAG,EAAE,CAAA;YAC7B,IAAI,QAAQ,GAAG,EAAE,CAAA;YACjB,IAAI,SAAS,GAAG,EAAE,CAAA;YAClB,IAAI,sBAAsB,GAAG,EAAE,CAAA;YAC/B,IAAI,oBAAoB,GAIpB,EAAE,CAAA;YACN,IAAI,YAAY,GAAsB,SAAS,CAAA;YAC/C,IAAI,cAAc,GAAmB,IAAI,CAAA;YACzC,IAAI,eAAe,GAAsB,IAAI,CAAA;YAC7C,IAAI,QAAQ,GAAmC,SAAS,CAAA;YAExD,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC9D,MAAM,KAAK,GAAG,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAA;gBACtD,MAAM,KAAK,CAAA,CAAC,2BAA2B;gBAEvC,wCAAwC;gBACxC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACnB,KAAK,wBAAwB;wBAC3B,WAAW,GAAG,KAAK,CAAC,IAAI,CAAA;wBACxB,aAAa,CAAC,MAAM,GAAG,CAAC,CAAA,CAAC,QAAQ;wBACjC,MAAK;oBAEP,KAAK,6BAA6B;wBAChC,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,cAAc,EAAE,CAAC;4BACzC,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAA;4BAC3B,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,CAAA;4BACjC,sBAAsB,GAAG,KAAK,CAAC,KAAK,CAAC,kBAAkB,IAAI,EAAE,CAAA;wBAC/D,CAAC;wBACD,oBAAoB,GAAG,EAAE,CAAA;wBACzB,eAAe,GAAG,EAAE,CAAA;wBACpB,oBAAoB,GAAG,EAAE,CAAA;wBACzB,MAAK;oBAEP,KAAK,6BAA6B;wBAChC,QAAQ,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;4BACzB,KAAK,WAAW;gCACd,eAAe,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,CAAA;gCACnC,MAAK;4BACP,KAAK,mBAAmB;gCACtB,oBAAoB,IAAI,KAAK,CAAC,KAAK,CAAC,KAAK,CAAA;gCACzC,MAAK;4BACP,KAAK,uBAAuB;gCAC1B,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI;oCAAE,oBAAoB,CAAC,IAAI,GAAG,CAAC,oBAAoB,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAA;gCACtG,IAAI,KAAK,CAAC,KAAK,CAAC,SAAS;oCAAE,oBAAoB,CAAC,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,CAAA;gCACjF,IAAI,KAAK,CAAC,KAAK,CAAC,eAAe;oCAAE,oBAAoB,CAAC,eAAe,GAAG,KAAK,CAAC,KAAK,CAAC,eAAe,CAAA;gCACnG,MAAK;wBACT,CAAC;wBACD,MAAK;oBAEP,KAAK,4BAA4B,CAAC,CAAC,CAAC;wBAClC,0CAA0C;wBAC1C,IAAI,KAAmB,CAAA;wBACvB,IAAI,CAAC;4BACH,IAAI,SAAS,EAAE,CAAC;gCACd,KAAK,GAAG,IAAI,YAAY,CAAC;oCACvB,IAAI,EAAE,QAAQ;oCACd,SAAS,EAAE,SAAS;oCACpB,KAAK,EAAE,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,EAAE;oCACnE,GAAG,CAAC,sBAAsB,IAAI,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,CAAC;iCAC9E,CAAC,CAAA;gCACF,SAAS,GAAG,EAAE,CAAA,CAAC,QAAQ;gCACvB,QAAQ,GAAG,EAAE,CAAA;gCACb,sBAAsB,GAAG,EAAE,CAAA;4BAC7B,CAAC;iCAAM,IAAI,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gCACxD,KAAK,GAAG,IAAI,cAAc,CAAC;oCACzB,GAAG,oBAAoB;iCACxB,CAAC,CAAA;gCACF,oBAAoB,GAAG,EAAE,CAAA,CAAC,uCAAuC;4BACnE,CAAC;iCAAM,CAAC;gCACN,KAAK,GAAG,IAAI,SAAS,CAAC,eAAe,CAAC,CAAA;4BACxC,CAAC;4BACD,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;4BACzB,MAAM,KAAK,CAAA;wBACb,CAAC;wBAAC,OAAO,CAAU,EAAE,CAAC;4BACpB,IAAI,CAAC,YAAY,WAAW,EAAE,CAAC;gCAC7B,OAAO,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAA;gCAC7C,YAAY,GAAG,CAAC,CAAA;4BAClB,CAAC;wBACH,CAAC;wBACD,MAAK;oBACP,CAAC;oBAED,KAAK,uBAAuB;wBAC1B,gCAAgC;wBAChC,IAAI,WAAW,EAAE,CAAC;4BAChB,cAAc,GAAG,IAAI,OAAO,CAAC;gCAC3B,IAAI,EAAE,WAAW;gCACjB,OAAO,EAAE,CAAC,GAAG,aAAa,CAAC;6BAC5B,CAAC,CAAA;4BACF,eAAe,GAAG,KAAK,CAAC,UAAW,CAAA;wBACrC,CAAC;wBACD,MAAK;oBAEP,KAAK,oBAAoB;wBACvB,gEAAgE;wBAChE,QAAQ,GAAG,KAAK,CAAA;wBAChB,MAAK;oBAEP;wBACE,MAAK;gBACT,CAAC;YACH,CAAC;YAED,IAAI,CAAC,cAAc,IAAI,CAAC,eAAe,EAAE,CAAC;gBACxC,kFAAkF;gBAClF,MAAM,IAAI,UAAU,CAClB,2CAA2C,EAC3C,YAAY,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CACnD,CAAA;YACH,CAAC;YAED,qBAAqB;YACrB,IAAI,eAAe,KAAK,WAAW,EAAE,CAAC;gBACpC,MAAM,cAAc,GAAG,IAAI,cAAc,CACvC,+FAA+F,EAC/F,cAAc,CACf,CAAA;gBACD,YAAY,GAAG,cAAc,CAAA;YAC/B,CAAC;YAED,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAC/B,MAAM,YAAY,CAAA;YACpB,CAAC;YAED,kEAAkE;YAClE,MAAM,MAAM,GAA2B;gBACrC,OAAO,EAAE,cAAc;gBACvB,UAAU,EAAE,eAAe;aAC5B,CAAA;YACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBAC3B,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAA;YAC5B,CAAC;YACD,OAAO,MAAM,CAAA;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,2CAA2C;YAC3C,IAAI,KAAK,YAAY,UAAU,EAAE,CAAC;gBAChC,MAAM,KAAK,CAAA;YACb,CAAC;YACD,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC,CAAA;YAC7C,MAAM,IAAI,UAAU,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAA;QACjE,CAAC;IACH,CAAC;CACF"}
1
+ {"version":3,"file":"model.js","sourceRoot":"","sources":["../../../src/models/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,OAAO,EACP,cAAc,EAId,SAAS,EACT,YAAY,GACb,MAAM,sBAAsB,CAAA;AAC7B,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAA;AAGtD,OAAO,EACL,2BAA2B,EAC3B,2BAA2B,EAC3B,0BAA0B,EAC1B,sBAAsB,EACtB,qBAAqB,EACrB,kBAAkB,EAClB,mBAAmB,GAEpB,MAAM,gBAAgB,CAAA;AACvB,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,cAAc,CAAA;AAGzE,MAAM,mBAAmB;IACvB,SAAS,GAAe,EAAE,CAAA;IAC1B,OAAO,GAA+B,EAAE,CAAA;IAExC,IAAI,CAAC,SAAqB,EAAE,OAAmC;QAC7D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAA;QACjC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAA;IAC/B,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAA;IAClC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,SAAS,GAAG,EAAE,CAAA;QACnB,IAAI,CAAC,OAAO,GAAG,EAAE,CAAA;IACnB,CAAC;CACF;AAsFD;;;;;;;;GAQG;AACH,MAAM,OAAgB,KAAK;IAgBzB;;OAEG;IACH,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC,OAAO,CAAA;IACjC,CAAC;IAYD;;;;;OAKG;IACK,uBAAuB,CAAC,UAA4B;QAC1D,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;YACxB,KAAK,wBAAwB;gBAC3B,OAAO,IAAI,sBAAsB,CAAC,UAAU,CAAC,CAAA;YAC/C,KAAK,6BAA6B;gBAChC,OAAO,IAAI,2BAA2B,CAAC,UAAU,CAAC,CAAA;YACpD,KAAK,6BAA6B;gBAChC,OAAO,IAAI,2BAA2B,CAAC,UAAU,CAAC,CAAA;YACpD,KAAK,4BAA4B;gBAC/B,OAAO,IAAI,0BAA0B,CAAC,UAAU,CAAC,CAAA;YACnD,KAAK,uBAAuB;gBAC1B,OAAO,IAAI,qBAAqB,CAAC,UAAU,CAAC,CAAA;YAC9C,KAAK,oBAAoB;gBACvB,OAAO,IAAI,kBAAkB,CAAC,UAAU,CAAC,CAAA;YAC3C,KAAK,qBAAqB;gBACxB,OAAO,IAAI,mBAAmB,CAAC,UAAU,CAAC,CAAA;YAC5C;gBACE,MAAM,IAAI,KAAK,CAAC,2BAA2B,UAAU,EAAE,CAAC,CAAA;QAC5D,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;OA0BG;IACH,KAAK,CAAC,CAAC,gBAAgB,CACrB,QAAmB,EACnB,OAAuB;QAEvB,IAAI,CAAC;YACH,8BAA8B;YAC9B,IAAI,WAAW,GAAgB,IAAI,CAAA;YACnC,MAAM,aAAa,GAAmB,EAAE,CAAA;YACxC,IAAI,eAAe,GAAG,EAAE,CAAA;YACxB,IAAI,oBAAoB,GAAG,EAAE,CAAA;YAC7B,IAAI,QAAQ,GAAG,EAAE,CAAA;YACjB,IAAI,SAAS,GAAG,EAAE,CAAA;YAClB,IAAI,sBAAsB,GAAG,EAAE,CAAA;YAC/B,IAAI,oBAAoB,GAIpB,EAAE,CAAA;YACN,MAAM,oBAAoB,GAAG,IAAI,mBAAmB,EAAE,CAAA;YACtD,IAAI,YAAY,GAAsB,SAAS,CAAA;YAC/C,IAAI,cAAc,GAAmB,IAAI,CAAA;YACzC,IAAI,eAAe,GAAsB,IAAI,CAAA;YAC7C,IAAI,QAAQ,GAAmC,SAAS,CAAA;YACxD,IAAI,gBAAgB,GAAuB,SAAS,CAAA;YAEpD,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC9D,MAAM,KAAK,GAAG,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAA;gBACtD,MAAM,KAAK,CAAA,CAAC,2BAA2B;gBAEvC,wCAAwC;gBACxC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACnB,KAAK,wBAAwB;wBAC3B,WAAW,GAAG,KAAK,CAAC,IAAI,CAAA;wBACxB,aAAa,CAAC,MAAM,GAAG,CAAC,CAAA,CAAC,QAAQ;wBACjC,MAAK;oBAEP,KAAK,6BAA6B;wBAChC,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,cAAc,EAAE,CAAC;4BACzC,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAA;4BAC3B,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,CAAA;4BACjC,sBAAsB,GAAG,KAAK,CAAC,KAAK,CAAC,kBAAkB,IAAI,EAAE,CAAA;wBAC/D,CAAC;wBACD,oBAAoB,GAAG,EAAE,CAAA;wBACzB,eAAe,GAAG,EAAE,CAAA;wBACpB,oBAAoB,GAAG,EAAE,CAAA;wBACzB,oBAAoB,CAAC,KAAK,EAAE,CAAA;wBAC5B,MAAK;oBAEP,KAAK,6BAA6B,CAAC,CAAC,CAAC;wBACnC,QAAQ,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;4BACzB,KAAK,WAAW;gCACd,eAAe,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,CAAA;gCACnC,MAAK;4BACP,KAAK,mBAAmB;gCACtB,oBAAoB,IAAI,KAAK,CAAC,KAAK,CAAC,KAAK,CAAA;gCACzC,MAAK;4BACP,KAAK,uBAAuB;gCAC1B,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI;oCAAE,oBAAoB,CAAC,IAAI,GAAG,CAAC,oBAAoB,CAAC,IAAI,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAA;gCACtG,IAAI,KAAK,CAAC,KAAK,CAAC,SAAS;oCAAE,oBAAoB,CAAC,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,CAAA;gCACjF,IAAI,KAAK,CAAC,KAAK,CAAC,eAAe;oCAAE,oBAAoB,CAAC,eAAe,GAAG,KAAK,CAAC,KAAK,CAAC,eAAe,CAAA;gCACnG,MAAK;4BACP,KAAK,gBAAgB;gCACnB,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;gCACrE,MAAK;wBACT,CAAC;wBACD,MAAK;oBACP,CAAC;oBAED,KAAK,4BAA4B,CAAC,CAAC,CAAC;wBAClC,0CAA0C;wBAC1C,IAAI,KAAmB,CAAA;wBACvB,IAAI,CAAC;4BACH,IAAI,SAAS,EAAE,CAAC;gCACd,KAAK,GAAG,IAAI,YAAY,CAAC;oCACvB,IAAI,EAAE,QAAQ;oCACd,SAAS,EAAE,SAAS;oCACpB,KAAK,EAAE,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,EAAE;oCACnE,GAAG,CAAC,sBAAsB,IAAI,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,CAAC;iCAC9E,CAAC,CAAA;gCACF,SAAS,GAAG,EAAE,CAAA,CAAC,QAAQ;gCACvB,QAAQ,GAAG,EAAE,CAAA;gCACb,sBAAsB,GAAG,EAAE,CAAA;4BAC7B,CAAC;iCAAM,IAAI,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gCACxD,KAAK,GAAG,IAAI,cAAc,CAAC;oCACzB,GAAG,oBAAoB;iCACxB,CAAC,CAAA;gCACF,oBAAoB,GAAG,EAAE,CAAA,CAAC,uCAAuC;4BACnE,CAAC;iCAAM,IAAI,oBAAoB,CAAC,OAAO,EAAE,EAAE,CAAC;gCAC1C,KAAK,GAAG,IAAI,cAAc,CAAC;oCACzB,SAAS,EAAE,oBAAoB,CAAC,SAAS;oCACzC,OAAO,EAAE,oBAAoB,CAAC,OAAO;iCACtC,CAAC,CAAA;gCACF,oBAAoB,CAAC,KAAK,EAAE,CAAA;4BAC9B,CAAC;iCAAM,CAAC;gCACN,KAAK,GAAG,IAAI,SAAS,CAAC,eAAe,CAAC,CAAA;4BACxC,CAAC;4BACD,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;4BACzB,MAAM,KAAK,CAAA;wBACb,CAAC;wBAAC,OAAO,CAAU,EAAE,CAAC;4BACpB,IAAI,CAAC,YAAY,WAAW,EAAE,CAAC;gCAC7B,OAAO,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAA;gCAC7C,YAAY,GAAG,CAAC,CAAA;4BAClB,CAAC;wBACH,CAAC;wBACD,MAAK;oBACP,CAAC;oBAED,KAAK,uBAAuB;wBAC1B,gCAAgC;wBAChC,IAAI,WAAW,EAAE,CAAC;4BAChB,cAAc,GAAG,IAAI,OAAO,CAAC;gCAC3B,IAAI,EAAE,WAAW;gCACjB,OAAO,EAAE,CAAC,GAAG,aAAa,CAAC;6BAC5B,CAAC,CAAA;4BACF,eAAe,GAAG,KAAK,CAAC,UAAW,CAAA;wBACrC,CAAC;wBACD,MAAK;oBAEP,KAAK,oBAAoB;wBACvB,gEAAgE;wBAChE,QAAQ,GAAG,KAAK,CAAA;wBAChB,MAAK;oBAEP,KAAK,qBAAqB;wBACxB,2CAA2C;wBAC3C,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;4BACzB,sEAAsE;4BACtE,gBAAgB,GAAG,KAAK,CAAC,cAAc,CAAC,cAAc,CAAA;wBACxD,CAAC;wBACD,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;4BAC1B,uDAAuD;4BACvD,oFAAoF;4BACpF,cAAc,GAAG,IAAI,OAAO,CAAC;gCAC3B,IAAI,EAAE,WAAW;gCACjB,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,KAAK,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;6BAC/D,CAAC,CAAA;wBACJ,CAAC;wBACD,MAAK;oBAEP;wBACE,MAAK;gBACT,CAAC;YACH,CAAC;YAED,IAAI,CAAC,cAAc,IAAI,CAAC,eAAe,EAAE,CAAC;gBACxC,kFAAkF;gBAClF,MAAM,IAAI,UAAU,CAClB,2CAA2C,EAC3C,YAAY,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CACnD,CAAA;YACH,CAAC;YAED,qBAAqB;YACrB,IAAI,eAAe,KAAK,WAAW,EAAE,CAAC;gBACpC,MAAM,cAAc,GAAG,IAAI,cAAc,CACvC,+FAA+F,EAC/F,cAAc,CACf,CAAA;gBACD,YAAY,GAAG,cAAc,CAAA;YAC/B,CAAC;YAED,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAC/B,MAAM,YAAY,CAAA;YACpB,CAAC;YAED,kEAAkE;YAClE,MAAM,MAAM,GAA2B;gBACrC,OAAO,EAAE,cAAc;gBACvB,UAAU,EAAE,eAAe;aAC5B,CAAA;YACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBAC3B,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAA;YAC5B,CAAC;YACD,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;gBACnC,MAAM,CAAC,SAAS,GAAG,EAAE,WAAW,EAAE,gBAAgB,EAAE,CAAA;YACtD,CAAC;YACD,OAAO,MAAM,CAAA;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,2CAA2C;YAC3C,IAAI,KAAK,YAAY,UAAU,EAAE,CAAC;gBAChC,MAAM,KAAK,CAAA;YACb,CAAC;YACD,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC,CAAA;YAC7C,MAAM,IAAI,UAAU,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAA;QACjE,CAAC;IACH,CAAC;CACF"}
@@ -1,9 +1,10 @@
1
1
  import type { Role, StopReason } from '../types/messages.js';
2
2
  import type { JSONValue } from '../types/json.js';
3
+ import type { Citation, CitationGeneratedContent } from '../types/citations.js';
3
4
  /**
4
5
  * ModelStreamEvent types for Model interactions.
5
6
  *
6
- * This module follows a pattern where <name>Data interfaces define the structure
7
+ * This module follows a pattern where "Data" interfaces define the structure
7
8
  * for objects, while corresponding classes extend those interfaces with additional
8
9
  * functionality and type discrimination.
9
10
  */
@@ -13,7 +14,7 @@ import type { JSONValue } from '../types/json.js';
13
14
  *
14
15
  * This allows for type-safe event handling using switch statements.
15
16
  */
16
- export type ModelStreamEvent = ModelMessageStartEventData | ModelContentBlockStartEventData | ModelContentBlockDeltaEventData | ModelContentBlockStopEventData | ModelMessageStopEventData | ModelMetadataEventData;
17
+ export type ModelStreamEvent = ModelMessageStartEventData | ModelContentBlockStartEventData | ModelContentBlockDeltaEventData | ModelContentBlockStopEventData | ModelMessageStopEventData | ModelMetadataEventData | ModelRedactionEventData;
17
18
  /**
18
19
  * Type guard to check if an event with a type discriminator is a ModelStreamEvent.
19
20
  * @param event - The event to check
@@ -207,6 +208,68 @@ export declare class ModelMetadataEvent implements ModelMetadataEventData {
207
208
  readonly trace?: unknown;
208
209
  constructor(data: ModelMetadataEventData);
209
210
  }
211
+ /**
212
+ * Information about input content redaction.
213
+ * Does not include redactedContent since the original input is already available
214
+ * in the messages array from BeforeModelCallEvent.
215
+ */
216
+ export interface RedactInputContent {
217
+ /**
218
+ * The content to replace the redacted input with.
219
+ */
220
+ replaceContent: string;
221
+ }
222
+ /**
223
+ * Information about output content redaction.
224
+ * May include the original content if captured during streaming.
225
+ */
226
+ export interface RedactOutputContent {
227
+ /**
228
+ * The original content that was blocked by guardrails.
229
+ * May not be available for all providers.
230
+ */
231
+ redactedContent?: string;
232
+ /**
233
+ * The content to replace the redacted output with.
234
+ */
235
+ replaceContent: string;
236
+ }
237
+ /**
238
+ * Data for a redact event.
239
+ * Emitted when guardrails block content and redaction is enabled.
240
+ */
241
+ export interface ModelRedactionEventData {
242
+ /**
243
+ * Discriminator for redact events.
244
+ */
245
+ type: 'modelRedactionEvent';
246
+ /**
247
+ * Input redaction information (when input is blocked).
248
+ */
249
+ inputRedaction?: RedactInputContent;
250
+ /**
251
+ * Output redaction information (when output is blocked).
252
+ */
253
+ outputRedaction?: RedactOutputContent;
254
+ }
255
+ /**
256
+ * Event emitted when guardrails block content and trigger redaction.
257
+ */
258
+ export declare class ModelRedactionEvent implements ModelRedactionEventData {
259
+ /**
260
+ * Discriminator for redact events.
261
+ */
262
+ readonly type: "modelRedactionEvent";
263
+ /**
264
+ * Input redaction information (when input is blocked).
265
+ */
266
+ readonly inputRedaction?: RedactInputContent;
267
+ /**
268
+ * Output redaction information (when output is blocked).
269
+ */
270
+ readonly outputRedaction?: RedactOutputContent;
271
+ constructor(data: ModelRedactionEventData);
272
+ }
210
273
  /**
211
274
  * Information about a content block that is starting.
212
275
  * Currently only represents tool use starts.
@@ -240,7 +303,7 @@ export interface ToolUseStart {
240
303
  *
241
304
  * This is a discriminated union for type-safe delta handling.
242
305
  */
243
- export type ContentBlockDelta = TextDelta | ToolUseInputDelta | ReasoningContentDelta;
306
+ export type ContentBlockDelta = TextDelta | ToolUseInputDelta | ReasoningContentDelta | CitationsDelta;
244
307
  /**
245
308
  * Text delta within a content block.
246
309
  * Represents incremental text content from the model.
@@ -291,6 +354,24 @@ export interface ReasoningContentDelta {
291
354
  */
292
355
  redactedContent?: Uint8Array;
293
356
  }
357
+ /**
358
+ * Citations content delta within a content block.
359
+ * Represents a citations content block from the model.
360
+ */
361
+ export interface CitationsDelta {
362
+ /**
363
+ * Discriminator for citations content delta.
364
+ */
365
+ type: 'citationsDelta';
366
+ /**
367
+ * Array of citations linking generated content to source locations.
368
+ */
369
+ citations: Citation[];
370
+ /**
371
+ * The generated content associated with these citations.
372
+ */
373
+ content: CitationGeneratedContent[];
374
+ }
294
375
  /**
295
376
  * Token usage statistics for a model invocation.
296
377
  * Tracks input, output, and total tokens, plus cache-related metrics.
@@ -1 +1 @@
1
- {"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../../src/models/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAA;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAEjD;;;;;;GAMG;AAEH;;;;;GAKG;AACH,MAAM,MAAM,gBAAgB,GACxB,0BAA0B,GAC1B,+BAA+B,GAC/B,+BAA+B,GAC/B,8BAA8B,GAC9B,yBAAyB,GACzB,sBAAsB,CAAA;AAY1B;;;;GAIG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,GAAG,KAAK,IAAI,gBAAgB,CAErF;AAED;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC;;OAEG;IACH,IAAI,EAAE,wBAAwB,CAAA;IAE9B;;OAEG;IACH,IAAI,EAAE,IAAI,CAAA;CACX;AAED;;GAEG;AACH,qBAAa,sBAAuB,YAAW,0BAA0B;IACvE;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,wBAAwB,CAAS;IAEjD;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAA;gBAEP,IAAI,EAAE,0BAA0B;CAG7C;AAED;;GAEG;AACH,MAAM,WAAW,+BAA+B;IAC9C;;OAEG;IACH,IAAI,EAAE,6BAA6B,CAAA;IAEnC;;;OAGG;IACH,KAAK,CAAC,EAAE,iBAAiB,CAAA;CAC1B;AAED;;GAEG;AACH,qBAAa,2BAA4B,YAAW,+BAA+B;IACjF;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,6BAA6B,CAAS;IAEtD;;;OAGG;IACH,QAAQ,CAAC,KAAK,CAAC,EAAE,iBAAiB,CAAA;gBAEtB,IAAI,EAAE,+BAA+B;CAKlD;AAED;;GAEG;AACH,MAAM,WAAW,+BAA+B;IAC9C;;OAEG;IACH,IAAI,EAAE,6BAA6B,CAAA;IAEnC;;OAEG;IACH,KAAK,EAAE,iBAAiB,CAAA;CACzB;AAED;;GAEG;AACH,qBAAa,2BAA4B,YAAW,+BAA+B;IACjF;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,6BAA6B,CAAS;IAEtD;;OAEG;IACH,QAAQ,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAA;IAEnC;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,iBAAiB,CAAA;gBAErB,IAAI,EAAE,+BAA+B;CAGlD;AAED;;GAEG;AACH,MAAM,WAAW,8BAA8B;IAC7C;;OAEG;IACH,IAAI,EAAE,4BAA4B,CAAA;CACnC;AAED;;GAEG;AACH,qBAAa,0BAA2B,YAAW,8BAA8B;IAC/E;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,4BAA4B,CAAS;gBAEzC,KAAK,EAAE,8BAA8B;CAClD;AAED;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACxC;;OAEG;IACH,IAAI,EAAE,uBAAuB,CAAA;IAE7B;;OAEG;IACH,UAAU,EAAE,UAAU,CAAA;IAEtB;;OAEG;IACH,6BAA6B,CAAC,EAAE,SAAS,CAAA;CAC1C;AAED;;GAEG;AACH,qBAAa,qBAAsB,YAAW,yBAAyB;IACrE;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,uBAAuB,CAAS;IAEhD;;OAEG;IACH,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAA;IAE/B;;OAEG;IACH,QAAQ,CAAC,6BAA6B,CAAC,EAAE,SAAS,CAAA;gBAEtC,IAAI,EAAE,yBAAyB;CAM5C;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC;;OAEG;IACH,IAAI,EAAE,oBAAoB,CAAA;IAE1B;;OAEG;IACH,KAAK,CAAC,EAAE,KAAK,CAAA;IAEb;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAA;IAEjB;;OAEG;IACH,KAAK,CAAC,EAAE,OAAO,CAAA;CAChB;AAED;;;GAGG;AACH,qBAAa,kBAAmB,YAAW,sBAAsB;IAC/D;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,oBAAoB,CAAS;IAE7C;;OAEG;IACH,QAAQ,CAAC,KAAK,CAAC,EAAE,KAAK,CAAA;IAEtB;;OAEG;IACH,QAAQ,CAAC,OAAO,CAAC,EAAE,OAAO,CAAA;IAE1B;;OAEG;IACH,QAAQ,CAAC,KAAK,CAAC,EAAE,OAAO,CAAA;gBAEZ,IAAI,EAAE,sBAAsB;CAWzC;AAED;;;GAGG;AACH,MAAM,MAAM,iBAAiB,GAAG,YAAY,CAAA;AAE5C;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B;;OAEG;IACH,IAAI,EAAE,cAAc,CAAA;IAEpB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAA;IAEZ;;OAEG;IACH,SAAS,EAAE,MAAM,CAAA;IAEjB;;;OAGG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAA;CAC5B;AAED;;;;;GAKG;AACH,MAAM,MAAM,iBAAiB,GAAG,SAAS,GAAG,iBAAiB,GAAG,qBAAqB,CAAA;AAErF;;;GAGG;AACH,MAAM,WAAW,SAAS;IACxB;;OAEG;IACH,IAAI,EAAE,WAAW,CAAA;IAEjB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAA;CACb;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC;;OAEG;IACH,IAAI,EAAE,mBAAmB,CAAA;IAEzB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAA;CACd;AAED;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC;;OAEG;IACH,IAAI,EAAE,uBAAuB,CAAA;IAE7B;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAA;IAEb;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAA;IAElB;;OAEG;IACH,eAAe,CAAC,EAAE,UAAU,CAAA;CAC7B;AAED;;;GAGG;AACH,MAAM,WAAW,KAAK;IACpB;;OAEG;IACH,WAAW,EAAE,MAAM,CAAA;IAEnB;;OAEG;IACH,YAAY,EAAE,MAAM,CAAA;IAEpB;;OAEG;IACH,WAAW,EAAE,MAAM,CAAA;IAEnB;;;OAGG;IACH,oBAAoB,CAAC,EAAE,MAAM,CAAA;IAE7B;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAA;CAC/B;AAED;;GAEG;AACH,MAAM,WAAW,OAAO;IACtB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAA;CAClB"}
1
+ {"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../../src/models/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAA;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AACjD,OAAO,KAAK,EAAE,QAAQ,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAA;AAE/E;;;;;;GAMG;AAEH;;;;;GAKG;AACH,MAAM,MAAM,gBAAgB,GACxB,0BAA0B,GAC1B,+BAA+B,GAC/B,+BAA+B,GAC/B,8BAA8B,GAC9B,yBAAyB,GACzB,sBAAsB,GACtB,uBAAuB,CAAA;AAa3B;;;;GAIG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,GAAG,KAAK,IAAI,gBAAgB,CAErF;AAED;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC;;OAEG;IACH,IAAI,EAAE,wBAAwB,CAAA;IAE9B;;OAEG;IACH,IAAI,EAAE,IAAI,CAAA;CACX;AAED;;GAEG;AACH,qBAAa,sBAAuB,YAAW,0BAA0B;IACvE;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,wBAAwB,CAAS;IAEjD;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAA;gBAEP,IAAI,EAAE,0BAA0B;CAG7C;AAED;;GAEG;AACH,MAAM,WAAW,+BAA+B;IAC9C;;OAEG;IACH,IAAI,EAAE,6BAA6B,CAAA;IAEnC;;;OAGG;IACH,KAAK,CAAC,EAAE,iBAAiB,CAAA;CAC1B;AAED;;GAEG;AACH,qBAAa,2BAA4B,YAAW,+BAA+B;IACjF;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,6BAA6B,CAAS;IAEtD;;;OAGG;IACH,QAAQ,CAAC,KAAK,CAAC,EAAE,iBAAiB,CAAA;gBAEtB,IAAI,EAAE,+BAA+B;CAKlD;AAED;;GAEG;AACH,MAAM,WAAW,+BAA+B;IAC9C;;OAEG;IACH,IAAI,EAAE,6BAA6B,CAAA;IAEnC;;OAEG;IACH,KAAK,EAAE,iBAAiB,CAAA;CACzB;AAED;;GAEG;AACH,qBAAa,2BAA4B,YAAW,+BAA+B;IACjF;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,6BAA6B,CAAS;IAEtD;;OAEG;IACH,QAAQ,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAA;IAEnC;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,iBAAiB,CAAA;gBAErB,IAAI,EAAE,+BAA+B;CAGlD;AAED;;GAEG;AACH,MAAM,WAAW,8BAA8B;IAC7C;;OAEG;IACH,IAAI,EAAE,4BAA4B,CAAA;CACnC;AAED;;GAEG;AACH,qBAAa,0BAA2B,YAAW,8BAA8B;IAC/E;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,4BAA4B,CAAS;gBAEzC,KAAK,EAAE,8BAA8B;CAClD;AAED;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACxC;;OAEG;IACH,IAAI,EAAE,uBAAuB,CAAA;IAE7B;;OAEG;IACH,UAAU,EAAE,UAAU,CAAA;IAEtB;;OAEG;IACH,6BAA6B,CAAC,EAAE,SAAS,CAAA;CAC1C;AAED;;GAEG;AACH,qBAAa,qBAAsB,YAAW,yBAAyB;IACrE;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,uBAAuB,CAAS;IAEhD;;OAEG;IACH,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAA;IAE/B;;OAEG;IACH,QAAQ,CAAC,6BAA6B,CAAC,EAAE,SAAS,CAAA;gBAEtC,IAAI,EAAE,yBAAyB;CAM5C;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC;;OAEG;IACH,IAAI,EAAE,oBAAoB,CAAA;IAE1B;;OAEG;IACH,KAAK,CAAC,EAAE,KAAK,CAAA;IAEb;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAA;IAEjB;;OAEG;IACH,KAAK,CAAC,EAAE,OAAO,CAAA;CAChB;AAED;;;GAGG;AACH,qBAAa,kBAAmB,YAAW,sBAAsB;IAC/D;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,oBAAoB,CAAS;IAE7C;;OAEG;IACH,QAAQ,CAAC,KAAK,CAAC,EAAE,KAAK,CAAA;IAEtB;;OAEG;IACH,QAAQ,CAAC,OAAO,CAAC,EAAE,OAAO,CAAA;IAE1B;;OAEG;IACH,QAAQ,CAAC,KAAK,CAAC,EAAE,OAAO,CAAA;gBAEZ,IAAI,EAAE,sBAAsB;CAWzC;AAED;;;;GAIG;AACH,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,cAAc,EAAE,MAAM,CAAA;CACvB;AAED;;;GAGG;AACH,MAAM,WAAW,mBAAmB;IAClC;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAA;IAExB;;OAEG;IACH,cAAc,EAAE,MAAM,CAAA;CACvB;AAED;;;GAGG;AACH,MAAM,WAAW,uBAAuB;IACtC;;OAEG;IACH,IAAI,EAAE,qBAAqB,CAAA;IAE3B;;OAEG;IACH,cAAc,CAAC,EAAE,kBAAkB,CAAA;IAEnC;;OAEG;IACH,eAAe,CAAC,EAAE,mBAAmB,CAAA;CACtC;AAED;;GAEG;AACH,qBAAa,mBAAoB,YAAW,uBAAuB;IACjE;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAG,qBAAqB,CAAS;IAE9C;;OAEG;IACH,QAAQ,CAAC,cAAc,CAAC,EAAE,kBAAkB,CAAA;IAE5C;;OAEG;IACH,QAAQ,CAAC,eAAe,CAAC,EAAE,mBAAmB,CAAA;gBAElC,IAAI,EAAE,uBAAuB;CAQ1C;AAED;;;GAGG;AACH,MAAM,MAAM,iBAAiB,GAAG,YAAY,CAAA;AAE5C;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B;;OAEG;IACH,IAAI,EAAE,cAAc,CAAA;IAEpB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAA;IAEZ;;OAEG;IACH,SAAS,EAAE,MAAM,CAAA;IAEjB;;;OAGG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAA;CAC5B;AAED;;;;;GAKG;AACH,MAAM,MAAM,iBAAiB,GAAG,SAAS,GAAG,iBAAiB,GAAG,qBAAqB,GAAG,cAAc,CAAA;AAEtG;;;GAGG;AACH,MAAM,WAAW,SAAS;IACxB;;OAEG;IACH,IAAI,EAAE,WAAW,CAAA;IAEjB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAA;CACb;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC;;OAEG;IACH,IAAI,EAAE,mBAAmB,CAAA;IAEzB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAA;CACd;AAED;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC;;OAEG;IACH,IAAI,EAAE,uBAAuB,CAAA;IAE7B;;OAEG;IACH,IAAI,CAAC,EAAE,MAAM,CAAA;IAEb;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAA;IAElB;;OAEG;IACH,eAAe,CAAC,EAAE,UAAU,CAAA;CAC7B;AAED;;;GAGG;AACH,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,IAAI,EAAE,gBAAgB,CAAA;IAEtB;;OAEG;IACH,SAAS,EAAE,QAAQ,EAAE,CAAA;IAErB;;OAEG;IACH,OAAO,EAAE,wBAAwB,EAAE,CAAA;CACpC;AAED;;;GAGG;AACH,MAAM,WAAW,KAAK;IACpB;;OAEG;IACH,WAAW,EAAE,MAAM,CAAA;IAEnB;;OAEG;IACH,YAAY,EAAE,MAAM,CAAA;IAEpB;;OAEG;IACH,WAAW,EAAE,MAAM,CAAA;IAEnB;;;OAGG;IACH,oBAAoB,CAAC,EAAE,MAAM,CAAA;IAE7B;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAA;CAC/B;AAED;;GAEG;AACH,MAAM,WAAW,OAAO;IACtB;;OAEG;IACH,SAAS,EAAE,MAAM,CAAA;CAClB"}
@@ -6,6 +6,7 @@ const modelStreamEventTypes = new Set([
6
6
  'modelContentBlockStopEvent',
7
7
  'modelMessageStopEvent',
8
8
  'modelMetadataEvent',
9
+ 'modelRedactionEvent',
9
10
  ]);
10
11
  /**
11
12
  * Type guard to check if an event with a type discriminator is a ModelStreamEvent.
@@ -136,4 +137,29 @@ export class ModelMetadataEvent {
136
137
  }
137
138
  }
138
139
  }
140
+ /**
141
+ * Event emitted when guardrails block content and trigger redaction.
142
+ */
143
+ export class ModelRedactionEvent {
144
+ /**
145
+ * Discriminator for redact events.
146
+ */
147
+ type = 'modelRedactionEvent';
148
+ /**
149
+ * Input redaction information (when input is blocked).
150
+ */
151
+ inputRedaction;
152
+ /**
153
+ * Output redaction information (when output is blocked).
154
+ */
155
+ outputRedaction;
156
+ constructor(data) {
157
+ if (data.inputRedaction !== undefined) {
158
+ this.inputRedaction = data.inputRedaction;
159
+ }
160
+ if (data.outputRedaction !== undefined) {
161
+ this.outputRedaction = data.outputRedaction;
162
+ }
163
+ }
164
+ }
139
165
  //# sourceMappingURL=streaming.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../../src/models/streaming.ts"],"names":[],"mappings":"AAyBA,uDAAuD;AACvD,MAAM,qBAAqB,GAAwB,IAAI,GAAG,CAA2B;IACnF,wBAAwB;IACxB,6BAA6B;IAC7B,6BAA6B;IAC7B,4BAA4B;IAC5B,uBAAuB;IACvB,oBAAoB;CACrB,CAAC,CAAA;AAEF;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAuB;IACxD,OAAO,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;AAC9C,CAAC;AAiBD;;GAEG;AACH,MAAM,OAAO,sBAAsB;IACjC;;OAEG;IACM,IAAI,GAAG,wBAAiC,CAAA;IAEjD;;OAEG;IACM,IAAI,CAAM;IAEnB,YAAY,IAAgC;QAC1C,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;IACvB,CAAC;CACF;AAkBD;;GAEG;AACH,MAAM,OAAO,2BAA2B;IACtC;;OAEG;IACM,IAAI,GAAG,6BAAsC,CAAA;IAEtD;;;OAGG;IACM,KAAK,CAAoB;IAElC,YAAY,IAAqC;QAC/C,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACzB,CAAC;IACH,CAAC;CACF;AAiBD;;GAEG;AACH,MAAM,OAAO,2BAA2B;IACtC;;OAEG;IACM,IAAI,GAAG,6BAAsC,CAAA;IAEtD;;OAEG;IACM,iBAAiB,CAAS;IAEnC;;OAEG;IACM,KAAK,CAAmB;IAEjC,YAAY,IAAqC;QAC/C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;IACzB,CAAC;CACF;AAYD;;GAEG;AACH,MAAM,OAAO,0BAA0B;IACrC;;OAEG;IACM,IAAI,GAAG,4BAAqC,CAAA;IAErD,YAAY,KAAqC,IAAG,CAAC;CACtD;AAsBD;;GAEG;AACH,MAAM,OAAO,qBAAqB;IAChC;;OAEG;IACM,IAAI,GAAG,uBAAgC,CAAA;IAEhD;;OAEG;IACM,UAAU,CAAY;IAE/B;;OAEG;IACM,6BAA6B,CAAY;IAElD,YAAY,IAA+B;QACzC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAA;QACjC,IAAI,IAAI,CAAC,6BAA6B,KAAK,SAAS,EAAE,CAAC;YACrD,IAAI,CAAC,6BAA6B,GAAG,IAAI,CAAC,6BAA6B,CAAA;QACzE,CAAC;IACH,CAAC;CACF;AA2BD;;;GAGG;AACH,MAAM,OAAO,kBAAkB;IAC7B;;OAEG;IACM,IAAI,GAAG,oBAA6B,CAAA;IAE7C;;OAEG;IACM,KAAK,CAAQ;IAEtB;;OAEG;IACM,OAAO,CAAU;IAE1B;;OAEG;IACM,KAAK,CAAU;IAExB,YAAY,IAA4B;QACtC,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACzB,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAA;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACzB,CAAC;IACH,CAAC;CACF"}
1
+ {"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../../src/models/streaming.ts"],"names":[],"mappings":"AA2BA,uDAAuD;AACvD,MAAM,qBAAqB,GAAwB,IAAI,GAAG,CAA2B;IACnF,wBAAwB;IACxB,6BAA6B;IAC7B,6BAA6B;IAC7B,4BAA4B;IAC5B,uBAAuB;IACvB,oBAAoB;IACpB,qBAAqB;CACtB,CAAC,CAAA;AAEF;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAuB;IACxD,OAAO,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;AAC9C,CAAC;AAiBD;;GAEG;AACH,MAAM,OAAO,sBAAsB;IACjC;;OAEG;IACM,IAAI,GAAG,wBAAiC,CAAA;IAEjD;;OAEG;IACM,IAAI,CAAM;IAEnB,YAAY,IAAgC;QAC1C,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;IACvB,CAAC;CACF;AAkBD;;GAEG;AACH,MAAM,OAAO,2BAA2B;IACtC;;OAEG;IACM,IAAI,GAAG,6BAAsC,CAAA;IAEtD;;;OAGG;IACM,KAAK,CAAoB;IAElC,YAAY,IAAqC;QAC/C,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACzB,CAAC;IACH,CAAC;CACF;AAiBD;;GAEG;AACH,MAAM,OAAO,2BAA2B;IACtC;;OAEG;IACM,IAAI,GAAG,6BAAsC,CAAA;IAEtD;;OAEG;IACM,iBAAiB,CAAS;IAEnC;;OAEG;IACM,KAAK,CAAmB;IAEjC,YAAY,IAAqC;QAC/C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;IACzB,CAAC;CACF;AAYD;;GAEG;AACH,MAAM,OAAO,0BAA0B;IACrC;;OAEG;IACM,IAAI,GAAG,4BAAqC,CAAA;IAErD,YAAY,KAAqC,IAAG,CAAC;CACtD;AAsBD;;GAEG;AACH,MAAM,OAAO,qBAAqB;IAChC;;OAEG;IACM,IAAI,GAAG,uBAAgC,CAAA;IAEhD;;OAEG;IACM,UAAU,CAAY;IAE/B;;OAEG;IACM,6BAA6B,CAAY;IAElD,YAAY,IAA+B;QACzC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAA;QACjC,IAAI,IAAI,CAAC,6BAA6B,KAAK,SAAS,EAAE,CAAC;YACrD,IAAI,CAAC,6BAA6B,GAAG,IAAI,CAAC,6BAA6B,CAAA;QACzE,CAAC;IACH,CAAC;CACF;AA2BD;;;GAGG;AACH,MAAM,OAAO,kBAAkB;IAC7B;;OAEG;IACM,IAAI,GAAG,oBAA6B,CAAA;IAE7C;;OAEG;IACM,KAAK,CAAQ;IAEtB;;OAEG;IACM,OAAO,CAAU;IAE1B;;OAEG;IACM,KAAK,CAAU;IAExB,YAAY,IAA4B;QACtC,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACzB,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAA;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACzB,CAAC;IACH,CAAC;CACF;AAoDD;;GAEG;AACH,MAAM,OAAO,mBAAmB;IAC9B;;OAEG;IACM,IAAI,GAAG,qBAA8B,CAAA;IAE9C;;OAEG;IACM,cAAc,CAAqB;IAE5C;;OAEG;IACM,eAAe,CAAsB;IAE9C,YAAY,IAA6B;QACvC,IAAI,IAAI,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;YACtC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAA;QAC3C,CAAC;QACD,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YACvC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;QAC7C,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=events.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"events.test.d.ts","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/events.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,205 @@
1
+ import { describe, expect, it } from 'vitest';
2
+ import { MultiAgentInitializedEvent, BeforeMultiAgentInvocationEvent, AfterMultiAgentInvocationEvent, BeforeNodeCallEvent, AfterNodeCallEvent, NodeStreamUpdateEvent, NodeResultEvent, NodeCancelEvent, MultiAgentHandoffEvent, MultiAgentResultEvent, } from '../events.js';
3
+ import { MultiAgentResult, MultiAgentState, NodeResult, Status } from '../state.js';
4
+ const mockOrchestrator = {
5
+ id: 'test-orchestrator',
6
+ invoke: async () => new MultiAgentResult({ results: [], duration: 0 }),
7
+ // eslint-disable-next-line require-yield
8
+ async *stream() {
9
+ return new MultiAgentResult({ results: [], duration: 0 });
10
+ },
11
+ };
12
+ describe('MultiAgentInitializedEvent', () => {
13
+ it('creates instance with correct properties', () => {
14
+ const event = new MultiAgentInitializedEvent({ orchestrator: mockOrchestrator });
15
+ expect(event).toEqual({
16
+ type: 'multiAgentInitializedEvent',
17
+ orchestrator: mockOrchestrator,
18
+ });
19
+ // @ts-expect-error verifying that property is readonly
20
+ event.orchestrator = mockOrchestrator;
21
+ });
22
+ it('returns false for _shouldReverseCallbacks', () => {
23
+ const event = new MultiAgentInitializedEvent({ orchestrator: mockOrchestrator });
24
+ expect(event._shouldReverseCallbacks()).toBe(false);
25
+ });
26
+ });
27
+ describe('BeforeMultiAgentInvocationEvent', () => {
28
+ it('creates instance with correct properties', () => {
29
+ const state = new MultiAgentState();
30
+ const event = new BeforeMultiAgentInvocationEvent({ orchestrator: mockOrchestrator, state });
31
+ expect(event).toEqual({
32
+ type: 'beforeMultiAgentInvocationEvent',
33
+ orchestrator: mockOrchestrator,
34
+ state,
35
+ });
36
+ // @ts-expect-error verifying that property is readonly
37
+ event.orchestrator = mockOrchestrator;
38
+ // @ts-expect-error verifying that property is readonly
39
+ event.state = state;
40
+ });
41
+ it('returns false for _shouldReverseCallbacks', () => {
42
+ const state = new MultiAgentState();
43
+ const event = new BeforeMultiAgentInvocationEvent({ orchestrator: mockOrchestrator, state });
44
+ expect(event._shouldReverseCallbacks()).toBe(false);
45
+ });
46
+ });
47
+ describe('AfterMultiAgentInvocationEvent', () => {
48
+ it('creates instance with correct properties', () => {
49
+ const state = new MultiAgentState();
50
+ const event = new AfterMultiAgentInvocationEvent({ orchestrator: mockOrchestrator, state });
51
+ expect(event).toEqual({
52
+ type: 'afterMultiAgentInvocationEvent',
53
+ orchestrator: mockOrchestrator,
54
+ state,
55
+ });
56
+ // @ts-expect-error verifying that property is readonly
57
+ event.orchestrator = mockOrchestrator;
58
+ // @ts-expect-error verifying that property is readonly
59
+ event.state = state;
60
+ });
61
+ it('returns true for _shouldReverseCallbacks', () => {
62
+ const state = new MultiAgentState();
63
+ const event = new AfterMultiAgentInvocationEvent({ orchestrator: mockOrchestrator, state });
64
+ expect(event._shouldReverseCallbacks()).toBe(true);
65
+ });
66
+ });
67
+ describe('BeforeNodeCallEvent', () => {
68
+ it('creates instance with correct properties', () => {
69
+ const state = new MultiAgentState();
70
+ const event = new BeforeNodeCallEvent({ orchestrator: mockOrchestrator, state, nodeId: 'node-1' });
71
+ expect(event).toEqual({
72
+ type: 'beforeNodeCallEvent',
73
+ orchestrator: mockOrchestrator,
74
+ state,
75
+ nodeId: 'node-1',
76
+ cancel: false,
77
+ });
78
+ // @ts-expect-error verifying that property is readonly
79
+ event.orchestrator = mockOrchestrator;
80
+ // @ts-expect-error verifying that property is readonly
81
+ event.state = state;
82
+ // @ts-expect-error verifying that property is readonly
83
+ event.nodeId = 'node-1';
84
+ });
85
+ it('returns false for _shouldReverseCallbacks', () => {
86
+ const state = new MultiAgentState();
87
+ const event = new BeforeNodeCallEvent({ orchestrator: mockOrchestrator, state, nodeId: 'node-1' });
88
+ expect(event._shouldReverseCallbacks()).toBe(false);
89
+ });
90
+ it('allows cancel to be set to true', () => {
91
+ const state = new MultiAgentState();
92
+ const event = new BeforeNodeCallEvent({ orchestrator: mockOrchestrator, state, nodeId: 'node-1' });
93
+ expect(event.cancel).toBe(false);
94
+ event.cancel = true;
95
+ expect(event.cancel).toBe(true);
96
+ });
97
+ it('allows cancel to be set to a string message', () => {
98
+ const state = new MultiAgentState();
99
+ const event = new BeforeNodeCallEvent({ orchestrator: mockOrchestrator, state, nodeId: 'node-1' });
100
+ event.cancel = 'node is not ready';
101
+ expect(event.cancel).toBe('node is not ready');
102
+ });
103
+ });
104
+ describe('AfterNodeCallEvent', () => {
105
+ it('creates instance with correct properties', () => {
106
+ const state = new MultiAgentState();
107
+ const error = new Error('node failed');
108
+ const event = new AfterNodeCallEvent({ orchestrator: mockOrchestrator, state, nodeId: 'node-1', error });
109
+ expect(event).toEqual({
110
+ type: 'afterNodeCallEvent',
111
+ orchestrator: mockOrchestrator,
112
+ state,
113
+ nodeId: 'node-1',
114
+ error,
115
+ });
116
+ // @ts-expect-error verifying that property is readonly
117
+ event.orchestrator = mockOrchestrator;
118
+ // @ts-expect-error verifying that property is readonly
119
+ event.state = state;
120
+ // @ts-expect-error verifying that property is readonly
121
+ event.nodeId = 'node-1';
122
+ });
123
+ it('returns true for _shouldReverseCallbacks', () => {
124
+ const state = new MultiAgentState();
125
+ const event = new AfterNodeCallEvent({ orchestrator: mockOrchestrator, state, nodeId: 'node-1' });
126
+ expect(event._shouldReverseCallbacks()).toBe(true);
127
+ });
128
+ });
129
+ describe('NodeStreamUpdateEvent', () => {
130
+ it('creates instance with correct properties', () => {
131
+ const innerEvent = { type: 'beforeInvocationEvent' };
132
+ const event = new NodeStreamUpdateEvent({ nodeId: 'node-1', nodeType: 'agentNode', event: innerEvent });
133
+ expect(event).toEqual({
134
+ type: 'nodeStreamUpdateEvent',
135
+ nodeId: 'node-1',
136
+ nodeType: 'agentNode',
137
+ event: innerEvent,
138
+ });
139
+ // @ts-expect-error verifying that property is readonly
140
+ event.nodeId = 'node-1';
141
+ // @ts-expect-error verifying that property is readonly
142
+ event.nodeType = 'agentNode';
143
+ // @ts-expect-error verifying that property is readonly
144
+ event.event = innerEvent;
145
+ });
146
+ });
147
+ describe('NodeResultEvent', () => {
148
+ it('creates instance with correct properties', () => {
149
+ const result = new NodeResult({ nodeId: 'node-1', status: Status.COMPLETED, duration: 100 });
150
+ const event = new NodeResultEvent({ nodeId: 'node-1', nodeType: 'agentNode', result });
151
+ expect(event).toEqual({
152
+ type: 'nodeResultEvent',
153
+ nodeId: 'node-1',
154
+ nodeType: 'agentNode',
155
+ result,
156
+ });
157
+ // @ts-expect-error verifying that property is readonly
158
+ event.nodeId = 'node-1';
159
+ // @ts-expect-error verifying that property is readonly
160
+ event.nodeType = 'agentNode';
161
+ // @ts-expect-error verifying that property is readonly
162
+ event.result = result;
163
+ });
164
+ });
165
+ describe('NodeCancelEvent', () => {
166
+ it('creates instance with correct properties', () => {
167
+ const event = new NodeCancelEvent({ nodeId: 'node-1', message: 'cancelled by hook' });
168
+ expect(event).toEqual({
169
+ type: 'nodeCancelEvent',
170
+ nodeId: 'node-1',
171
+ message: 'cancelled by hook',
172
+ });
173
+ // @ts-expect-error verifying that property is readonly
174
+ event.nodeId = 'node-1';
175
+ // @ts-expect-error verifying that property is readonly
176
+ event.message = 'cancelled by hook';
177
+ });
178
+ });
179
+ describe('MultiAgentHandoffEvent', () => {
180
+ it('creates instance with correct properties', () => {
181
+ const event = new MultiAgentHandoffEvent({ source: 'node-a', targets: ['node-b', 'node-c'] });
182
+ expect(event).toEqual({
183
+ type: 'multiAgentHandoffEvent',
184
+ source: 'node-a',
185
+ targets: ['node-b', 'node-c'],
186
+ });
187
+ // @ts-expect-error verifying that property is readonly
188
+ event.source = 'node-a';
189
+ // @ts-expect-error verifying that property is readonly
190
+ event.targets = [];
191
+ });
192
+ });
193
+ describe('MultiAgentResultEvent', () => {
194
+ it('creates instance with correct properties', () => {
195
+ const result = new MultiAgentResult({ results: [], duration: 0 });
196
+ const event = new MultiAgentResultEvent({ result });
197
+ expect(event).toEqual({
198
+ type: 'multiAgentResultEvent',
199
+ result,
200
+ });
201
+ // @ts-expect-error verifying that property is readonly
202
+ event.result = result;
203
+ });
204
+ });
205
+ //# sourceMappingURL=events.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"events.test.js","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/events.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EACL,0BAA0B,EAC1B,+BAA+B,EAC/B,8BAA8B,EAC9B,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,eAAe,EACf,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,cAAc,CAAA;AACrB,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AAInF,MAAM,gBAAgB,GAAmB;IACvC,EAAE,EAAE,mBAAmB;IACvB,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,gBAAgB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;IACtE,yCAAyC;IACzC,KAAK,CAAC,CAAC,MAAM;QACX,OAAO,IAAI,gBAAgB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAA;IAC3D,CAAC;CACF,CAAA;AAED,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,0BAA0B,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC,CAAA;QAEhF,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,4BAA4B;YAClC,YAAY,EAAE,gBAAgB;SAC/B,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,YAAY,GAAG,gBAAgB,CAAA;IACvC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,KAAK,GAAG,IAAI,0BAA0B,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC,CAAA;QAChF,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACrD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,+BAA+B,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAA;QAE5F,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,iCAAiC;YACvC,YAAY,EAAE,gBAAgB;YAC9B,KAAK;SACN,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,YAAY,GAAG,gBAAgB,CAAA;QACrC,uDAAuD;QACvD,KAAK,CAAC,KAAK,GAAG,KAAK,CAAA;IACrB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,+BAA+B,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAA;QAC5F,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACrD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,8BAA8B,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAA;QAE3F,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,gCAAgC;YACtC,YAAY,EAAE,gBAAgB;YAC9B,KAAK;SACN,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,YAAY,GAAG,gBAAgB,CAAA;QACrC,uDAAuD;QACvD,KAAK,CAAC,KAAK,GAAG,KAAK,CAAA;IACrB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,8BAA8B,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAA;QAC3F,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACpD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAA;QAElG,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,qBAAqB;YAC3B,YAAY,EAAE,gBAAgB;YAC9B,KAAK;YACL,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,KAAK;SACd,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,YAAY,GAAG,gBAAgB,CAAA;QACrC,uDAAuD;QACvD,KAAK,CAAC,KAAK,GAAG,KAAK,CAAA;QACnB,uDAAuD;QACvD,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAA;IACzB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAA;QAClG,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACrD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAA;QAElG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAChC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAA;QACnB,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAA;QAElG,KAAK,CAAC,MAAM,GAAG,mBAAmB,CAAA;QAClC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;IAChD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,aAAa,CAAC,CAAA;QACtC,MAAM,KAAK,GAAG,IAAI,kBAAkB,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAA;QAExG,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,oBAAoB;YAC1B,YAAY,EAAE,gBAAgB;YAC9B,KAAK;YACL,MAAM,EAAE,QAAQ;YAChB,KAAK;SACN,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,YAAY,GAAG,gBAAgB,CAAA;QACrC,uDAAuD;QACvD,KAAK,CAAC,KAAK,GAAG,KAAK,CAAA;QACnB,uDAAuD;QACvD,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAA;IACzB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAA;QACnC,MAAM,KAAK,GAAG,IAAI,kBAAkB,CAAC,EAAE,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAA;QACjG,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACpD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,UAAU,GAAG,EAAE,IAAI,EAAE,uBAAuB,EAAsB,CAAA;QACxE,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAA;QAEvG,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,uBAAuB;YAC7B,MAAM,EAAE,QAAQ;YAChB,QAAQ,EAAE,WAAW;YACrB,KAAK,EAAE,UAAU;SAClB,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAA;QACvB,uDAAuD;QACvD,KAAK,CAAC,QAAQ,GAAG,WAAW,CAAA;QAC5B,uDAAuD;QACvD,KAAK,CAAC,KAAK,GAAG,UAAU,CAAA;IAC1B,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAA;QAC5F,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAA;QAEtF,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,iBAAiB;YACvB,MAAM,EAAE,QAAQ;YAChB,QAAQ,EAAE,WAAW;YACrB,MAAM;SACP,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAA;QACvB,uDAAuD;QACvD,KAAK,CAAC,QAAQ,GAAG,WAAW,CAAA;QAC5B,uDAAuD;QACvD,KAAK,CAAC,MAAM,GAAG,MAAM,CAAA;IACvB,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAA;QAErF,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,iBAAiB;YACvB,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,mBAAmB;SAC7B,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAA;QACvB,uDAAuD;QACvD,KAAK,CAAC,OAAO,GAAG,mBAAmB,CAAA;IACrC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,sBAAsB,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAA;QAE7F,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,wBAAwB;YAC9B,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;SAC9B,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAA;QACvB,uDAAuD;QACvD,KAAK,CAAC,OAAO,GAAG,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAA;QACjE,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,EAAE,MAAM,EAAE,CAAC,CAAA;QAEnD,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,IAAI,EAAE,uBAAuB;YAC7B,MAAM;SACP,CAAC,CAAA;QACF,uDAAuD;QACvD,KAAK,CAAC,MAAM,GAAG,MAAM,CAAA;IACvB,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=graph.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"graph.test.d.ts","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/graph.test.ts"],"names":[],"mappings":""}