@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
@@ -0,0 +1,438 @@
1
+ import { describe, expect, it, vi } from 'vitest';
2
+ import { Agent } from '../../agent/agent.js';
3
+ import { MockMessageModel } from '../../__fixtures__/mock-message-model.js';
4
+ import { collectGenerator } from '../../__fixtures__/model-test-helpers.js';
5
+ import { AfterNodeCallEvent, BeforeNodeCallEvent, MultiAgentInitializedEvent } from '../events.js';
6
+ import { TextBlock } from '../../types/messages.js';
7
+ import { Status } from '../state.js';
8
+ import { AgentNode, MultiAgentNode } from '../nodes.js';
9
+ import { Graph } from '../graph.js';
10
+ function makeAgent(id, text = 'reply') {
11
+ const model = new MockMessageModel().addTurn(new TextBlock(text));
12
+ return new Agent({ model, printer: false, agentId: id });
13
+ }
14
+ describe('Graph', () => {
15
+ describe('constructor', () => {
16
+ it('defaults id to "graph"', () => {
17
+ const graph = new Graph({
18
+ nodes: [makeAgent('a')],
19
+ edges: [],
20
+ });
21
+ expect(graph.id).toBe('graph');
22
+ });
23
+ it('accepts a custom id', () => {
24
+ const graph = new Graph({
25
+ nodes: [makeAgent('a')],
26
+ edges: [],
27
+ id: 'my-graph',
28
+ });
29
+ expect(graph.id).toBe('my-graph');
30
+ });
31
+ it('accepts agent node options', () => {
32
+ const graph = new Graph({
33
+ nodes: [{ type: 'agent', agent: makeAgent('a') }],
34
+ edges: [],
35
+ });
36
+ expect(graph.nodes.get('a')).toBeInstanceOf(AgentNode);
37
+ });
38
+ it('accepts multiAgent node options', () => {
39
+ const inner = new Graph({
40
+ id: 'inner',
41
+ nodes: [makeAgent('x')],
42
+ edges: [],
43
+ });
44
+ const graph = new Graph({
45
+ nodes: [{ type: 'multiAgent', orchestrator: inner }],
46
+ edges: [],
47
+ });
48
+ expect(graph.nodes.get('inner')).toBeInstanceOf(MultiAgentNode);
49
+ });
50
+ it('accepts pre-built Node instances', () => {
51
+ const node = new AgentNode({ agent: makeAgent('a') });
52
+ const graph = new Graph({
53
+ nodes: [node],
54
+ edges: [],
55
+ });
56
+ expect(graph.nodes.get('a')).toBe(node);
57
+ });
58
+ it('accepts edge options', () => {
59
+ const graph = new Graph({
60
+ nodes: [makeAgent('a'), makeAgent('b')],
61
+ edges: [{ source: 'a', target: 'b' }],
62
+ });
63
+ expect(graph.edges).toHaveLength(1);
64
+ expect(graph.edges[0]).toEqual(expect.objectContaining({
65
+ source: expect.objectContaining({ id: 'a' }),
66
+ target: expect.objectContaining({ id: 'b' }),
67
+ }));
68
+ });
69
+ it('throws on duplicate node IDs', () => {
70
+ const agent = makeAgent('a');
71
+ expect(() => new Graph({
72
+ nodes: [agent, agent],
73
+ edges: [],
74
+ })).toThrow('node_id=<a> | duplicate node id');
75
+ });
76
+ it('throws on edge referencing unknown source node', () => {
77
+ expect(() => new Graph({
78
+ nodes: [makeAgent('a')],
79
+ edges: [['missing', 'a']],
80
+ })).toThrow('source=<missing> | edge references unknown source node');
81
+ });
82
+ it('throws on edge referencing unknown target node', () => {
83
+ expect(() => new Graph({
84
+ nodes: [makeAgent('a')],
85
+ edges: [['a', 'missing']],
86
+ })).toThrow('target=<missing> | edge references unknown target node');
87
+ });
88
+ it('throws when graph has no source nodes', () => {
89
+ expect(() => new Graph({
90
+ nodes: [makeAgent('a'), makeAgent('b')],
91
+ edges: [
92
+ ['a', 'b'],
93
+ ['b', 'a'],
94
+ ],
95
+ })).toThrow('graph has no source nodes');
96
+ });
97
+ it('throws on unreachable nodes', () => {
98
+ expect(() => new Graph({
99
+ nodes: [makeAgent('a'), makeAgent('b'), makeAgent('island1'), makeAgent('island2')],
100
+ edges: [
101
+ ['a', 'b'],
102
+ ['island1', 'island2'],
103
+ ['island2', 'island1'],
104
+ ],
105
+ })).toThrow('node_id=<island1> | unreachable from any source node');
106
+ });
107
+ it('throws when explicit source references unknown node', () => {
108
+ expect(() => new Graph({
109
+ nodes: [makeAgent('a')],
110
+ edges: [],
111
+ sources: ['missing'],
112
+ })).toThrow('source=<missing> | source references unknown node');
113
+ });
114
+ it('throws when maxSteps < 1', () => {
115
+ expect(() => new Graph({
116
+ nodes: [makeAgent('a')],
117
+ edges: [],
118
+ maxSteps: 0,
119
+ })).toThrow('max_steps=<0> | must be at least 1');
120
+ });
121
+ it('throws when maxConcurrency < 1', () => {
122
+ expect(() => new Graph({
123
+ nodes: [makeAgent('a')],
124
+ edges: [],
125
+ maxConcurrency: 0,
126
+ })).toThrow('max_concurrency=<0> | must be at least 1');
127
+ });
128
+ });
129
+ describe('invoke', () => {
130
+ it('executes linear graph (A -> B -> C) in order', async () => {
131
+ const graph = new Graph({
132
+ nodes: [makeAgent('a', 'a-reply'), makeAgent('b', 'b-reply'), makeAgent('c', 'c-reply')],
133
+ edges: [
134
+ ['a', 'b'],
135
+ ['b', 'c'],
136
+ ],
137
+ });
138
+ const result = await graph.invoke('start');
139
+ expect(result).toEqual(expect.objectContaining({
140
+ status: Status.COMPLETED,
141
+ content: expect.arrayContaining([expect.objectContaining({ type: 'textBlock', text: 'c-reply' })]),
142
+ duration: expect.any(Number),
143
+ }));
144
+ expect(result.results.map((r) => r.nodeId)).toStrictEqual(['a', 'b', 'c']);
145
+ });
146
+ it('executes parallel graph (A -> B, A -> C) with B and C after A', async () => {
147
+ const graph = new Graph({
148
+ nodes: [makeAgent('a', 'a-reply'), makeAgent('b', 'b-reply'), makeAgent('c', 'c-reply')],
149
+ edges: [
150
+ ['a', 'b'],
151
+ ['a', 'c'],
152
+ ],
153
+ });
154
+ const result = await graph.invoke('start');
155
+ expect(result).toEqual(expect.objectContaining({
156
+ status: Status.COMPLETED,
157
+ content: expect.arrayContaining([
158
+ expect.objectContaining({ type: 'textBlock', text: 'b-reply' }),
159
+ expect.objectContaining({ type: 'textBlock', text: 'c-reply' }),
160
+ ]),
161
+ duration: expect.any(Number),
162
+ }));
163
+ expect(result.results.map((r) => r.nodeId).sort()).toStrictEqual(['a', 'b', 'c']);
164
+ });
165
+ it('waits for all dependencies before executing join node (A -> C, B -> C)', async () => {
166
+ const graph = new Graph({
167
+ nodes: [makeAgent('a', 'a-reply'), makeAgent('b', 'b-reply'), makeAgent('c', 'c-reply')],
168
+ edges: [
169
+ ['a', 'c'],
170
+ ['b', 'c'],
171
+ ],
172
+ maxConcurrency: 1,
173
+ });
174
+ const result = await graph.invoke('start');
175
+ expect(result).toEqual(expect.objectContaining({
176
+ status: Status.COMPLETED,
177
+ content: expect.arrayContaining([expect.objectContaining({ type: 'textBlock', text: 'c-reply' })]),
178
+ duration: expect.any(Number),
179
+ }));
180
+ expect(result.results).toHaveLength(3);
181
+ });
182
+ it('executes nested graph through MultiAgentNode', async () => {
183
+ const inner = new Graph({
184
+ id: 'inner',
185
+ nodes: [makeAgent('x', 'inner-reply')],
186
+ edges: [],
187
+ });
188
+ const graph = new Graph({
189
+ nodes: [makeAgent('a', 'a-reply'), inner],
190
+ edges: [['a', 'inner']],
191
+ });
192
+ const result = await graph.invoke('start');
193
+ expect(result).toEqual(expect.objectContaining({
194
+ status: Status.COMPLETED,
195
+ content: expect.arrayContaining([expect.objectContaining({ type: 'textBlock', text: 'inner-reply' })]),
196
+ duration: expect.any(Number),
197
+ }));
198
+ expect(result.results.map((r) => r.nodeId)).toStrictEqual(['a', 'inner']);
199
+ });
200
+ it('uses explicit sources instead of auto-detection', async () => {
201
+ const graph = new Graph({
202
+ nodes: [makeAgent('a'), makeAgent('b')],
203
+ edges: [['a', 'b'], { source: 'b', target: 'a', handler: () => false }],
204
+ sources: ['a'],
205
+ });
206
+ const result = await graph.invoke('go');
207
+ expect(result).toEqual(expect.objectContaining({
208
+ status: Status.COMPLETED,
209
+ duration: expect.any(Number),
210
+ }));
211
+ expect(result.results.map((r) => r.nodeId)).toStrictEqual(['a', 'b']);
212
+ });
213
+ it('evaluates conditional edges', async () => {
214
+ const graph = new Graph({
215
+ nodes: [makeAgent('a', 'a-reply'), makeAgent('b', 'b-reply'), makeAgent('c', 'c-reply')],
216
+ edges: [
217
+ { source: 'a', target: 'b', handler: () => true },
218
+ { source: 'a', target: 'c', handler: () => false },
219
+ ],
220
+ });
221
+ const result = await graph.invoke('start');
222
+ expect(result).toEqual(expect.objectContaining({
223
+ status: Status.COMPLETED,
224
+ duration: expect.any(Number),
225
+ }));
226
+ expect(result.results.map((r) => r.nodeId)).toStrictEqual(['a', 'b']);
227
+ });
228
+ it('passes task + dependency content to downstream nodes', async () => {
229
+ const agentB = makeAgent('b');
230
+ const streamSpy = vi.spyOn(agentB, 'stream');
231
+ const graph = new Graph({
232
+ nodes: [makeAgent('a', 'from-a'), agentB],
233
+ edges: [['a', 'b']],
234
+ });
235
+ await graph.invoke('task-input');
236
+ expect(streamSpy).toHaveBeenCalled();
237
+ const input = streamSpy.mock.calls[0][0];
238
+ expect(input.map((b) => b.text)).toStrictEqual(['task-input', '[node: a]', 'from-a']);
239
+ });
240
+ it('returns failed result when agent throws', async () => {
241
+ const model = new MockMessageModel().addTurn(new Error('agent exploded'));
242
+ const agent = new Agent({ model, printer: false, agentId: 'a' });
243
+ const graph = new Graph({
244
+ nodes: [agent, makeAgent('b', 'b-reply')],
245
+ edges: [['a', 'b']],
246
+ });
247
+ const result = await graph.invoke('go');
248
+ expect(result).toEqual(expect.objectContaining({
249
+ status: Status.FAILED,
250
+ duration: expect.any(Number),
251
+ }));
252
+ expect(result.results).toHaveLength(1);
253
+ expect(result.results[0]).toEqual(expect.objectContaining({ nodeId: 'a', status: Status.FAILED }));
254
+ });
255
+ it('propagates unexpected errors from node execution', async () => {
256
+ const graph = new Graph({
257
+ nodes: [makeAgent('a')],
258
+ edges: [],
259
+ });
260
+ const node = graph.nodes.get('a');
261
+ // eslint-disable-next-line require-yield
262
+ vi.spyOn(node, 'stream').mockImplementation(async function* () {
263
+ throw new Error('unexpected failure');
264
+ });
265
+ await expect(graph.invoke('go')).rejects.toThrow('unexpected failure');
266
+ });
267
+ it('throws when maxSteps is exceeded', async () => {
268
+ const graph = new Graph({
269
+ nodes: [makeAgent('a'), makeAgent('b'), makeAgent('c')],
270
+ edges: [
271
+ ['a', 'b'],
272
+ ['b', 'c'],
273
+ ],
274
+ maxSteps: 2,
275
+ });
276
+ await expect(graph.invoke('go')).rejects.toThrow('max steps reached');
277
+ });
278
+ it('calls initialize only once across invocations', async () => {
279
+ let callCount = 0;
280
+ const graph = new Graph({
281
+ nodes: [makeAgent('a')],
282
+ edges: [],
283
+ });
284
+ graph.hooks.addCallback(MultiAgentInitializedEvent, () => {
285
+ callCount++;
286
+ });
287
+ await graph.invoke('first');
288
+ await graph.invoke('second');
289
+ expect(callCount).toBe(1);
290
+ });
291
+ it('respects maxConcurrency limit', async () => {
292
+ let concurrent = 0;
293
+ let maxConcurrent = 0;
294
+ const graph = new Graph({
295
+ nodes: [makeAgent('a'), makeAgent('b'), makeAgent('c')],
296
+ edges: [
297
+ ['a', 'b'],
298
+ ['a', 'c'],
299
+ ],
300
+ maxConcurrency: 1,
301
+ });
302
+ graph.hooks.addCallback(BeforeNodeCallEvent, () => {
303
+ concurrent++;
304
+ maxConcurrent = Math.max(maxConcurrent, concurrent);
305
+ });
306
+ graph.hooks.addCallback(AfterNodeCallEvent, () => {
307
+ concurrent--;
308
+ });
309
+ const result = await graph.invoke('go');
310
+ expect(result.status).toBe(Status.COMPLETED);
311
+ expect(maxConcurrent).toBe(1);
312
+ });
313
+ it('preserves agent messages and state after execution', async () => {
314
+ const agent = makeAgent('a', 'reply');
315
+ const messagesBefore = [...agent.messages];
316
+ const stateBefore = agent.state.getAll();
317
+ const graph = new Graph({
318
+ nodes: [agent],
319
+ edges: [],
320
+ });
321
+ await graph.invoke('hello');
322
+ expect(agent.messages).toStrictEqual(messagesBefore);
323
+ expect(agent.state.getAll()).toStrictEqual(stateBefore);
324
+ });
325
+ it('executes join node exactly once when all parents complete concurrently', async () => {
326
+ const graph = new Graph({
327
+ nodes: [makeAgent('a', 'a-reply'), makeAgent('b', 'b-reply'), makeAgent('c', 'c-reply')],
328
+ edges: [
329
+ ['a', 'c'],
330
+ ['b', 'c'],
331
+ ],
332
+ });
333
+ const nodeC = graph.nodes.get('c');
334
+ const streamSpy = vi.spyOn(nodeC, 'stream');
335
+ const result = await graph.invoke('go');
336
+ expect(result.status).toBe(Status.COMPLETED);
337
+ expect(streamSpy).toHaveBeenCalledTimes(1);
338
+ });
339
+ it('re-executes node in a cycle when conditional edge allows re-entry', async () => {
340
+ let visits = 0;
341
+ const graph = new Graph({
342
+ nodes: [makeAgent('a', 'a-reply')],
343
+ edges: [
344
+ {
345
+ source: 'a',
346
+ target: 'a',
347
+ handler: () => {
348
+ visits++;
349
+ return visits < 2;
350
+ },
351
+ },
352
+ ],
353
+ sources: ['a'],
354
+ });
355
+ const result = await graph.invoke('go');
356
+ expect(result.status).toBe(Status.COMPLETED);
357
+ expect(result.results).toHaveLength(2);
358
+ expect(result.results.map((r) => r.nodeId)).toStrictEqual(['a', 'a']);
359
+ expect(visits).toBe(2);
360
+ });
361
+ });
362
+ describe('stream', () => {
363
+ it('yields lifecycle events in correct order for single node', async () => {
364
+ const graph = new Graph({
365
+ nodes: [makeAgent('a', 'a-reply')],
366
+ edges: [],
367
+ });
368
+ const { items, result } = await collectGenerator(graph.stream('go'));
369
+ const eventTypes = items.map((e) => e.type);
370
+ expect(result.status).toBe(Status.COMPLETED);
371
+ expect(result.results.map((r) => r.nodeId)).toStrictEqual(['a']);
372
+ expect(eventTypes).toStrictEqual([
373
+ 'beforeMultiAgentInvocationEvent',
374
+ 'beforeNodeCallEvent',
375
+ ...eventTypes.filter((t) => t === 'nodeStreamUpdateEvent'),
376
+ 'nodeResultEvent',
377
+ 'afterNodeCallEvent',
378
+ 'afterMultiAgentInvocationEvent',
379
+ 'multiAgentResultEvent',
380
+ ]);
381
+ });
382
+ it('yields handoff events on transitions between nodes', async () => {
383
+ const graph = new Graph({
384
+ nodes: [makeAgent('a', 'a-reply'), makeAgent('b', 'b-reply')],
385
+ edges: [['a', 'b']],
386
+ });
387
+ const { items } = await collectGenerator(graph.stream('go'));
388
+ const handoffEvents = items.filter((e) => e.type === 'multiAgentHandoffEvent');
389
+ expect(handoffEvents).toHaveLength(1);
390
+ expect(handoffEvents[0]).toEqual(expect.objectContaining({
391
+ type: 'multiAgentHandoffEvent',
392
+ source: 'a',
393
+ targets: ['b'],
394
+ }));
395
+ });
396
+ it('returns cancelled result when cancel is true', async () => {
397
+ const graph = new Graph({
398
+ nodes: [makeAgent('a')],
399
+ edges: [],
400
+ });
401
+ graph.hooks.addCallback(BeforeNodeCallEvent, (event) => {
402
+ event.cancel = true;
403
+ });
404
+ const { items, result } = await collectGenerator(graph.stream('go'));
405
+ expect(result.status).toBe(Status.CANCELLED);
406
+ expect(result.results).toHaveLength(1);
407
+ expect(result.results[0]).toEqual(expect.objectContaining({ nodeId: 'a', status: Status.CANCELLED, duration: 0 }));
408
+ const cancelEvent = items.find((e) => e.type === 'nodeCancelEvent');
409
+ expect(cancelEvent).toEqual(expect.objectContaining({ nodeId: 'a', message: 'node cancelled by hook' }));
410
+ });
411
+ it('returns cancelled result with custom message when cancel is a string', async () => {
412
+ const graph = new Graph({
413
+ nodes: [makeAgent('a')],
414
+ edges: [],
415
+ });
416
+ graph.hooks.addCallback(BeforeNodeCallEvent, (event) => {
417
+ event.cancel = 'node not ready';
418
+ });
419
+ const { items, result } = await collectGenerator(graph.stream('go'));
420
+ expect(result.status).toBe(Status.CANCELLED);
421
+ const cancelEvent = items.find((e) => e.type === 'nodeCancelEvent');
422
+ expect(cancelEvent).toEqual(expect.objectContaining({ nodeId: 'a', message: 'node not ready' }));
423
+ });
424
+ it('cleans up running nodes when consumer breaks mid-stream', async () => {
425
+ const graph = new Graph({
426
+ nodes: [makeAgent('a', 'a-reply'), makeAgent('b', 'b-reply')],
427
+ edges: [['a', 'b']],
428
+ });
429
+ const gen = graph.stream('go');
430
+ const first = await gen.next();
431
+ expect(first.done).toBe(false);
432
+ // Simulates consumer break — should not hang waiting for node streams
433
+ const result = await gen.return(undefined);
434
+ expect(result.done).toBe(true);
435
+ });
436
+ });
437
+ });
438
+ //# sourceMappingURL=graph.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"graph.test.js","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/graph.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACjD,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAA;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,0BAA0B,EAAE,MAAM,cAAc,CAAA;AAClG,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAA;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AACpC,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AACvD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAA;AAEnC,SAAS,SAAS,CAAC,EAAU,EAAE,IAAI,GAAG,OAAO;IAC3C,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAA;IACjE,OAAO,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAA;AAC1D,CAAC;AAED,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;IACrB,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;YAChC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YACF,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;YAC7B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;gBACT,EAAE,EAAE,UAAU;aACf,CAAC,CAAA;YACF,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;gBACjD,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YACF,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAA;QACxD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YAEF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;gBACpD,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YACF,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,cAAc,CAAC,CAAA;QACjE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,IAAI,GAAG,IAAI,SAAS,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YACrD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,IAAI,CAAC;gBACb,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YACF,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACzC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;YAC9B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvC,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;aACtC,CAAC,CAAA;YACF,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACnC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC5B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;gBAC5C,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;aAC7C,CAAC,CACH,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAA;YAC5B,MAAM,CACJ,GAAG,EAAE,CACH,IAAI,KAAK,CAAC;gBACR,KAAK,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;gBACrB,KAAK,EAAE,EAAE;aACV,CAAC,CACL,CAAC,OAAO,CAAC,iCAAiC,CAAC,CAAA;QAC9C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CACJ,GAAG,EAAE,CACH,IAAI,KAAK,CAAC;gBACR,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,CAAC,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;aAC1B,CAAC,CACL,CAAC,OAAO,CAAC,wDAAwD,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CACJ,GAAG,EAAE,CACH,IAAI,KAAK,CAAC;gBACR,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;aAC1B,CAAC,CACL,CAAC,OAAO,CAAC,wDAAwD,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CACJ,GAAG,EAAE,CACH,IAAI,KAAK,CAAC;gBACR,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvC,KAAK,EAAE;oBACL,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;iBACX;aACF,CAAC,CACL,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAA;QACxC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,CACJ,GAAG,EAAE,CACH,IAAI,KAAK,CAAC;gBACR,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;gBACnF,KAAK,EAAE;oBACL,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,SAAS,EAAE,SAAS,CAAC;oBACtB,CAAC,SAAS,EAAE,SAAS,CAAC;iBACvB;aACF,CAAC,CACL,CAAC,OAAO,CAAC,sDAAsD,CAAC,CAAA;QACnE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,CACJ,GAAG,EAAE,CACH,IAAI,KAAK,CAAC;gBACR,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;gBACT,OAAO,EAAE,CAAC,SAAS,CAAC;aACrB,CAAC,CACL,CAAC,OAAO,CAAC,mDAAmD,CAAC,CAAA;QAChE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,CACJ,GAAG,EAAE,CACH,IAAI,KAAK,CAAC;gBACR,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;gBACT,QAAQ,EAAE,CAAC;aACZ,CAAC,CACL,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAA;QACjD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CACJ,GAAG,EAAE,CACH,IAAI,KAAK,CAAC;gBACR,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;gBACT,cAAc,EAAE,CAAC;aAClB,CAAC,CACL,CAAC,OAAO,CAAC,0CAA0C,CAAC,CAAA;QACvD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACxF,KAAK,EAAE;oBACL,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;iBACX;aACF,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;gBAClG,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CACH,CAAA;YACD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;QAC5E,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACxF,KAAK,EAAE;oBACL,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;iBACX;aACF,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC;oBAC9B,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;oBAC/D,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;iBAChE,CAAC;gBACF,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CACH,CAAA;YACD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;QACnF,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACxF,KAAK,EAAE;oBACL,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;iBACX;gBACD,cAAc,EAAE,CAAC;aAClB,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;gBAClG,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CACH,CAAA;YACD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACxC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;gBACtC,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YAEF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,KAAK,CAAC;gBACzC,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;aACxB,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;gBACtG,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CACH,CAAA;YACD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAA;QAC3E,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvC,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC;gBACvE,OAAO,EAAE,CAAC,GAAG,CAAC;aACf,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAEvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CACH,CAAA;YACD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACxF,KAAK,EAAE;oBACL,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE;oBACjD,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE;iBACnD;aACF,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CACH,CAAA;YACD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAA;YAC7B,MAAM,SAAS,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;YAE5C,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;gBACzC,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;aACpB,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,CAAA;YAEhC,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAA;YACpC,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAgB,CAAA;YACxD,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,YAAY,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAA;QACvF,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAA;YACzE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAA;YAEhE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,KAAK,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACzC,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;aACpB,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAEvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CACH,CAAA;YACD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACtC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACpG,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YAEF,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAE,CAAA;YAClC,yCAAyC;YACzC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,KAAK,SAAS,CAAC;gBACzD,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAA;YACvC,CAAC,CAAC,CAAA;YAEF,MAAM,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAA;QACxE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvD,KAAK,EAAE;oBACL,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;iBACX;gBACD,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAA;YAEF,MAAM,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,IAAI,SAAS,GAAG,CAAC,CAAA;YAEjB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YAEF,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,0BAA0B,EAAE,GAAG,EAAE;gBACvD,SAAS,EAAE,CAAA;YACb,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAC3B,MAAM,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;YAE5B,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC3B,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,IAAI,UAAU,GAAG,CAAC,CAAA;YAClB,IAAI,aAAa,GAAG,CAAC,CAAA;YAErB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvD,KAAK,EAAE;oBACL,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;iBACX;gBACD,cAAc,EAAE,CAAC;aAClB,CAAC,CAAA;YAEF,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,mBAAmB,EAAE,GAAG,EAAE;gBAChD,UAAU,EAAE,CAAA;gBACZ,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,EAAE,UAAU,CAAC,CAAA;YACrD,CAAC,CAAC,CAAA;YACF,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,kBAAkB,EAAE,GAAG,EAAE;gBAC/C,UAAU,EAAE,CAAA;YACd,CAAC,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAEvC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAC5C,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC/B,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,EAAE,OAAO,CAAC,CAAA;YACrC,MAAM,cAAc,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAA;YAC1C,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAA;YAExC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,KAAK,CAAC;gBACd,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC,cAAc,CAAC,CAAA;YACpD,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,aAAa,CAAC,WAAW,CAAC,CAAA;QACzD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACxF,KAAK,EAAE;oBACL,CAAC,GAAG,EAAE,GAAG,CAAC;oBACV,CAAC,GAAG,EAAE,GAAG,CAAC;iBACX;aACF,CAAC,CAAA;YAEF,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAE,CAAA;YACnC,MAAM,SAAS,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;YAE3C,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAEvC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAC5C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,IAAI,MAAM,GAAG,CAAC,CAAA;YAEd,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBAClC,KAAK,EAAE;oBACL;wBACE,MAAM,EAAE,GAAG;wBACX,MAAM,EAAE,GAAG;wBACX,OAAO,EAAE,GAAG,EAAE;4BACZ,MAAM,EAAE,CAAA;4BACR,OAAO,MAAM,GAAG,CAAC,CAAA;wBACnB,CAAC;qBACF;iBACF;gBACD,OAAO,EAAE,CAAC,GAAG,CAAC;aACf,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAEvC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACtC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;YACrE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACxB,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBAClC,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YAEF,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;YACpE,MAAM,UAAU,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;YAE3C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,CAAA;YAChE,MAAM,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC;gBAC/B,iCAAiC;gBACjC,qBAAqB;gBACrB,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,uBAAuB,CAAC;gBAC1D,iBAAiB;gBACjB,oBAAoB;gBACpB,gCAAgC;gBAChC,uBAAuB;aACxB,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBAC7D,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;aACpB,CAAC,CAAA;YAEF,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;YAE5D,MAAM,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,wBAAwB,CAAC,CAAA;YAC9E,MAAM,CAAC,aAAa,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAErC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAC9B,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,wBAAwB;gBAC9B,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,CAAC,GAAG,CAAC;aACf,CAAC,CACH,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YAEF,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,mBAAmB,EAAE,CAAC,KAA0B,EAAE,EAAE;gBAC1E,KAAK,CAAC,MAAM,GAAG,IAAI,CAAA;YACrB,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;YAEpE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACtC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;YAElH,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAA;YACnE,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,wBAAwB,EAAE,CAAC,CAAC,CAAA;QAC1G,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvB,KAAK,EAAE,EAAE;aACV,CAAC,CAAA;YAEF,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,mBAAmB,EAAE,CAAC,KAA0B,EAAE,EAAE;gBAC1E,KAAK,CAAC,MAAM,GAAG,gBAAgB,CAAA;YACjC,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;YAEpE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAE5C,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAA;YACnE,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAA;QAClG,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBAC7D,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;aACpB,CAAC,CAAA;YAEF,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;YAC9B,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;YAC9B,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAE9B,sEAAsE;YACtE,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,SAAkB,CAAC,CAAA;YACnD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
@@ -1,17 +1,19 @@
1
1
  import { beforeEach, describe, expect, it } from 'vitest';
2
+ import { z } from 'zod';
2
3
  import { Agent } from '../../agent/agent.js';
3
4
  import { MockMessageModel } from '../../__fixtures__/mock-message-model.js';
4
5
  import { collectGenerator } from '../../__fixtures__/model-test-helpers.js';
5
6
  import { TextBlock } from '../../types/messages.js';
6
- import { MultiAgentState, Status } from '../state.js';
7
- import { AgentNode, Node } from '../nodes.js';
7
+ import { MultiAgentResult, MultiAgentState, NodeResult, Status } from '../state.js';
8
+ import { MultiAgentHandoffEvent, NodeStreamUpdateEvent } from '../events.js';
9
+ import { AgentNode, MultiAgentNode, Node } from '../nodes.js';
8
10
  /**
9
11
  * Concrete Node subclass for testing the abstract base class.
10
12
  */
11
13
  class TestNode extends Node {
12
14
  _fn;
13
15
  constructor(id, fn) {
14
- super(id);
16
+ super(id, {});
15
17
  this._fn = fn;
16
18
  }
17
19
  async *handle(args, state) {
@@ -21,7 +23,7 @@ class TestNode extends Node {
21
23
  describe('Node', () => {
22
24
  let state;
23
25
  beforeEach(() => {
24
- state = new MultiAgentState();
26
+ state = new MultiAgentState({ nodeIds: ['test-node', 'fail-node'] });
25
27
  });
26
28
  describe('stream', () => {
27
29
  it('returns COMPLETED NodeResult on successful execution', async () => {
@@ -62,17 +64,20 @@ describe('AgentNode', () => {
62
64
  let state;
63
65
  beforeEach(() => {
64
66
  const model = new MockMessageModel().addTurn(new TextBlock('reply'));
65
- agent = new Agent({ model, printer: false, state: { key1: 'value1' } });
66
- node = new AgentNode('agent-1', agent);
67
- state = new MultiAgentState();
67
+ agent = new Agent({ model, printer: false, state: { key1: 'value1' }, agentId: 'agent-1' });
68
+ node = new AgentNode({ agent });
69
+ state = new MultiAgentState({ nodeIds: ['agent-1'] });
68
70
  });
69
71
  describe('handle', () => {
70
72
  it('wraps agent events and returns content', async () => {
71
73
  const { items, result } = await collectGenerator(node.stream([new TextBlock('prompt')], state));
72
- expect(items.length).toBeGreaterThan(0);
73
- for (const event of items) {
74
+ const streamEvents = items.filter((e) => e.type === 'nodeStreamUpdateEvent');
75
+ expect(streamEvents.length).toBeGreaterThan(0);
76
+ for (const event of streamEvents) {
74
77
  expect(event).toEqual(expect.objectContaining({ type: 'nodeStreamUpdateEvent', nodeId: 'agent-1', nodeType: 'agentNode' }));
75
78
  }
79
+ const resultEvent = items.find((e) => e.type === 'nodeResultEvent');
80
+ expect(resultEvent).toEqual(expect.objectContaining({ type: 'nodeResultEvent', nodeId: 'agent-1', nodeType: 'agentNode', result }));
76
81
  expect(result).toEqual({
77
82
  type: 'nodeResult',
78
83
  nodeId: 'agent-1',
@@ -88,6 +93,22 @@ describe('AgentNode', () => {
88
93
  expect(agent.messages).toStrictEqual(messagesBefore);
89
94
  expect(agent.state.getAll()).toStrictEqual(stateBefore);
90
95
  });
96
+ it('passes structuredOutputSchema from state to the agent', async () => {
97
+ const schema = z.object({ agentName: z.string().optional(), message: z.string() });
98
+ const model = new MockMessageModel()
99
+ .addTurn({
100
+ type: 'toolUseBlock',
101
+ name: 'strands_structured_output',
102
+ toolUseId: 'tool-1',
103
+ input: { message: 'hello' },
104
+ })
105
+ .addTurn({ type: 'textBlock', text: 'Done' });
106
+ agent = new Agent({ model, printer: false, agentId: 'schema-agent' });
107
+ node = new AgentNode({ agent });
108
+ state = new MultiAgentState({ nodeIds: ['schema-agent'], structuredOutputSchema: schema });
109
+ const { result } = await collectGenerator(node.stream('test', state));
110
+ expect(result.structuredOutput).toStrictEqual({ message: 'hello' });
111
+ });
91
112
  });
92
113
  describe('agent', () => {
93
114
  it('exposes the wrapped agent instance', () => {
@@ -95,4 +116,79 @@ describe('AgentNode', () => {
95
116
  });
96
117
  });
97
118
  });
119
+ describe('MultiAgentNode', () => {
120
+ const content = [new TextBlock('inner-result')];
121
+ /**
122
+ * Creates a mock orchestrator that yields the given events and returns a result with the given content.
123
+ */
124
+ function mockOrchestrator(id, events) {
125
+ return {
126
+ id,
127
+ invoke: async () => new MultiAgentResult({ results: [], duration: 0 }),
128
+ async *stream() {
129
+ for (const event of events) {
130
+ yield event;
131
+ }
132
+ return new MultiAgentResult({
133
+ results: [new NodeResult({ nodeId: id, status: Status.COMPLETED, duration: 0, content })],
134
+ content,
135
+ duration: 0,
136
+ });
137
+ },
138
+ };
139
+ }
140
+ let node;
141
+ let state;
142
+ beforeEach(() => {
143
+ const orchestrator = mockOrchestrator('inner', []);
144
+ node = new MultiAgentNode({ orchestrator });
145
+ state = new MultiAgentState({ nodeIds: ['inner'] });
146
+ });
147
+ describe('constructor', () => {
148
+ it('derives id from orchestrator', () => {
149
+ expect(node.id).toBe('inner');
150
+ });
151
+ });
152
+ describe('handle', () => {
153
+ it('passes through inner NodeStreamUpdateEvents', async () => {
154
+ const innerUpdate = new MultiAgentHandoffEvent({ source: 'x', targets: ['y'] });
155
+ const innerEvent = new NodeStreamUpdateEvent({
156
+ nodeId: 'deep-node',
157
+ nodeType: 'agentNode',
158
+ event: innerUpdate,
159
+ });
160
+ const orchestrator = mockOrchestrator('inner', [innerEvent]);
161
+ node = new MultiAgentNode({ orchestrator });
162
+ const { items } = await collectGenerator(node.stream([], state));
163
+ const streamEvents = items.filter((e) => e.type === 'nodeStreamUpdateEvent');
164
+ const passthrough = streamEvents.find((e) => e.nodeId === 'deep-node');
165
+ expect(passthrough).toBe(innerEvent);
166
+ });
167
+ it('wraps non-NodeStreamUpdateEvents with this node identity', async () => {
168
+ const handoff = new MultiAgentHandoffEvent({ source: 'a', targets: ['b'] });
169
+ const orchestrator = mockOrchestrator('inner', [handoff]);
170
+ node = new MultiAgentNode({ orchestrator });
171
+ const { items } = await collectGenerator(node.stream([], state));
172
+ const streamEvents = items.filter((e) => e.type === 'nodeStreamUpdateEvent');
173
+ const wrapped = streamEvents.find((e) => e.nodeId === 'inner' && e.event === handoff);
174
+ expect(wrapped).toBeDefined();
175
+ expect(wrapped.nodeType).toBe('multiAgentNode');
176
+ });
177
+ it('returns orchestrator content', async () => {
178
+ const { result } = await collectGenerator(node.stream([], state));
179
+ expect(result).toEqual(expect.objectContaining({
180
+ nodeId: 'inner',
181
+ status: Status.COMPLETED,
182
+ content,
183
+ }));
184
+ });
185
+ });
186
+ describe('orchestrator', () => {
187
+ it('exposes the wrapped orchestrator instance', () => {
188
+ const orchestrator = mockOrchestrator('test', []);
189
+ node = new MultiAgentNode({ orchestrator });
190
+ expect(node.orchestrator).toBe(orchestrator);
191
+ });
192
+ });
193
+ });
98
194
  //# sourceMappingURL=nodes.test.js.map