@strands-agents/sdk 1.6.0 → 1.8.0

package/dist/src/multiagent/__tests__/graph.invocation-state.test.js

@@ -1,95 +0,0 @@
-import { describe, expect, it } from 'vitest';
-import { Agent } from '../../agent/agent.js';
-import { MockMessageModel } from '../../__fixtures__/mock-message-model.js';
-import { BeforeModelCallEvent } from '../../hooks/events.js';
-import { TextBlock } from '../../types/messages.js';
-import { Graph } from '../graph.js';
-import { AfterMultiAgentInvocationEvent, AfterNodeCallEvent, BeforeMultiAgentInvocationEvent, BeforeNodeCallEvent, MultiAgentHandoffEvent, MultiAgentResultEvent, NodeResultEvent, NodeStreamUpdateEvent, } from '../events.js';
-describe('Graph invocationState forwarding', () => {
-    it('forwards invocationState to every node and mutations from one node are visible to the next', async () => {
-        const nodeAObserved = [];
-        const nodeBObserved = [];
-        const agentA = new Agent({
-            model: new MockMessageModel().addTurn(new TextBlock('A done')),
-            printer: false,
-            id: 'a',
-        });
-        agentA.addHook(BeforeModelCallEvent, (event) => {
-            nodeAObserved.push(event.invocationState);
-            event.invocationState.touchedByA = true;
-        });
-        const agentB = new Agent({
-            model: new MockMessageModel().addTurn(new TextBlock('B done')),
-            printer: false,
-            id: 'b',
-        });
-        agentB.addHook(BeforeModelCallEvent, (event) => {
-            nodeBObserved.push(event.invocationState);
-        });
-        const graph = new Graph({
-            nodes: [agentA, agentB],
-            edges: [{ source: 'a', target: 'b' }],
-        });
-        const state = { requestId: 'r-1' };
-        await graph.invoke('hello', { invocationState: state });
-        // Both nodes observe the same object reference.
-        expect(nodeAObserved[0]).toBe(state);
-        expect(nodeBObserved[0]).toBe(state);
-        // Node B sees node A's mutation.
-        expect(nodeBObserved[0]?.touchedByA).toBe(true);
-        expect(state.touchedByA).toBe(true);
-    });
-    it('defaults invocationState to {} when none is passed', async () => {
-        let observed;
-        const agentA = new Agent({
-            model: new MockMessageModel().addTurn(new TextBlock('A done')),
-            printer: false,
-            id: 'a',
-        });
-        agentA.addHook(BeforeModelCallEvent, (event) => {
-            observed = event.invocationState;
-        });
-        const graph = new Graph({
-            nodes: [agentA],
-            edges: [],
-        });
-        await graph.invoke('hello');
-        expect(observed).toEqual({});
-    });
-    it('every orchestrator and node event in a run carries the same invocationState reference', async () => {
-        const agentA = new Agent({
-            model: new MockMessageModel().addTurn(new TextBlock('A done')),
-            printer: false,
-            id: 'a',
-        });
-        const agentB = new Agent({
-            model: new MockMessageModel().addTurn(new TextBlock('B done')),
-            printer: false,
-            id: 'b',
-        });
-        const graph = new Graph({
-            nodes: [agentA, agentB],
-            edges: [{ source: 'a', target: 'b' }],
-        });
-        const state = { requestId: 'r-1' };
-        const observed = [];
-        const record = (label, ref) => {
-            observed.push({ label, ref });
-        };
-        graph.addHook(BeforeMultiAgentInvocationEvent, (e) => record('BeforeMultiAgentInvocation', e.invocationState));
-        graph.addHook(AfterMultiAgentInvocationEvent, (e) => record('AfterMultiAgentInvocation', e.invocationState));
-        graph.addHook(BeforeNodeCallEvent, (e) => record(`BeforeNodeCall:${e.nodeId}`, e.invocationState));
-        graph.addHook(AfterNodeCallEvent, (e) => record(`AfterNodeCall:${e.nodeId}`, e.invocationState));
-        graph.addHook(NodeStreamUpdateEvent, (e) => record(`NodeStreamUpdate:${e.nodeId}`, e.invocationState));
-        graph.addHook(NodeResultEvent, (e) => record(`NodeResult:${e.nodeId}`, e.invocationState));
-        graph.addHook(MultiAgentHandoffEvent, (e) => record('MultiAgentHandoff', e.invocationState));
-        graph.addHook(MultiAgentResultEvent, (e) => record('MultiAgentResult', e.invocationState));
-        await graph.invoke('hello', { invocationState: state });
-        // Every event observed at the orchestrator level must share the caller's reference.
-        expect(observed.length).toBeGreaterThan(0);
-        for (const { label, ref } of observed) {
-            expect(ref, `event=${label} saw a different invocationState object`).toBe(state);
-        }
-    });
-});
-//# sourceMappingURL=graph.invocation-state.test.js.map

package/dist/src/multiagent/__tests__/graph.invocation-state.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"graph.invocation-state.test.js","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/graph.invocation-state.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAA;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAA;AAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAA;AACnD,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAA;AACnC,OAAO,EACL,8BAA8B,EAC9B,kBAAkB,EAClB,+BAA+B,EAC/B,mBAAmB,EACnB,sBAAsB,EACtB,qBAAqB,EACrB,eAAe,EACf,qBAAqB,GACtB,MAAM,cAAc,CAAA;AAGrB,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;IAChD,EAAE,CAAC,4FAA4F,EAAE,KAAK,IAAI,EAAE;QAC1G,MAAM,aAAa,GAAsB,EAAE,CAAA;QAC3C,MAAM,aAAa,GAAsB,EAAE,CAAA;QAE3C,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;YACvB,KAAK,EAAE,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9D,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,GAAG;SACR,CAAC,CAAA;QACF,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC,KAAK,EAAE,EAAE;YAC7C,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAA;YACzC,KAAK,CAAC,eAAe,CAAC,UAAU,GAAG,IAAI,CAAA;QACzC,CAAC,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;YACvB,KAAK,EAAE,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9D,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,GAAG;SACR,CAAC,CAAA;QACF,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC,KAAK,EAAE,EAAE;YAC7C,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAA;QAC3C,CAAC,CAAC,CAAA;QAEF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;YACvB,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;SACtC,CAAC,CAAA;QAEF,MAAM,KAAK,GAAoB,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;QACnD,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC,CAAA;QAEvD,gDAAgD;QAChD,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACpC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAEpC,iCAAiC;QACjC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC/C,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACrC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,IAAI,QAAqC,CAAA;QAEzC,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;YACvB,KAAK,EAAE,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9D,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,GAAG;SACR,CAAC,CAAA;QACF,MAAM,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC,KAAK,EAAE,EAAE;YAC7C,QAAQ,GAAG,KAAK,CAAC,eAAe,CAAA;QAClC,CAAC,CAAC,CAAA;QAEF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,CAAC,MAAM,CAAC;YACf,KAAK,EAAE,EAAE;SACV,CAAC,CAAA;QAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAE3B,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IAC9B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;QACrG,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;YACvB,KAAK,EAAE,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9D,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,GAAG;SACR,CAAC,CAAA;QACF,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;YACvB,KAAK,EAAE,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9D,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,GAAG;SACR,CAAC,CAAA;QAEF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;YACvB,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;SACtC,CAAC,CAAA;QAEF,MAAM,KAAK,GAAoB,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;QACnD,MAAM,QAAQ,GAA8C,EAAE,CAAA;QAE9D,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,GAAoB,EAAQ,EAAE;YAC3D,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;QAC/B,CAAC,CAAA;QACD,KAAK,CAAC,OAAO,CAAC,+BAA+B,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,4BAA4B,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAC9G,KAAK,CAAC,OAAO,CAAC,8BAA8B,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,2BAA2B,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAC5G,KAAK,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAClG,KAAK,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAChG,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QACtG,KAAK,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAC1F,KAAK,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAC5F,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAE1F,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC,CAAA;QAEvD,oFAAoF;QACpF,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;QAC1C,KAAK,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,QAAQ,EAAE,CAAC;YACtC,MAAM,CAAC,GAAG,EAAE,SAAS,KAAK,yCAAyC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAClF,CAAC;IACH,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/multiagent/__tests__/graph.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=graph.test.d.ts.map

package/dist/src/multiagent/__tests__/graph.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"graph.test.d.ts","sourceRoot":"","sources":["../../../../src/multiagent/__tests__/graph.test.ts"],"names":[],"mappings":""}