@strands-agents/sdk 0.6.0 → 1.0.0-rc.0

package/dist/src/multiagent/__tests__/graph.tracer.test.js

@@ -0,0 +1,204 @@
+import { describe, expect, it, vi, beforeEach } from 'vitest';
+import { Agent } from '../../agent/agent.js';
+import { MockMessageModel } from '../../__fixtures__/mock-message-model.js';
+import { TextBlock } from '../../types/messages.js';
+import { Tracer } from '../../telemetry/tracer.js';
+import { Graph } from '../graph.js';
+import { BeforeNodeCallEvent } from '../events.js';
+import { Status } from '../state.js';
+vi.mock('../../telemetry/tracer.js', () => ({
+    Tracer: vi.fn(function () {
+        return {
+            startAgentSpan: vi.fn().mockReturnValue({ mock: 'agentSpan' }),
+            endAgentSpan: vi.fn(),
+            startAgentLoopSpan: vi.fn().mockReturnValue({ mock: 'loopSpan' }),
+            endAgentLoopSpan: vi.fn(),
+            startModelInvokeSpan: vi.fn().mockReturnValue({ mock: 'modelSpan' }),
+            endModelInvokeSpan: vi.fn(),
+            startToolCallSpan: vi.fn().mockReturnValue({ mock: 'toolSpan' }),
+            endToolCallSpan: vi.fn(),
+            startMultiAgentSpan: vi.fn().mockReturnValue({ mock: 'multiAgentSpan' }),
+            endMultiAgentSpan: vi.fn(),
+            startNodeSpan: vi.fn().mockReturnValue({ mock: 'nodeSpan' }),
+            endNodeSpan: vi.fn(),
+            withSpanContext: vi.fn((_span, fn) => fn()),
+        };
+    }),
+}));
+/**
+ * Returns the Tracer mock instance owned by the Graph.
+ * Agents are constructed before the Graph, so the Graph's Tracer
+ * is always the last one created during Graph construction.
+ */
+function getGraphTracer() {
+    return vi.mocked(Tracer).mock.results.at(-1).value;
+}
+function makeAgent(id, text = 'reply') {
+    const model = new MockMessageModel().addTurn(new TextBlock(text));
+    return new Agent({ model, printer: false, id });
+}
+function makeAgentWithUsage(id, text = 'reply') {
+    const model = new MockMessageModel().addTurn(new TextBlock(text), {
+        usage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+    });
+    return new Agent({ model, printer: false, id });
+}
+describe('Graph tracer integration', () => {
+    let graph;
+    let tracer;
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    describe('multi-agent span lifecycle', () => {
+        it('starts and ends multi-agent span on successful invocation', async () => {
+            graph = new Graph({ id: 'test-graph', nodes: [makeAgent('a')], edges: [] });
+            tracer = getGraphTracer();
+            await graph.invoke('Hello');
+            expect(tracer.startMultiAgentSpan.mock.calls).toEqual([
+                [{ orchestratorId: 'test-graph', orchestratorType: 'graph', input: 'Hello' }],
+            ]);
+            expect(tracer.endMultiAgentSpan.mock.calls.length).toBe(1);
+            const [span, endOpts] = tracer.endMultiAgentSpan.mock.calls[0];
+            expect(span).toStrictEqual({ mock: 'multiAgentSpan' });
+            expect(endOpts).toEqual({
+                duration: expect.any(Number),
+                usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+            });
+            expect(endOpts.duration).toBeGreaterThanOrEqual(0);
+        });
+        it('passes exact usage from result to endMultiAgentSpan', async () => {
+            graph = new Graph({ id: 'test-graph', nodes: [makeAgentWithUsage('a')], edges: [] });
+            tracer = getGraphTracer();
+            await graph.invoke('Hello');
+            const [, endOpts] = tracer.endMultiAgentSpan.mock.calls[0];
+            expect(endOpts.usage).toStrictEqual({ inputTokens: 10, outputTokens: 5, totalTokens: 15 });
+        });
+        it('ends multi-agent span with error when maxSteps exceeded', async () => {
+            graph = new Graph({
+                nodes: [makeAgent('a'), makeAgent('b')],
+                edges: [['a', 'b']],
+                maxSteps: 1,
+            });
+            tracer = getGraphTracer();
+            await expect(graph.invoke('Hello')).rejects.toThrow('max steps reached');
+            const [span, endOpts] = tracer.endMultiAgentSpan.mock.calls[0];
+            expect(span).toStrictEqual({ mock: 'multiAgentSpan' });
+            expect(endOpts).toEqual({
+                duration: expect.any(Number),
+                error: expect.objectContaining({
+                    message: expect.stringContaining('max steps reached'),
+                }),
+            });
+            expect(endOpts.duration).toBeGreaterThanOrEqual(0);
+        });
+    });
+    describe('node span lifecycle', () => {
+        it('starts and ends node span for each node execution', async () => {
+            graph = new Graph({ nodes: [makeAgent('a'), makeAgent('b')], edges: [['a', 'b']] });
+            tracer = getGraphTracer();
+            await graph.invoke('Hello');
+            expect(tracer.startNodeSpan.mock.calls).toEqual([
+                [{ nodeId: 'a', nodeType: 'agentNode' }],
+                [{ nodeId: 'b', nodeType: 'agentNode' }],
+            ]);
+            expect(tracer.endNodeSpan.mock.calls.length).toBe(2);
+        });
+        it('ends node span with COMPLETED status, duration, and zero usage on success', async () => {
+            graph = new Graph({ nodes: [makeAgent('a')], edges: [] });
+            tracer = getGraphTracer();
+            await graph.invoke('Hello');
+            const [span, endOpts] = tracer.endNodeSpan.mock.calls[0];
+            expect(span).toStrictEqual({ mock: 'nodeSpan' });
+            expect(endOpts).toEqual({
+                status: Status.COMPLETED,
+                duration: expect.any(Number),
+                usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+            });
+            expect(endOpts.duration).toBeGreaterThanOrEqual(0);
+        });
+        it('passes exact usage from node result to endNodeSpan', async () => {
+            graph = new Graph({ nodes: [makeAgentWithUsage('a')], edges: [] });
+            tracer = getGraphTracer();
+            await graph.invoke('Hello');
+            const [, endOpts] = tracer.endNodeSpan.mock.calls[0];
+            expect(endOpts.status).toBe(Status.COMPLETED);
+            expect(endOpts.usage).toStrictEqual({ inputTokens: 10, outputTokens: 5, totalTokens: 15 });
+        });
+        it('ends node span with FAILED status when node agent throws', async () => {
+            const model = new MockMessageModel().addTurn(new Error('agent exploded'));
+            graph = new Graph({ nodes: [new Agent({ model, printer: false, id: 'a' })], edges: [] });
+            tracer = getGraphTracer();
+            const result = await graph.invoke('Hello');
+            expect(result.status).toBe(Status.FAILED);
+            const [span, endOpts] = tracer.endNodeSpan.mock.calls[0];
+            expect(span).toStrictEqual({ mock: 'nodeSpan' });
+            expect(endOpts).toEqual({
+                status: Status.FAILED,
+                duration: expect.any(Number),
+            });
+            expect(endOpts.duration).toBeGreaterThanOrEqual(0);
+        });
+        it('ends node span with CANCELLED status and zero duration when cancelled by hook', async () => {
+            graph = new Graph({ nodes: [makeAgent('a')], edges: [] });
+            tracer = getGraphTracer();
+            graph.addHook(BeforeNodeCallEvent, (event) => {
+                event.cancel = 'cancelled by test';
+            });
+            await graph.invoke('Hello');
+            expect(tracer.endNodeSpan.mock.calls).toEqual([[{ mock: 'nodeSpan' }, { status: Status.CANCELLED, duration: 0 }]]);
+        });
+    });
+    describe('null span handling', () => {
+        it('completes successfully when startMultiAgentSpan returns null', async () => {
+            graph = new Graph({ nodes: [makeAgent('a')], edges: [] });
+            tracer = getGraphTracer();
+            tracer.startMultiAgentSpan.mockReturnValue(null);
+            const result = await graph.invoke('Hello');
+            expect(result.status).toBe(Status.COMPLETED);
+            const [span] = tracer.endMultiAgentSpan.mock.calls[0];
+            expect(span).toBeNull();
+        });
+        it('completes successfully when startNodeSpan returns null', async () => {
+            graph = new Graph({ nodes: [makeAgent('a')], edges: [] });
+            tracer = getGraphTracer();
+            tracer.startNodeSpan.mockReturnValue(null);
+            const result = await graph.invoke('Hello');
+            expect(result.status).toBe(Status.COMPLETED);
+            const [span] = tracer.endNodeSpan.mock.calls[0];
+            expect(span).toBeNull();
+        });
+    });
+    describe('span context propagation', () => {
+        it('passes node span to every withSpanContext call during node execution', async () => {
+            graph = new Graph({ nodes: [makeAgent('a')], edges: [] });
+            tracer = getGraphTracer();
+            await graph.invoke('Hello');
+            // First call: multiAgentSpan to create nodeSpan, then nodeSpan for node.stream() + gen.next() calls
+            const calls = tracer.withSpanContext.mock.calls;
+            expect(calls.length).toBeGreaterThanOrEqual(3);
+            // First call uses multiAgentSpan to create the nodeSpan
+            expect(calls[0]).toEqual([{ mock: 'multiAgentSpan' }, expect.any(Function)]);
+            // Subsequent calls use nodeSpan for node execution
+            const subsequentCalls = calls.slice(1);
+            expect(subsequentCalls).toEqual(expect.arrayContaining(Array(subsequentCalls.length).fill([{ mock: 'nodeSpan' }, expect.any(Function)])));
+        });
+    });
+    describe('parallel node execution', () => {
+        it('creates separate node spans for parallel source nodes', async () => {
+            graph = new Graph({
+                nodes: [makeAgent('a'), makeAgent('b'), makeAgent('c')],
+                edges: [
+                    ['a', 'c'],
+                    ['b', 'c'],
+                ],
+            });
+            tracer = getGraphTracer();
+            await graph.invoke('Hello');
+            const nodeIds = tracer.startNodeSpan.mock.calls.map((call) => call[0].nodeId);
+            expect(nodeIds).toEqual(expect.arrayContaining(['a', 'b', 'c']));
+            expect(tracer.startNodeSpan.mock.calls.length).toBe(3);
+            expect(tracer.endNodeSpan.mock.calls.length).toBe(3);
+        });
+    });
+});
+//# sourceMappingURL=graph.tracer.test.js.map

package/dist/src/multiagent/__tests__/graph.tracer.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph.tracer.test.js","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/graph.tracer.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAqB,MAAM,QAAQ,CAAA;AAChF,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAA;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAA;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,2BAA2B,CAAA;AAClD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAA;AACnC,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAA;AAClD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AAkBpC,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;QACZ,OAAO;YACL,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;YAC9D,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;YACrB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YACjE,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;YACzB,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;YACpE,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YAChE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;YACxB,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;YACxE,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC1B,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YAC5D,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAc,EAAE,EAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;SACpE,CAAA;IACH,CAAC,CAAC;CACH,CAAC,CAAC,CAAA;AAEH;;;;GAIG;AACH,SAAS,cAAc;IACrB,OAAO,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,KAAK,CAAA;AACrD,CAAC;AAED,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,EAAE,EAAE,CAAC,CAAA;AACjD,CAAC;AAED,SAAS,kBAAkB,CAAC,EAAU,EAAE,IAAI,GAAG,OAAO;IACpD,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,EAAE;QAChE,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;KAC7D,CAAC,CAAA;IACF,OAAO,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;AACjD,CAAC;AAED,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,KAAY,CAAA;IAChB,IAAI,MAA0B,CAAA;IAE9B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YAC3E,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;gBACpD,CAAC,EAAE,cAAc,EAAE,YAAY,EAAE,gBAAgB,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;aAC9E,CAAC,CAAA;YACF,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAE1D,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAA;YAC/D,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAA;YACtD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;aAC3D,CAAC,CAAA;YACF,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YACpF,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAA;YAC3D,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAA;QAC5F,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,KAAK,GAAG,IAAI,KAAK,CAAC;gBAChB,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC;gBACvC,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;gBACnB,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAA;YACF,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;YAExE,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAA;YAC/D,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAA;YACtD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC7B,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC;iBACtD,CAAC;aACH,CAAC,CAAA;YACF,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAA;YACnF,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;gBAC9C,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;gBACxC,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;aACzC,CAAC,CAAA;YACF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACtD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YACzD,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAA;YACzD,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAA;YAChD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;aAC3D,CAAC,CAAA;YACF,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YAClE,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAA;YACrD,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAC7C,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAA;QAC5F,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAA;YACzE,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YACxF,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE1C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;YACzC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAA;YACzD,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAA;YAChD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CAAA;YACF,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;YAC7F,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YACzD,MAAM,GAAG,cAAc,EAAE,CAAA;YACzB,KAAK,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC3C,KAAK,CAAC,MAAM,GAAG,mBAAmB,CAAA;YACpC,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;QACpH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YACzD,MAAM,GAAG,cAAc,EAAE,CAAA;YACzB,MAAM,CAAC,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;YAEhD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE1C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAC5C,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAA;YACtD,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YACzD,MAAM,GAAG,cAAc,EAAE,CAAA;YACzB,MAAM,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;YAE1C,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE1C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;YAC5C,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAA;YAChD,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAA;YACzD,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,oGAAoG;YACpG,MAAM,KAAK,GAAG,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAA;YAC/C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YAE9C,wDAAwD;YACxD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;YAE5E,mDAAmD;YACnD,MAAM,eAAe,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;YACtC,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAC7B,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CACzG,CAAA;QACH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,KAAK,GAAG,IAAI,KAAK,CAAC;gBAChB,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;aACF,CAAC,CAAA;YACF,MAAM,GAAG,cAAc,EAAE,CAAA;YAEzB,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,OAAO,GAAG,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAA;YAC7E,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAA;YAChE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YACtD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACtD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/multiagent/__tests__/nodes.test.js

@@ -28,11 +28,19 @@ describe('Node', () => {
     describe('stream', () => {
         it('returns COMPLETED NodeResult on successful execution', async () => {
             const content = [new TextBlock('result')];
-            // eslint-disable-next-line require-yield
             const node = new TestNode('test-node', async function* () {
+                yield* [];
                 return { content };
             });
-            const { result } = await collectGenerator(node.stream([], state));
+            const { items, result } = await collectGenerator(node.stream([], state));
+            const resultEvent = items.find((e) => e.type === 'nodeResultEvent');
+            expect(resultEvent).toEqual({
+                type: 'nodeResultEvent',
+                nodeId: 'test-node',
+                nodeType: 'node',
+                state,
+                result,
+            });
             expect(result).toEqual({
                 type: 'nodeResult',
                 nodeId: 'test-node',
@@ -42,11 +50,19 @@ describe('Node', () => {
             });
         });
         it('catches errors and returns FAILED NodeResult', async () => {
-            // eslint-disable-next-line require-yield
             const node = new TestNode('fail-node', async function* () {
+                yield* [];
                 throw new Error('boom');
             });
-            const { result } = await collectGenerator(node.stream([], state));
+            const { items, result } = await collectGenerator(node.stream([], state));
+            const resultEvent = items.find((e) => e.type === 'nodeResultEvent');
+            expect(resultEvent).toEqual({
+                type: 'nodeResultEvent',
+                nodeId: 'fail-node',
+                nodeType: 'node',
+                state,
+                result,
+            });
             expect(result).toEqual({
                 type: 'nodeResult',
                 nodeId: 'fail-node',
@@ -64,7 +80,7 @@ describe('AgentNode', () => {
     let state;
     beforeEach(() => {
         const model = new MockMessageModel().addTurn(new TextBlock('reply'));
-        agent = new Agent({ model, printer: false, state: { key1: 'value1' }, agentId: 'agent-1' });
+        agent = new Agent({ model, printer: false, appState: { key1: 'value1' }, id: 'agent-1' });
         node = new AgentNode({ agent });
         state = new MultiAgentState({ nodeIds: ['agent-1'] });
     });
@@ -84,16 +100,17 @@ describe('AgentNode', () => {
                 status: Status.COMPLETED,
                 content: expect.arrayContaining([expect.objectContaining({ type: 'textBlock', text: 'reply' })]),
                 duration: expect.any(Number),
+                usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
             });
         });
         it('restores agent messages and state after execution', async () => {
             const messagesBefore = [...agent.messages];
-            const stateBefore = agent.state.getAll();
+            const stateBefore = agent.appState.getAll();
             await collectGenerator(node.stream([new TextBlock('prompt')], state));
             expect(agent.messages).toStrictEqual(messagesBefore);
-            expect(agent.state.getAll()).toStrictEqual(stateBefore);
+            expect(agent.appState.getAll()).toStrictEqual(stateBefore);
         });
-        it('passes structuredOutputSchema from state to the agent', async () => {
+        it('passes structuredOutputSchema from options to the agent', async () => {
             const schema = z.object({ agentName: z.string().optional(), message: z.string() });
             const model = new MockMessageModel()
                 .addTurn({
@@ -103,10 +120,10 @@ describe('AgentNode', () => {
                 input: { message: 'hello' },
             })
                 .addTurn({ type: 'textBlock', text: 'Done' });
-            agent = new Agent({ model, printer: false, agentId: 'schema-agent' });
+            agent = new Agent({ model, printer: false, id: 'schema-agent' });
             node = new AgentNode({ agent });
-            state = new MultiAgentState({ nodeIds: ['schema-agent'], structuredOutputSchema: schema });
-            const { result } = await collectGenerator(node.stream('test', state));
+            state = new MultiAgentState({ nodeIds: ['schema-agent'] });
+            const { result } = await collectGenerator(node.stream('test', state, { structuredOutputSchema: schema }));
             expect(result.structuredOutput).toStrictEqual({ message: 'hello' });
         });
     });
@@ -135,6 +152,7 @@ describe('MultiAgentNode', () => {
                     duration: 0,
                 });
             },
+            addHook: () => () => { },
         };
     }
     let node;
@@ -151,11 +169,12 @@ describe('MultiAgentNode', () => {
     });
     describe('handle', () => {
         it('passes through inner NodeStreamUpdateEvents', async () => {
-            const innerUpdate = new MultiAgentHandoffEvent({ source: 'x', targets: ['y'] });
+            const innerUpdate = new MultiAgentHandoffEvent({ source: 'x', targets: ['y'], state });
             const innerEvent = new NodeStreamUpdateEvent({
                 nodeId: 'deep-node',
                 nodeType: 'agentNode',
-                event: innerUpdate,
+                state,
+                inner: { source: 'multiAgent', event: innerUpdate },
             });
             const orchestrator = mockOrchestrator('inner', [innerEvent]);
             node = new MultiAgentNode({ orchestrator });
@@ -165,12 +184,12 @@ describe('MultiAgentNode', () => {
             expect(passthrough).toBe(innerEvent);
         });
         it('wraps non-NodeStreamUpdateEvents with this node identity', async () => {
-            const handoff = new MultiAgentHandoffEvent({ source: 'a', targets: ['b'] });
+            const handoff = new MultiAgentHandoffEvent({ source: 'a', targets: ['b'], state });
             const orchestrator = mockOrchestrator('inner', [handoff]);
             node = new MultiAgentNode({ orchestrator });
             const { items } = await collectGenerator(node.stream([], state));
             const streamEvents = items.filter((e) => e.type === 'nodeStreamUpdateEvent');
-            const wrapped = streamEvents.find((e) => e.nodeId === 'inner' && e.event === handoff);
+            const wrapped = streamEvents.find((e) => e.nodeId === 'inner' && e.inner.event === handoff);
             expect(wrapped).toBeDefined();
             expect(wrapped.nodeType).toBe('multiAgentNode');
         });
@@ -182,6 +201,48 @@ describe('MultiAgentNode', () => {
                 content,
             }));
         });
+        it('propagates FAILED status from inner orchestrator', async () => {
+            const failedOrchestrator = {
+                id: 'inner',
+                invoke: async () => new MultiAgentResult({ results: [], duration: 0 }),
+                async *stream() {
+                    yield* [];
+                    return new MultiAgentResult({
+                        status: Status.FAILED,
+                        results: [
+                            new NodeResult({ nodeId: 'x', status: Status.FAILED, duration: 0, error: new Error('inner boom') }),
+                        ],
+                        content: [],
+                        duration: 0,
+                        error: new Error('inner boom'),
+                    });
+                },
+                addHook: () => () => { },
+            };
+            node = new MultiAgentNode({ orchestrator: failedOrchestrator });
+            const { result } = await collectGenerator(node.stream([], state));
+            expect(result.status).toBe(Status.FAILED);
+            expect(result.error?.message).toBe('inner boom');
+        });
+        it('propagates CANCELLED status from inner orchestrator', async () => {
+            const cancelledOrchestrator = {
+                id: 'inner',
+                invoke: async () => new MultiAgentResult({ results: [], duration: 0 }),
+                async *stream() {
+                    yield* [];
+                    return new MultiAgentResult({
+                        status: Status.CANCELLED,
+                        results: [],
+                        content: [],
+                        duration: 0,
+                    });
+                },
+                addHook: () => () => { },
+            };
+            node = new MultiAgentNode({ orchestrator: cancelledOrchestrator });
+            const { result } = await collectGenerator(node.stream([], state));
+            expect(result.status).toBe(Status.CANCELLED);
+        });
     });
     describe('orchestrator', () => {
         it('exposes the wrapped orchestrator instance', () => {

package/dist/src/multiagent/__tests__/nodes.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"nodes.test.js","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/nodes.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACzD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAA;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAA;AACnD,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AAEnF,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAA;AAC5E,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,aAAa,CAAA;AAI7D;;GAEG;AACH,MAAM,QAAS,SAAQ,IAAI;IACR,GAAG,CAGmD;IAEvE,YACE,EAAU,EACV,EAAoH;QAEpH,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QACb,IAAI,CAAC,GAAG,GAAG,EAAE,CAAA;IACf,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,IAAgB,EAChB,KAAsB;QAEtB,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;IACrC,CAAC;CACF;AAED,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;IACpB,IAAI,KAAsB,CAAA;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,OAAO,GAAG,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;YACzC,yCAAyC;YACzC,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,WAAW,EAAE,KAAK,SAAS,CAAC;gBACpD,OAAO,EAAE,OAAO,EAAE,CAAA;YACpB,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEjE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,WAAW;gBACnB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO;gBACP,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,yCAAyC;YACzC,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,WAAW,EAAE,KAAK,SAAS,CAAC;gBACpD,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,CAAA;YACzB,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEjE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,WAAW;gBACnB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,EAAE;gBACX,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aACpD,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,KAAY,CAAA;IAChB,IAAI,IAAe,CAAA;IACnB,IAAI,KAAsB,CAAA;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;QACpE,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAA;QAC3F,IAAI,GAAG,IAAI,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;QAC/B,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAA;IACvD,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAA;YAE/F,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAuB,CAAC,CAAA;YAC5E,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;YAC9C,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;gBACjC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CACnB,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CACrG,CAAA;YACH,CAAC;YAED,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,CACzB,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CACvG,CAAA;YAED,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;gBAChG,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,cAAc,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAA;YAC1C,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAA;YAExC,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAA;YAErE,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,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;YAElF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE;iBACjC,OAAO,CAAC;gBACP,IAAI,EAAE,cAAc;gBACpB,IAAI,EAAE,2BAA2B;gBACjC,SAAS,EAAE,QAAQ;gBACnB,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;aAC5B,CAAC;iBACD,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;YAE/C,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAA;YACrE,IAAI,GAAG,IAAI,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAC/B,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,cAAc,CAAC,EAAE,sBAAsB,EAAE,MAAM,EAAE,CAAC,CAAA;YAE1F,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAA;YAErE,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,OAAO,GAAG,CAAC,IAAI,SAAS,CAAC,cAAc,CAAC,CAAC,CAAA;IAE/C;;OAEG;IACH,SAAS,gBAAgB,CAAC,EAAU,EAAE,MAA+B;QACnE,OAAO;YACL,EAAE;YACF,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,gBAAgB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;YACtE,KAAK,CAAC,CAAC,MAAM;gBACX,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,MAAM,KAAK,CAAA;gBACb,CAAC;gBACD,OAAO,IAAI,gBAAgB,CAAC;oBAC1B,OAAO,EAAE,CAAC,IAAI,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;oBACzF,OAAO;oBACP,QAAQ,EAAE,CAAC;iBACZ,CAAC,CAAA;YACJ,CAAC;SACF,CAAA;IACH,CAAC;IAED,IAAI,IAAoB,CAAA;IACxB,IAAI,KAAsB,CAAA;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,YAAY,GAAG,gBAAgB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAA;QAClD,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC,CAAA;QAC3C,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IACrD,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC/B,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YAC/E,MAAM,UAAU,GAAG,IAAI,qBAAqB,CAAC;gBAC3C,MAAM,EAAE,WAAW;gBACnB,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,WAAW;aACnB,CAAC,CAAA;YACF,MAAM,YAAY,GAAG,gBAAgB,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC,CAAA;YAC5D,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC,CAAA;YAE3C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEhE,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAuB,CAA4B,CAAA;YACvG,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAA;YACtE,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACtC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YAC3E,MAAM,YAAY,GAAG,gBAAgB,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAA;YACzD,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC,CAAA;YAE3C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEhE,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAuB,CAA4B,CAAA;YACvG,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,OAAO,IAAI,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,CAAA;YACrF,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAA;YAC7B,MAAM,CAAC,OAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEjE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,OAAO;gBACf,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO;aACR,CAAC,CACH,CAAA;QACH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;YACjD,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC,CAAA;YAC3C,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QAC9C,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}
+{"version":3,"file":"nodes.test.js","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/nodes.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACzD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAA;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAA;AACnD,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AAEnF,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAA;AAC5E,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,aAAa,CAAA;AAI7D;;GAEG;AACH,MAAM,QAAS,SAAQ,IAAI;IACR,GAAG,CAGmD;IAEvE,YACE,EAAU,EACV,EAGuE;QAEvE,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QACb,IAAI,CAAC,GAAG,GAAG,EAAE,CAAA;IACf,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,IAAqB,EACrB,KAAsB;QAEtB,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;IACrC,CAAC;CACF;AAED,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;IACpB,IAAI,KAAsB,CAAA;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,OAAO,GAAG,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAA;YACzC,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,WAAW,EAAE,KAAK,SAAS,CAAC;gBACpD,KAAK,CAAC,CAAC,EAAE,CAAA;gBACT,OAAO,EAAE,OAAO,EAAE,CAAA;YACpB,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAExE,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;gBAC1B,IAAI,EAAE,iBAAiB;gBACvB,MAAM,EAAE,WAAW;gBACnB,QAAQ,EAAE,MAAM;gBAChB,KAAK;gBACL,MAAM;aACP,CAAC,CAAA;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,WAAW;gBACnB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO;gBACP,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;aAC7B,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,WAAW,EAAE,KAAK,SAAS,CAAC;gBACpD,KAAK,CAAC,CAAC,EAAE,CAAA;gBACT,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,CAAA;YACzB,CAAC,CAAC,CAAA;YAEF,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAExE,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;gBAC1B,IAAI,EAAE,iBAAiB;gBACvB,MAAM,EAAE,WAAW;gBACnB,QAAQ,EAAE,MAAM;gBAChB,KAAK;gBACL,MAAM;aACP,CAAC,CAAA;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,WAAW;gBACnB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,EAAE;gBACX,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;aACpD,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,KAAY,CAAA;IAChB,IAAI,IAAe,CAAA;IACnB,IAAI,KAAsB,CAAA;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;QACpE,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,CAAA;QACzF,IAAI,GAAG,IAAI,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;QAC/B,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAA;IACvD,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAA;YAE/F,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAuB,CAAC,CAAA;YAC5E,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;YAC9C,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;gBACjC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CACnB,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CACrG,CAAA;YACH,CAAC;YAED,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,CACzB,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CACvG,CAAA;YAED,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;gBAChG,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;aAC3D,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,cAAc,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,CAAA;YAC1C,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAA;YAE3C,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAA;YAErE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC,cAAc,CAAC,CAAA;YACpD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,aAAa,CAAC,WAAW,CAAC,CAAA;QAC5D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;YAElF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE;iBACjC,OAAO,CAAC;gBACP,IAAI,EAAE,cAAc;gBACpB,IAAI,EAAE,2BAA2B;gBACjC,SAAS,EAAE,QAAQ;gBACnB,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;aAC5B,CAAC;iBACD,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;YAE/C,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,CAAA;YAChE,IAAI,GAAG,IAAI,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAC/B,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,cAAc,CAAC,EAAE,CAAC,CAAA;YAE1D,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,sBAAsB,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;YAEzG,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,aAAa,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAA;QACrE,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,OAAO,GAAG,CAAC,IAAI,SAAS,CAAC,cAAc,CAAC,CAAC,CAAA;IAE/C;;OAEG;IACH,SAAS,gBAAgB,CAAC,EAAU,EAAE,MAA+B;QACnE,OAAO;YACL,EAAE;YACF,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,gBAAgB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;YACtE,KAAK,CAAC,CAAC,MAAM;gBACX,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,MAAM,KAAK,CAAA;gBACb,CAAC;gBACD,OAAO,IAAI,gBAAgB,CAAC;oBAC1B,OAAO,EAAE,CAAC,IAAI,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;oBACzF,OAAO;oBACP,QAAQ,EAAE,CAAC;iBACZ,CAAC,CAAA;YACJ,CAAC;YACD,OAAO,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAE,CAAC;SACxB,CAAA;IACH,CAAC;IAED,IAAI,IAAoB,CAAA;IACxB,IAAI,KAAsB,CAAA;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,YAAY,GAAG,gBAAgB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAA;QAClD,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC,CAAA;QAC3C,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IACrD,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC/B,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,WAAW,GAAG,IAAI,sBAAsB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YACtF,MAAM,UAAU,GAAG,IAAI,qBAAqB,CAAC;gBAC3C,MAAM,EAAE,WAAW;gBACnB,QAAQ,EAAE,WAAW;gBACrB,KAAK;gBACL,KAAK,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE;aACpD,CAAC,CAAA;YACF,MAAM,YAAY,GAAG,gBAAgB,CAAC,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC,CAAA;YAC5D,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC,CAAA;YAE3C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEhE,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAuB,CAA4B,CAAA;YACvG,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAA;YACtE,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACtC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAClF,MAAM,YAAY,GAAG,gBAAgB,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAA;YACzD,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC,CAAA;YAE3C,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEhE,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,uBAAuB,CAA4B,CAAA;YACvG,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,OAAO,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,KAAK,OAAO,CAAC,CAAA;YAC3F,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAA;YAC7B,MAAM,CAAC,OAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEjE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CACpB,MAAM,CAAC,gBAAgB,CAAC;gBACtB,MAAM,EAAE,OAAO;gBACf,MAAM,EAAE,MAAM,CAAC,SAAS;gBACxB,OAAO;aACR,CAAC,CACH,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,kBAAkB,GAAe;gBACrC,EAAE,EAAE,OAAO;gBACX,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,gBAAgB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;gBACtE,KAAK,CAAC,CAAC,MAAM;oBACX,KAAK,CAAC,CAAC,EAAE,CAAA;oBACT,OAAO,IAAI,gBAAgB,CAAC;wBAC1B,MAAM,EAAE,MAAM,CAAC,MAAM;wBACrB,OAAO,EAAE;4BACP,IAAI,UAAU,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;yBACpG;wBACD,OAAO,EAAE,EAAE;wBACX,QAAQ,EAAE,CAAC;wBACX,KAAK,EAAE,IAAI,KAAK,CAAC,YAAY,CAAC;qBAC/B,CAAC,CAAA;gBACJ,CAAC;gBACD,OAAO,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAE,CAAC;aACxB,CAAA;YACD,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,kBAAkB,EAAE,CAAC,CAAA;YAE/D,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEjE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;YACzC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,qBAAqB,GAAe;gBACxC,EAAE,EAAE,OAAO;gBACX,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,gBAAgB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;gBACtE,KAAK,CAAC,CAAC,MAAM;oBACX,KAAK,CAAC,CAAC,EAAE,CAAA;oBACT,OAAO,IAAI,gBAAgB,CAAC;wBAC1B,MAAM,EAAE,MAAM,CAAC,SAAS;wBACxB,OAAO,EAAE,EAAE;wBACX,OAAO,EAAE,EAAE;wBACX,QAAQ,EAAE,CAAC;qBACZ,CAAC,CAAA;gBACJ,CAAC;gBACD,OAAO,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAE,CAAC;aACxB,CAAA;YACD,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,qBAAqB,EAAE,CAAC,CAAA;YAElE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAA;YAEjE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;QAC9C,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;YACjD,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,CAAC,CAAA;YAC3C,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QAC9C,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/multiagent/__tests__/swarm.test.js

@@ -4,23 +4,21 @@ import { MockMessageModel } from '../../__fixtures__/mock-message-model.js';
 import { collectGenerator } from '../../__fixtures__/model-test-helpers.js';
 import { BeforeNodeCallEvent, MultiAgentInitializedEvent } from '../events.js';
 import { TextBlock } from '../../types/messages.js';
-import { Status } from '../state.js';
+import { Status, MultiAgentState } from '../state.js';
 import { AgentNode } from '../nodes.js';
 import { Swarm } from '../swarm.js';
 /**
  * Creates an agent that produces a structured output handoff via the strands_structured_output tool.
- * The model returns a toolUseBlock with the handoff payload, then a text block to finish.
+ * The agent exits after the structured output tool succeeds (early-exit behavior).
  */
 function createHandoffAgent(agentId, handoff, description = `Agent ${agentId}`) {
-    const model = new MockMessageModel()
-        .addTurn({
+    const model = new MockMessageModel().addTurn({
         type: 'toolUseBlock',
         name: 'strands_structured_output',
         toolUseId: 'tool-1',
         input: handoff,
-    })
-        .addTurn(new TextBlock('Done'));
-    return new Agent({ model, printer: false, agentId, description });
+    });
+    return new Agent({ model, printer: false, id: agentId, description });
 }
 /**
  * Creates a simple agent that produces a final response (no handoff).
@@ -52,12 +50,21 @@ describe('Swarm', () => {
             });
             expect(swarm.nodes.get('a')).toBeInstanceOf(AgentNode);
         });
+        it('defaults start to the first node when not specified', () => {
+            const swarm = new Swarm({
+                nodes: [createFinalAgent('first', 'hi'), createFinalAgent('second', 'bye')],
+            });
+            expect(swarm.start.id).toBe('first');
+        });
         it('throws when start references unknown agent', () => {
             expect(() => new Swarm({
                 nodes: [createFinalAgent('a', 'hi')],
                 start: 'missing',
             })).toThrow('start=<missing> | start references unknown agent');
         });
+        it('throws when nodes list is empty', () => {
+            expect(() => new Swarm({ nodes: [] })).toThrow('nodes list is empty');
+        });
         it('throws on duplicate agent ids', () => {
             const agent = createFinalAgent('a', 'hi');
             expect(() => new Swarm({
@@ -83,9 +90,10 @@ describe('Swarm', () => {
             expect(result).toEqual(expect.objectContaining({
                 status: Status.COMPLETED,
                 duration: expect.any(Number),
-                content: expect.arrayContaining([expect.objectContaining({ type: 'textBlock', text: 'Done' })]),
+                content: [expect.objectContaining({ type: 'textBlock', text: 'final answer' })],
             }));
             expect(result.results.map((r) => r.nodeId)).toStrictEqual(['a']);
+            expect(result.results[0]?.structuredOutput).toEqual({ message: 'final answer' });
         });
         it('hands off from A to B and returns final output', async () => {
             const swarm = new Swarm({
@@ -127,6 +135,25 @@ describe('Swarm', () => {
             expect(texts).toContainEqual('handle this');
             expect(texts).toContainEqual(expect.stringContaining(JSON.stringify(contextData, null, 2)));
         });
+        it('excludes current agent from handoff schema', async () => {
+            const agentA = createHandoffAgent('a', { agentId: 'b', message: 'go to b' });
+            const agentB = createFinalAgent('b', 'done');
+            const streamSpyA = vi.spyOn(agentA, 'stream');
+            const streamSpyB = vi.spyOn(agentB, 'stream');
+            const swarm = new Swarm({
+                nodes: [agentA, agentB],
+                start: 'a',
+            });
+            await swarm.invoke('start');
+            // Agent A's handoff schema allows B but rejects A
+            const schemaA = streamSpyA.mock.calls[0][1].structuredOutputSchema;
+            expect(schemaA.parse({ agentId: 'b', message: 'ok' })).toStrictEqual({ agentId: 'b', message: 'ok' });
+            expect(() => schemaA.parse({ agentId: 'a', message: 'ok' })).toThrow();
+            // Agent B's handoff schema allows A but rejects B
+            const schemaB = streamSpyB.mock.calls[0][1].structuredOutputSchema;
+            expect(schemaB.parse({ agentId: 'a', message: 'ok' })).toStrictEqual({ agentId: 'a', message: 'ok' });
+            expect(() => schemaB.parse({ agentId: 'b', message: 'ok' })).toThrow();
+        });
         it('throws when maxSteps is exceeded', async () => {
             const swarm = new Swarm({
                 nodes: [createHandoffAgent('a', { agentId: 'b', message: 'to b' }), createFinalAgent('b', 'done')],
@@ -135,9 +162,32 @@ describe('Swarm', () => {
             });
             await expect(swarm.invoke('start')).rejects.toThrow('swarm reached step limit');
         });
+        it('does not throw when swarm completes normally using exactly maxSteps', async () => {
+            const swarm = new Swarm({
+                nodes: [createHandoffAgent('a', { agentId: 'b', message: 'to b' }), createFinalAgent('b', 'done by b')],
+                start: 'a',
+                maxSteps: 2,
+            });
+            const result = await swarm.invoke('start');
+            expect(result.status).toBe(Status.COMPLETED);
+            expect(result.results.map((r) => r.nodeId)).toStrictEqual(['a', 'b']);
+        });
+        it('returns cancelled result with custom message when cancel is a string', async () => {
+            const swarm = new Swarm({
+                nodes: [createFinalAgent('a', 'hi')],
+                start: 'a',
+            });
+            swarm.addHook(BeforeNodeCallEvent, (event) => {
+                event.cancel = 'agent not ready';
+            });
+            const { items, result } = await collectGenerator(swarm.stream('go'));
+            expect(result.status).toBe(Status.CANCELLED);
+            const cancelEvent = items.find((e) => e.type === 'nodeCancelEvent');
+            expect(cancelEvent).toEqual(expect.objectContaining({ nodeId: 'a', state: expect.any(MultiAgentState), message: 'agent not ready' }));
+        });
         it('returns failed result when agent throws', async () => {
             const model = new MockMessageModel().addTurn(new Error('agent exploded'));
-            const agent = new Agent({ model, printer: false, agentId: 'a', description: 'Agent a' });
+            const agent = new Agent({ model, printer: false, id: 'a', description: 'Agent a' });
             const swarm = new Swarm({
                 nodes: [{ agent }],
                 start: 'a',
@@ -153,7 +203,7 @@ describe('Swarm', () => {
                 nodes: [createFinalAgent('a', 'hi')],
                 start: 'a',
             });
-            swarm.hooks.addCallback(MultiAgentInitializedEvent, () => {
+            swarm.addHook(MultiAgentInitializedEvent, () => {
                 callCount++;
             });
             await swarm.invoke('first');
@@ -163,14 +213,14 @@ describe('Swarm', () => {
         it('preserves agent messages and state after execution', async () => {
             const agent = createFinalAgent('a', 'reply');
             const messagesBefore = [...agent.messages];
-            const stateBefore = agent.state.getAll();
+            const stateBefore = agent.appState.getAll();
             const swarm = new Swarm({
                 nodes: [agent],
                 start: 'a',
             });
             await swarm.invoke('hello');
             expect(agent.messages).toStrictEqual(messagesBefore);
-            expect(agent.state.getAll()).toStrictEqual(stateBefore);
+            expect(agent.appState.getAll()).toStrictEqual(stateBefore);
         });
     });
     describe('stream', () => {
@@ -206,6 +256,7 @@ describe('Swarm', () => {
                 type: 'multiAgentHandoffEvent',
                 source: 'a',
                 targets: ['b'],
+                state: expect.any(MultiAgentState),
             }));
         });
         it('returns cancelled result with default message when cancel is true', async () => {
@@ -213,7 +264,7 @@ describe('Swarm', () => {
                 nodes: [createFinalAgent('a', 'hi')],
                 start: 'a',
             });
-            swarm.hooks.addCallback(BeforeNodeCallEvent, (event) => {
+            swarm.addHook(BeforeNodeCallEvent, (event) => {
                 event.cancel = true;
             });
             const { items, result } = await collectGenerator(swarm.stream('go'));
@@ -221,20 +272,20 @@ describe('Swarm', () => {
             expect(result.results).toHaveLength(1);
             expect(result.results[0]).toEqual(expect.objectContaining({ nodeId: 'a', status: Status.CANCELLED, duration: 0 }));
             const cancelEvent = items.find((e) => e.type === 'nodeCancelEvent');
-            expect(cancelEvent).toEqual(expect.objectContaining({ nodeId: 'a', message: 'node cancelled by hook' }));
+            expect(cancelEvent).toEqual(expect.objectContaining({ nodeId: 'a', state: expect.any(MultiAgentState), message: 'node cancelled by hook' }));
         });
         it('returns cancelled result with custom message when cancel is a string', async () => {
             const swarm = new Swarm({
                 nodes: [createFinalAgent('a', 'hi')],
                 start: 'a',
             });
-            swarm.hooks.addCallback(BeforeNodeCallEvent, (event) => {
+            swarm.addHook(BeforeNodeCallEvent, (event) => {
                 event.cancel = 'agent not ready';
             });
             const { items, result } = await collectGenerator(swarm.stream('go'));
             expect(result.status).toBe(Status.CANCELLED);
             const cancelEvent = items.find((e) => e.type === 'nodeCancelEvent');
-            expect(cancelEvent).toEqual(expect.objectContaining({ nodeId: 'a', message: 'agent not ready' }));
+            expect(cancelEvent).toEqual(expect.objectContaining({ nodeId: 'a', state: expect.any(MultiAgentState), message: 'agent not ready' }));
         });
     });
 });