@strands-agents/sdk 1.0.0-rc.5 → 1.1.0

package/dist/src/multiagent/events.d.ts

@@ -1,5 +1,5 @@
 import { HookableEvent, StreamEvent } from '../hooks/events.js';
-import type { AgentStreamEvent } from '../types/agent.js';
+import type { AgentStreamEvent, InvocationState } from '../types/agent.js';
 import type { MultiAgentResult, MultiAgentState, NodeResult } from './state.js';
 import type { MultiAgent } from './multiagent.js';
 import type { NodeType } from './nodes.js';
@@ -21,9 +21,11 @@ export declare class BeforeMultiAgentInvocationEvent extends HookableEvent {
     readonly type: "beforeMultiAgentInvocationEvent";
     readonly orchestrator: MultiAgent;
     readonly state: MultiAgentState;
+    readonly invocationState: InvocationState;
     constructor(data: {
         orchestrator: MultiAgent;
         state: MultiAgentState;
+        invocationState: InvocationState;
     });
     toJSON(): Pick<BeforeMultiAgentInvocationEvent, 'type'>;
 }
@@ -34,9 +36,11 @@ export declare class AfterMultiAgentInvocationEvent extends HookableEvent {
     readonly type: "afterMultiAgentInvocationEvent";
     readonly orchestrator: MultiAgent;
     readonly state: MultiAgentState;
+    readonly invocationState: InvocationState;
     constructor(data: {
         orchestrator: MultiAgent;
         state: MultiAgentState;
+        invocationState: InvocationState;
     });
     _shouldReverseCallbacks(): boolean;
     toJSON(): Pick<AfterMultiAgentInvocationEvent, 'type'>;
@@ -50,6 +54,7 @@ export declare class BeforeNodeCallEvent extends HookableEvent {
     readonly orchestrator: MultiAgent;
     readonly state: MultiAgentState;
     readonly nodeId: string;
+    readonly invocationState: InvocationState;
     /**
      * Set by hook callbacks to cancel node execution.
      * When set to `true`, a default cancel message is used.
@@ -60,6 +65,7 @@ export declare class BeforeNodeCallEvent extends HookableEvent {
         orchestrator: MultiAgent;
         state: MultiAgentState;
         nodeId: string;
+        invocationState: InvocationState;
     });
     toJSON(): Pick<BeforeNodeCallEvent, 'type' | 'nodeId'>;
 }
@@ -71,11 +77,13 @@ export declare class AfterNodeCallEvent extends HookableEvent {
     readonly orchestrator: MultiAgent;
     readonly state: MultiAgentState;
     readonly nodeId: string;
+    readonly invocationState: InvocationState;
     readonly error?: Error;
     constructor(data: {
         orchestrator: MultiAgent;
         state: MultiAgentState;
         nodeId: string;
+        invocationState: InvocationState;
         error?: Error;
     });
     _shouldReverseCallbacks(): boolean;
@@ -122,11 +130,13 @@ export declare class NodeStreamUpdateEvent extends HookableEvent {
     readonly nodeType: NodeType;
     readonly state: MultiAgentState;
     readonly inner: NodeStreamUpdateInnerEvent;
+    readonly invocationState: InvocationState;
     constructor(data: {
         nodeId: string;
         nodeType: NodeType;
         state: MultiAgentState;
         inner: NodeStreamUpdateInnerEvent;
+        invocationState: InvocationState;
     });
     toJSON(): Pick<NodeStreamUpdateEvent, 'type' | 'nodeId' | 'nodeType' | 'inner'>;
 }
@@ -140,11 +150,13 @@ export declare class NodeResultEvent extends HookableEvent {
     readonly nodeType: NodeType;
     readonly state: MultiAgentState;
     readonly result: NodeResult;
+    readonly invocationState: InvocationState;
     constructor(data: {
         nodeId: string;
         nodeType: NodeType;
         state: MultiAgentState;
         result: NodeResult;
+        invocationState: InvocationState;
     });
     toJSON(): Pick<NodeResultEvent, 'type' | 'nodeId' | 'nodeType' | 'result'>;
 }
@@ -156,10 +168,12 @@ export declare class MultiAgentHandoffEvent extends HookableEvent {
     readonly source: string;
     readonly targets: string[];
     readonly state: MultiAgentState;
+    readonly invocationState: InvocationState;
     constructor(data: {
         source: string;
         targets: string[];
         state: MultiAgentState;
+        invocationState: InvocationState;
     });
     toJSON(): Pick<MultiAgentHandoffEvent, 'type' | 'source' | 'targets'>;
 }
@@ -171,10 +185,12 @@ export declare class NodeCancelEvent extends HookableEvent {
     readonly nodeId: string;
     readonly state: MultiAgentState;
     readonly message: string;
+    readonly invocationState: InvocationState;
     constructor(data: {
         nodeId: string;
         state: MultiAgentState;
         message: string;
+        invocationState: InvocationState;
     });
     toJSON(): Pick<NodeCancelEvent, 'type' | 'nodeId' | 'message'>;
 }
@@ -185,8 +201,10 @@ export declare class NodeCancelEvent extends HookableEvent {
 export declare class MultiAgentResultEvent extends HookableEvent {
     readonly type: "multiAgentResultEvent";
     readonly result: MultiAgentResult;
+    readonly invocationState: InvocationState;
     constructor(data: {
         result: MultiAgentResult;
+        invocationState: InvocationState;
     });
     toJSON(): Pick<MultiAgentResultEvent, 'type' | 'result'>;
 }

package/dist/src/multiagent/events.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"events.d.ts","sourceRoot":"","sources":["../../../src/multiagent/events.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAC/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAA;AACzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AAC/E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AACjD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAE1C;;GAEG;AACH,qBAAa,0BAA2B,SAAQ,aAAa;IAC3D,QAAQ,CAAC,IAAI,EAAG,4BAA4B,CAAS;IACrD,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;gBAErB,IAAI,EAAE;QAAE,YAAY,EAAE,UAAU,CAAA;KAAE;IAK9C,MAAM,IAAI,IAAI,CAAC,0BAA0B,EAAE,MAAM,CAAC;CAGnD;AAED;;GAEG;AACH,qBAAa,+BAAgC,SAAQ,aAAa;IAChE,QAAQ,CAAC,IAAI,EAAG,iCAAiC,CAAS;IAC1D,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;IACjC,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;gBAEnB,IAAI,EAAE;QAAE,YAAY,EAAE,UAAU,CAAC;QAAC,KAAK,EAAE,eAAe,CAAA;KAAE;IAMtE,MAAM,IAAI,IAAI,CAAC,+BAA+B,EAAE,MAAM,CAAC;CAGxD;AAED;;GAEG;AACH,qBAAa,8BAA+B,SAAQ,aAAa;IAC/D,QAAQ,CAAC,IAAI,EAAG,gCAAgC,CAAS;IACzD,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;IACjC,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;gBAEnB,IAAI,EAAE;QAAE,YAAY,EAAE,UAAU,CAAC;QAAC,KAAK,EAAE,eAAe,CAAA;KAAE;IAM7D,uBAAuB,IAAI,OAAO;IAI3C,MAAM,IAAI,IAAI,CAAC,8BAA8B,EAAE,MAAM,CAAC;CAGvD;AAED;;;GAGG;AACH,qBAAa,mBAAoB,SAAQ,aAAa;IACpD,QAAQ,CAAC,IAAI,EAAG,qBAAqB,CAAS;IAC9C,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;IACjC,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IAEvB;;;;OAIG;IACH,MAAM,EAAE,OAAO,GAAG,MAAM,CAAQ;gBAEpB,IAAI,EAAE;QAAE,YAAY,EAAE,UAAU,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE;IAOtF,MAAM,IAAI,IAAI,CAAC,mBAAmB,EAAE,MAAM,GAAG,QAAQ,CAAC;CAGvD;AAED;;GAEG;AACH,qBAAa,kBAAmB,SAAQ,aAAa;IACnD,QAAQ,CAAC,IAAI,EAAG,oBAAoB,CAAS;IAC7C,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;IACjC,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,KAAK,CAAC,EAAE,KAAK,CAAA;gBAEV,IAAI,EAAE;QAAE,YAAY,EAAE,UAAU,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,KAAK,CAAA;KAAE;IAU5F,uBAAuB,IAAI,OAAO;IAI3C,MAAM,IAAI,IAAI,CAAC,kBAAkB,EAAE,MAAM,GAAG,QAAQ,CAAC,GAAG;QAAE,KAAK,CAAC,EAAE;YAAE,OAAO,CAAC,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE;CAOzF;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,MAAM,0BAA0B,GAClC;IAAE,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,KAAK,EAAE,gBAAgB,CAAA;CAAE,GAC9D;IAAE,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;IAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,qBAAqB,EAAE,qBAAqB,CAAC,CAAA;CAAE,GACxG;IAAE,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC;IAAC,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAA;CAAE,CAAA;AAE9D;;;;GAIG;AACH,qBAAa,qBAAsB,SAAQ,aAAa;IACtD,QAAQ,CAAC,IAAI,EAAG,uBAAuB,CAAS;IAChD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAA;IAC3B,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,KAAK,EAAE,0BAA0B,CAAA;gBAE9B,IAAI,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,QAAQ,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,KAAK,EAAE,0BAA0B,CAAA;KAAE;IAQnH,MAAM,IAAI,IAAI,CAAC,qBAAqB,EAAE,MAAM,GAAG,QAAQ,GAAG,UAAU,GAAG,OAAO,CAAC;CAGhF;AAED;;;GAGG;AACH,qBAAa,eAAgB,SAAQ,aAAa;IAChD,QAAQ,CAAC,IAAI,EAAG,iBAAiB,CAAS;IAC1C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAA;IAC3B,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAA;gBAEf,IAAI,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,QAAQ,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,MAAM,EAAE,UAAU,CAAA;KAAE;IAQpG,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,MAAM,GAAG,QAAQ,GAAG,UAAU,GAAG,QAAQ,CAAC;CAG3E;AAED;;GAEG;AACH,qBAAa,sBAAuB,SAAQ,aAAa;IACvD,QAAQ,CAAC,IAAI,EAAG,wBAAwB,CAAS;IACjD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,CAAA;IAC1B,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;gBAEnB,IAAI,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,EAAE,CAAC;QAAC,KAAK,EAAE,eAAe,CAAA;KAAE;IAO/E,MAAM,IAAI,IAAI,CAAC,sBAAsB,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC;CAGtE;AAED;;GAEG;AACH,qBAAa,eAAgB,SAAQ,aAAa;IAChD,QAAQ,CAAC,IAAI,EAAG,iBAAiB,CAAS;IAC1C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;gBAEZ,IAAI,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE;IAO7E,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC;CAG/D;AAED;;;GAGG;AACH,qBAAa,qBAAsB,SAAQ,aAAa;IACtD,QAAQ,CAAC,IAAI,EAAG,uBAAuB,CAAS;IAChD,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAA;gBAErB,IAAI,EAAE;QAAE,MAAM,EAAE,gBAAgB,CAAA;KAAE;IAK9C,MAAM,IAAI,IAAI,CAAC,qBAAqB,EAAE,MAAM,GAAG,QAAQ,CAAC;CAGzD;AAED;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAC7B,+BAA+B,GAC/B,8BAA8B,GAC9B,mBAAmB,GACnB,kBAAkB,GAClB,qBAAqB,GACrB,eAAe,GACf,eAAe,GACf,sBAAsB,GACtB,qBAAqB,CAAA"}
+{"version":3,"file":"events.d.ts","sourceRoot":"","sources":["../../../src/multiagent/events.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAC/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAA;AAC1E,OAAO,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AAC/E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AACjD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAE1C;;GAEG;AACH,qBAAa,0BAA2B,SAAQ,aAAa;IAC3D,QAAQ,CAAC,IAAI,EAAG,4BAA4B,CAAS;IACrD,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;gBAErB,IAAI,EAAE;QAAE,YAAY,EAAE,UAAU,CAAA;KAAE;IAK9C,MAAM,IAAI,IAAI,CAAC,0BAA0B,EAAE,MAAM,CAAC;CAGnD;AAED;;GAEG;AACH,qBAAa,+BAAgC,SAAQ,aAAa;IAChE,QAAQ,CAAC,IAAI,EAAG,iCAAiC,CAAS;IAC1D,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;IACjC,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;gBAE7B,IAAI,EAAE;QAAE,YAAY,EAAE,UAAU,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,eAAe,EAAE,eAAe,CAAA;KAAE;IAOxG,MAAM,IAAI,IAAI,CAAC,+BAA+B,EAAE,MAAM,CAAC;CAGxD;AAED;;GAEG;AACH,qBAAa,8BAA+B,SAAQ,aAAa;IAC/D,QAAQ,CAAC,IAAI,EAAG,gCAAgC,CAAS;IACzD,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;IACjC,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;gBAE7B,IAAI,EAAE;QAAE,YAAY,EAAE,UAAU,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,eAAe,EAAE,eAAe,CAAA;KAAE;IAO/F,uBAAuB,IAAI,OAAO;IAI3C,MAAM,IAAI,IAAI,CAAC,8BAA8B,EAAE,MAAM,CAAC;CAGvD;AAED;;;GAGG;AACH,qBAAa,mBAAoB,SAAQ,aAAa;IACpD,QAAQ,CAAC,IAAI,EAAG,qBAAqB,CAAS;IAC9C,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;IACjC,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;IAEzC;;;;OAIG;IACH,MAAM,EAAE,OAAO,GAAG,MAAM,CAAQ;gBAEpB,IAAI,EAAE;QAChB,YAAY,EAAE,UAAU,CAAA;QACxB,KAAK,EAAE,eAAe,CAAA;QACtB,MAAM,EAAE,MAAM,CAAA;QACd,eAAe,EAAE,eAAe,CAAA;KACjC;IAQD,MAAM,IAAI,IAAI,CAAC,mBAAmB,EAAE,MAAM,GAAG,QAAQ,CAAC;CAGvD;AAED;;GAEG;AACH,qBAAa,kBAAmB,SAAQ,aAAa;IACnD,QAAQ,CAAC,IAAI,EAAG,oBAAoB,CAAS;IAC7C,QAAQ,CAAC,YAAY,EAAE,UAAU,CAAA;IACjC,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;IACzC,QAAQ,CAAC,KAAK,CAAC,EAAE,KAAK,CAAA;gBAEV,IAAI,EAAE;QAChB,YAAY,EAAE,UAAU,CAAA;QACxB,KAAK,EAAE,eAAe,CAAA;QACtB,MAAM,EAAE,MAAM,CAAA;QACd,eAAe,EAAE,eAAe,CAAA;QAChC,KAAK,CAAC,EAAE,KAAK,CAAA;KACd;IAWQ,uBAAuB,IAAI,OAAO;IAI3C,MAAM,IAAI,IAAI,CAAC,kBAAkB,EAAE,MAAM,GAAG,QAAQ,CAAC,GAAG;QAAE,KAAK,CAAC,EAAE;YAAE,OAAO,CAAC,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE;CAOzF;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,MAAM,0BAA0B,GAClC;IAAE,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,KAAK,EAAE,gBAAgB,CAAA;CAAE,GAC9D;IAAE,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC;IAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,qBAAqB,EAAE,qBAAqB,CAAC,CAAA;CAAE,GACxG;IAAE,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC;IAAC,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAA;CAAE,CAAA;AAE9D;;;;GAIG;AACH,qBAAa,qBAAsB,SAAQ,aAAa;IACtD,QAAQ,CAAC,IAAI,EAAG,uBAAuB,CAAS;IAChD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAA;IAC3B,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,KAAK,EAAE,0BAA0B,CAAA;IAC1C,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;gBAE7B,IAAI,EAAE;QAChB,MAAM,EAAE,MAAM,CAAA;QACd,QAAQ,EAAE,QAAQ,CAAA;QAClB,KAAK,EAAE,eAAe,CAAA;QACtB,KAAK,EAAE,0BAA0B,CAAA;QACjC,eAAe,EAAE,eAAe,CAAA;KACjC;IASD,MAAM,IAAI,IAAI,CAAC,qBAAqB,EAAE,MAAM,GAAG,QAAQ,GAAG,UAAU,GAAG,OAAO,CAAC;CAGhF;AAED;;;GAGG;AACH,qBAAa,eAAgB,SAAQ,aAAa;IAChD,QAAQ,CAAC,IAAI,EAAG,iBAAiB,CAAS;IAC1C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAA;IAC3B,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAA;IAC3B,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;gBAE7B,IAAI,EAAE;QAChB,MAAM,EAAE,MAAM,CAAA;QACd,QAAQ,EAAE,QAAQ,CAAA;QAClB,KAAK,EAAE,eAAe,CAAA;QACtB,MAAM,EAAE,UAAU,CAAA;QAClB,eAAe,EAAE,eAAe,CAAA;KACjC;IASD,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,MAAM,GAAG,QAAQ,GAAG,UAAU,GAAG,QAAQ,CAAC;CAG3E;AAED;;GAEG;AACH,qBAAa,sBAAuB,SAAQ,aAAa;IACvD,QAAQ,CAAC,IAAI,EAAG,wBAAwB,CAAS;IACjD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,CAAA;IAC1B,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;gBAE7B,IAAI,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,EAAE,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,eAAe,EAAE,eAAe,CAAA;KAAE;IAQjH,MAAM,IAAI,IAAI,CAAC,sBAAsB,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC;CAGtE;AAED;;GAEG;AACH,qBAAa,eAAgB,SAAQ,aAAa;IAChD,QAAQ,CAAC,IAAI,EAAG,iBAAiB,CAAS;IAC1C,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,KAAK,EAAE,eAAe,CAAA;IAC/B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;IACxB,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;gBAE7B,IAAI,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,eAAe,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,eAAe,EAAE,eAAe,CAAA;KAAE;IAQ/G,MAAM,IAAI,IAAI,CAAC,eAAe,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC;CAG/D;AAED;;;GAGG;AACH,qBAAa,qBAAsB,SAAQ,aAAa;IACtD,QAAQ,CAAC,IAAI,EAAG,uBAAuB,CAAS;IAChD,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAA;IACjC,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;gBAE7B,IAAI,EAAE;QAAE,MAAM,EAAE,gBAAgB,CAAC;QAAC,eAAe,EAAE,eAAe,CAAA;KAAE;IAMhF,MAAM,IAAI,IAAI,CAAC,qBAAqB,EAAE,MAAM,GAAG,QAAQ,CAAC;CAGzD;AAED;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAC7B,+BAA+B,GAC/B,8BAA8B,GAC9B,mBAAmB,GACnB,kBAAkB,GAClB,qBAAqB,GACrB,eAAe,GACf,eAAe,GACf,sBAAsB,GACtB,qBAAqB,CAAA"}

package/dist/src/multiagent/events.js

@@ -20,10 +20,12 @@ export class BeforeMultiAgentInvocationEvent extends HookableEvent {
     type = 'beforeMultiAgentInvocationEvent';
     orchestrator;
     state;
+    invocationState;
     constructor(data) {
         super();
         this.orchestrator = data.orchestrator;
         this.state = data.state;
+        this.invocationState = data.invocationState;
     }
     toJSON() {
         return { type: this.type };
@@ -36,10 +38,12 @@ export class AfterMultiAgentInvocationEvent extends HookableEvent {
     type = 'afterMultiAgentInvocationEvent';
     orchestrator;
     state;
+    invocationState;
     constructor(data) {
         super();
         this.orchestrator = data.orchestrator;
         this.state = data.state;
+        this.invocationState = data.invocationState;
     }
     _shouldReverseCallbacks() {
         return true;
@@ -57,6 +61,7 @@ export class BeforeNodeCallEvent extends HookableEvent {
     orchestrator;
     state;
     nodeId;
+    invocationState;
     /**
      * Set by hook callbacks to cancel node execution.
      * When set to `true`, a default cancel message is used.
@@ -68,6 +73,7 @@ export class BeforeNodeCallEvent extends HookableEvent {
         this.orchestrator = data.orchestrator;
         this.state = data.state;
         this.nodeId = data.nodeId;
+        this.invocationState = data.invocationState;
     }
     toJSON() {
         return { type: this.type, nodeId: this.nodeId };
@@ -81,12 +87,14 @@ export class AfterNodeCallEvent extends HookableEvent {
     orchestrator;
     state;
     nodeId;
+    invocationState;
     error;
     constructor(data) {
         super();
         this.orchestrator = data.orchestrator;
         this.state = data.state;
         this.nodeId = data.nodeId;
+        this.invocationState = data.invocationState;
         if (data.error !== undefined) {
             this.error = data.error;
         }
@@ -113,12 +121,14 @@ export class NodeStreamUpdateEvent extends HookableEvent {
     nodeType;
     state;
     inner;
+    invocationState;
     constructor(data) {
         super();
         this.nodeId = data.nodeId;
         this.nodeType = data.nodeType;
         this.state = data.state;
         this.inner = data.inner;
+        this.invocationState = data.invocationState;
     }
     toJSON() {
         return { type: this.type, nodeId: this.nodeId, nodeType: this.nodeType, inner: this.inner };
@@ -134,12 +144,14 @@ export class NodeResultEvent extends HookableEvent {
     nodeType;
     state;
     result;
+    invocationState;
     constructor(data) {
         super();
         this.nodeId = data.nodeId;
         this.nodeType = data.nodeType;
         this.state = data.state;
         this.result = data.result;
+        this.invocationState = data.invocationState;
     }
     toJSON() {
         return { type: this.type, nodeId: this.nodeId, nodeType: this.nodeType, result: this.result };
@@ -153,11 +165,13 @@ export class MultiAgentHandoffEvent extends HookableEvent {
     source;
     targets;
     state;
+    invocationState;
     constructor(data) {
         super();
         this.source = data.source;
         this.targets = data.targets;
         this.state = data.state;
+        this.invocationState = data.invocationState;
     }
     toJSON() {
         return { type: this.type, source: this.source, targets: this.targets };
@@ -171,11 +185,13 @@ export class NodeCancelEvent extends HookableEvent {
     nodeId;
     state;
     message;
+    invocationState;
     constructor(data) {
         super();
         this.nodeId = data.nodeId;
         this.state = data.state;
         this.message = data.message;
+        this.invocationState = data.invocationState;
     }
     toJSON() {
         return { type: this.type, nodeId: this.nodeId, message: this.message };
@@ -188,9 +204,11 @@ export class NodeCancelEvent extends HookableEvent {
 export class MultiAgentResultEvent extends HookableEvent {
     type = 'multiAgentResultEvent';
     result;
+    invocationState;
     constructor(data) {
         super();
         this.result = data.result;
+        this.invocationState = data.invocationState;
     }
     toJSON() {
         return { type: this.type, result: this.result };

package/dist/src/multiagent/events.js.map

@@ -1 +1 @@
-{"version":3,"file":"events.js","sourceRoot":"","sources":["../../../src/multiagent/events.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAM/D;;GAEG;AACH,MAAM,OAAO,0BAA2B,SAAQ,aAAa;IAClD,IAAI,GAAG,4BAAqC,CAAA;IAC5C,YAAY,CAAY;IAEjC,YAAY,IAAkC;QAC5C,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;IACvC,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAA;IAC5B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,+BAAgC,SAAQ,aAAa;IACvD,IAAI,GAAG,iCAA0C,CAAA;IACjD,YAAY,CAAY;IACxB,KAAK,CAAiB;IAE/B,YAAY,IAA0D;QACpE,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QACrC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;IACzB,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAA;IAC5B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,8BAA+B,SAAQ,aAAa;IACtD,IAAI,GAAG,gCAAyC,CAAA;IAChD,YAAY,CAAY;IACxB,KAAK,CAAiB;IAE/B,YAAY,IAA0D;QACpE,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QACrC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;IACzB,CAAC;IAEQ,uBAAuB;QAC9B,OAAO,IAAI,CAAA;IACb,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAA;IAC5B,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,mBAAoB,SAAQ,aAAa;IAC3C,IAAI,GAAG,qBAA8B,CAAA;IACrC,YAAY,CAAY;IACxB,KAAK,CAAiB;IACtB,MAAM,CAAQ;IAEvB;;;;OAIG;IACH,MAAM,GAAqB,KAAK,CAAA;IAEhC,YAAY,IAA0E;QACpF,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QACrC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;IAC3B,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;IACjD,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,kBAAmB,SAAQ,aAAa;IAC1C,IAAI,GAAG,oBAA6B,CAAA;IACpC,YAAY,CAAY;IACxB,KAAK,CAAiB;IACtB,MAAM,CAAQ;IACd,KAAK,CAAQ;IAEtB,YAAY,IAAyF;QACnG,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QACrC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACzB,CAAC;IACH,CAAC;IAEQ,uBAAuB;QAC9B,OAAO,IAAI,CAAA;IACb,CAAC;IAED,MAAM;QACJ,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,GAAG,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;SAC5E,CAAA;IACH,CAAC;CACF;AAuBD;;;;GAIG;AACH,MAAM,OAAO,qBAAsB,SAAQ,aAAa;IAC7C,IAAI,GAAG,uBAAgC,CAAA;IACvC,MAAM,CAAQ;IACd,QAAQ,CAAU;IAClB,KAAK,CAAiB;IACtB,KAAK,CAA4B;IAE1C,YAAY,IAAuG;QACjH,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAA;QAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;IACzB,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAA;IAC7F,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,eAAgB,SAAQ,aAAa;IACvC,IAAI,GAAG,iBAA0B,CAAA;IACjC,MAAM,CAAQ;IACd,QAAQ,CAAU;IAClB,KAAK,CAAiB;IACtB,MAAM,CAAY;IAE3B,YAAY,IAAwF;QAClG,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAA;QAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;IAC3B,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;IAC/F,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,sBAAuB,SAAQ,aAAa;IAC9C,IAAI,GAAG,wBAAiC,CAAA;IACxC,MAAM,CAAQ;IACd,OAAO,CAAU;IACjB,KAAK,CAAiB;IAE/B,YAAY,IAAmE;QAC7E,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAA;QAC3B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;IACzB,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAA;IACxE,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,eAAgB,SAAQ,aAAa;IACvC,IAAI,GAAG,iBAA0B,CAAA;IACjC,MAAM,CAAQ;IACd,KAAK,CAAiB;IACtB,OAAO,CAAQ;IAExB,YAAY,IAAiE;QAC3E,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAA;IAC7B,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAA;IACxE,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,qBAAsB,SAAQ,aAAa;IAC7C,IAAI,GAAG,uBAAgC,CAAA;IACvC,MAAM,CAAkB;IAEjC,YAAY,IAAkC;QAC5C,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;IAC3B,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;IACjD,CAAC;CACF"}
+{"version":3,"file":"events.js","sourceRoot":"","sources":["../../../src/multiagent/events.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAM/D;;GAEG;AACH,MAAM,OAAO,0BAA2B,SAAQ,aAAa;IAClD,IAAI,GAAG,4BAAqC,CAAA;IAC5C,YAAY,CAAY;IAEjC,YAAY,IAAkC;QAC5C,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;IACvC,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAA;IAC5B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,+BAAgC,SAAQ,aAAa;IACvD,IAAI,GAAG,iCAA0C,CAAA;IACjD,YAAY,CAAY;IACxB,KAAK,CAAiB;IACtB,eAAe,CAAiB;IAEzC,YAAY,IAA4F;QACtG,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QACrC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC7C,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAA;IAC5B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,8BAA+B,SAAQ,aAAa;IACtD,IAAI,GAAG,gCAAyC,CAAA;IAChD,YAAY,CAAY;IACxB,KAAK,CAAiB;IACtB,eAAe,CAAiB;IAEzC,YAAY,IAA4F;QACtG,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QACrC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC7C,CAAC;IAEQ,uBAAuB;QAC9B,OAAO,IAAI,CAAA;IACb,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAA;IAC5B,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,mBAAoB,SAAQ,aAAa;IAC3C,IAAI,GAAG,qBAA8B,CAAA;IACrC,YAAY,CAAY;IACxB,KAAK,CAAiB;IACtB,MAAM,CAAQ;IACd,eAAe,CAAiB;IAEzC;;;;OAIG;IACH,MAAM,GAAqB,KAAK,CAAA;IAEhC,YAAY,IAKX;QACC,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QACrC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC7C,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;IACjD,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,kBAAmB,SAAQ,aAAa;IAC1C,IAAI,GAAG,oBAA6B,CAAA;IACpC,YAAY,CAAY;IACxB,KAAK,CAAiB;IACtB,MAAM,CAAQ;IACd,eAAe,CAAiB;IAChC,KAAK,CAAQ;IAEtB,YAAY,IAMX;QACC,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QACrC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;QAC3C,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACzB,CAAC;IACH,CAAC;IAEQ,uBAAuB;QAC9B,OAAO,IAAI,CAAA;IACb,CAAC;IAED,MAAM;QACJ,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,GAAG,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;SAC5E,CAAA;IACH,CAAC;CACF;AAuBD;;;;GAIG;AACH,MAAM,OAAO,qBAAsB,SAAQ,aAAa;IAC7C,IAAI,GAAG,uBAAgC,CAAA;IACvC,MAAM,CAAQ;IACd,QAAQ,CAAU;IAClB,KAAK,CAAiB;IACtB,KAAK,CAA4B;IACjC,eAAe,CAAiB;IAEzC,YAAY,IAMX;QACC,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAA;QAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC7C,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAA;IAC7F,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,eAAgB,SAAQ,aAAa;IACvC,IAAI,GAAG,iBAA0B,CAAA;IACjC,MAAM,CAAQ;IACd,QAAQ,CAAU;IAClB,KAAK,CAAiB;IACtB,MAAM,CAAY;IAClB,eAAe,CAAiB;IAEzC,YAAY,IAMX;QACC,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAA;QAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC7C,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;IAC/F,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,sBAAuB,SAAQ,aAAa;IAC9C,IAAI,GAAG,wBAAiC,CAAA;IACxC,MAAM,CAAQ;IACd,OAAO,CAAU;IACjB,KAAK,CAAiB;IACtB,eAAe,CAAiB;IAEzC,YAAY,IAAqG;QAC/G,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAA;QAC3B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC7C,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAA;IACxE,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,eAAgB,SAAQ,aAAa;IACvC,IAAI,GAAG,iBAA0B,CAAA;IACjC,MAAM,CAAQ;IACd,KAAK,CAAiB;IACtB,OAAO,CAAQ;IACf,eAAe,CAAiB;IAEzC,YAAY,IAAmG;QAC7G,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;QACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAA;QAC3B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC7C,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAA;IACxE,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,qBAAsB,SAAQ,aAAa;IAC7C,IAAI,GAAG,uBAAgC,CAAA;IACvC,MAAM,CAAkB;IACxB,eAAe,CAAiB;IAEzC,YAAY,IAAoE;QAC9E,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAA;IAC7C,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;IACjD,CAAC;CACF"}

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

@@ -1,5 +1,5 @@
 import type { AttributeValue } from '@opentelemetry/api';
-import type { MultiAgentInput } from './multiagent.js';
+import type { MultiAgentInput, MultiAgentInvokeOptions } from './multiagent.js';
 import { HookableEvent } from '../hooks/events.js';
 import type { HookCallback, HookableEventConstructor, HookCleanup } from '../hooks/types.js';
 import type { MultiAgentPlugin } from './plugins.js';
@@ -15,10 +15,30 @@ import { Edge } from './edge.js';
  * Runtime configuration for graph execution.
  */
 export interface GraphConfig {
-    /** Max nodes executing in parallel. */
+    /** Max nodes executing in parallel. Defaults to `Infinity` (no limit). */
     maxConcurrency?: number;
-    /** Max total steps (prevents infinite loops in cyclic graphs). */
+    /** Max total steps (prevents infinite loops in cyclic graphs). Defaults to `Infinity` (no limit). */
     maxSteps?: number;
+    /**
+     * Wall-clock ceiling for the entire graph invocation, in milliseconds. Defaults to `Infinity`
+     * (no limit).
+     *
+     * Does not propagate into nested orchestrators wrapped via `MultiAgentNode` — a nested
+     * `Swarm`/`Graph` runs to completion under its own timeout config; the parent graph's
+     * timeout only fires once the nested node returns.
+     */
+    timeout?: number;
+    /**
+     * Fallback per-node wall-clock ceiling in milliseconds. Applied to any `AgentNode` that
+     * doesn't set its own `timeout`. Defaults to `Infinity` (no limit).
+     *
+     * Does not apply to `MultiAgentNode`. Set `timeout`/`nodeTimeout` on the nested
+     * orchestrator to bound it.
+     *
+     * Enforced via `AbortSignal` — cancellation is cooperative, so a tool that neither polls
+     * its cancel signal nor forwards it to a cancellable API can run past this deadline.
+     */
+    nodeTimeout?: number;
 }
 /**
  * Options for creating a Graph instance.
@@ -94,9 +114,10 @@ export declare class Graph implements MultiAgent {
      * Invoke graph and return final result (consumes stream).
      *
      * @param input - The input to pass to entry point nodes
+     * @param options - Optional per-invocation options (e.g., {@link InvocationState})
      * @returns Promise resolving to the final MultiAgentResult
      */
-    invoke(input: MultiAgentInput): Promise<MultiAgentResult>;
+    invoke(input: MultiAgentInput, options?: MultiAgentInvokeOptions): Promise<MultiAgentResult>;
     /**
      * Register a hook callback for a specific graph event type.
      *
@@ -110,9 +131,10 @@ export declare class Graph implements MultiAgent {
      * Invokes hook callbacks for each event before yielding.
      *
      * @param input - The input to pass to entry nodes
+     * @param options - Optional per-invocation options (e.g., {@link InvocationState})
      * @returns Async generator yielding streaming events and returning a MultiAgentResult
      */
-    stream(input: MultiAgentInput): AsyncGenerator<MultiAgentStreamEvent, MultiAgentResult, undefined>;
+    stream(input: MultiAgentInput, options?: MultiAgentInvokeOptions): AsyncGenerator<MultiAgentStreamEvent, MultiAgentResult, undefined>;
     private _stream;
     /**
      * Executes a single node, pushing streaming events to the shared queue in real-time.

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

@@ -1 +1 @@
-{"version":3,"file":"graph.d.ts","sourceRoot":"","sources":["../../../src/multiagent/graph.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAA;AAExD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAA;AAItD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAElD,OAAO,KAAK,EAAE,YAAY,EAAE,wBAAwB,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAC5F,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAA;AACpD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAA;AAEnE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAChD,OAAO,EAA6B,IAAI,EAAE,MAAM,YAAY,CAAA;AAC5D,OAAO,EAAmB,gBAAgB,EAAsB,MAAM,YAAY,CAAA;AAClF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAEjD,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAA;AAWxD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAC/C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAMhC;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,uCAAuC;IACvC,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,kEAAkE;IAClE,QAAQ,CAAC,EAAE,MAAM,CAAA;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,YAAa,SAAQ,WAAW;IAC/C,+DAA+D;IAC/D,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,oDAAoD;IACpD,KAAK,EAAE,cAAc,EAAE,CAAA;IACvB,6DAA6D;IAC7D,KAAK,EAAE,cAAc,EAAE,CAAA;IACvB,6FAA6F;IAC7F,OAAO,CAAC,EAAE,MAAM,EAAE,CAAA;IAClB,+DAA+D;IAC/D,cAAc,CAAC,EAAE,cAAc,CAAA;IAC/B,8CAA8C;IAC9C,OAAO,CAAC,EAAE,gBAAgB,EAAE,CAAA;IAC5B,uDAAuD;IACvD,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;CACjD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,qBAAa,KAAM,YAAW,UAAU;IACtC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAA;IACnB,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;IACzC,QAAQ,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,CAAA;IAC/B,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAA;IACtC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAA0B;IAC1D,OAAO,CAAC,QAAQ,CAAC,aAAa,CAA4B;IAC1D,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAQ;IACjC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAQ;IAChC,QAAQ,CAAC,cAAc,CAAC,EAAE,cAAc,GAAG,SAAS,CAAA;IACpD,OAAO,CAAC,YAAY,CAAS;gBAEjB,OAAO,EAAE,YAAY;IA+BjC;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAOjC;;;;;OAKG;IACG,MAAM,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAS/D;;;;;;OAMG;IACH,OAAO,CAAC,CAAC,SAAS,aAAa,EAAE,SAAS,EAAE,wBAAwB,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,GAAG,WAAW;IAIhH;;;;;;OAMG;IACI,MAAM,CAAC,KAAK,EAAE,eAAe,GAAG,cAAc,CAAC,qBAAqB,EAAE,gBAAgB,EAAE,SAAS,CAAC;YAmB1F,OAAO;IA0FtB;;OAEG;YACW,WAAW;IA6EzB,OAAO,CAAC,eAAe;IASvB,OAAO,CAAC,gBAAgB;IA8BxB,OAAO,CAAC,aAAa;IA2BrB,OAAO,CAAC,aAAa;IAoBrB,OAAO,CAAC,eAAe;IAiBvB;;;;OAIG;IACH,OAAO,CAAC,eAAe;IAavB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAkBzB;;;;;;;;;;;;;;OAcG;YACW,kBAAkB;IA6BhC;;OAEG;YACW,yBAAyB;IAQvC,OAAO,CAAC,WAAW;IAMnB;;;;;;;;;OASG;YACW,UAAU;CAsBzB"}
+{"version":3,"file":"graph.d.ts","sourceRoot":"","sources":["../../../src/multiagent/graph.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAA;AAExD,OAAO,KAAK,EAAE,eAAe,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAA;AAK/E,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAElD,OAAO,KAAK,EAAE,YAAY,EAAE,wBAAwB,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAC5F,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAA;AACpD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAA;AAEnE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAChD,OAAO,EAA6B,IAAI,EAAE,MAAM,YAAY,CAAA;AAC5D,OAAO,EAAmB,gBAAgB,EAAsB,MAAM,YAAY,CAAA;AAClF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAEjD,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAA;AAWxD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAC/C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAMhC;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,0EAA0E;IAC1E,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,qGAAqG;IACrG,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB;;;;;;;OAOG;IACH,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB;;;;;;;;;OASG;IACH,WAAW,CAAC,EAAE,MAAM,CAAA;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,YAAa,SAAQ,WAAW;IAC/C,+DAA+D;IAC/D,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,oDAAoD;IACpD,KAAK,EAAE,cAAc,EAAE,CAAA;IACvB,6DAA6D;IAC7D,KAAK,EAAE,cAAc,EAAE,CAAA;IACvB,6FAA6F;IAC7F,OAAO,CAAC,EAAE,MAAM,EAAE,CAAA;IAClB,+DAA+D;IAC/D,cAAc,CAAC,EAAE,cAAc,CAAA;IAC/B,8CAA8C;IAC9C,OAAO,CAAC,EAAE,gBAAgB,EAAE,CAAA;IAC5B,uDAAuD;IACvD,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;CACjD;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,qBAAa,KAAM,YAAW,UAAU;IACtC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAA;IACnB,QAAQ,CAAC,KAAK,EAAE,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;IACzC,QAAQ,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,CAAA;IAC/B,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAA;IACtC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAA0B;IAC1D,OAAO,CAAC,QAAQ,CAAC,aAAa,CAA4B;IAC1D,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAQ;IACjC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAQ;IAChC,QAAQ,CAAC,cAAc,CAAC,EAAE,cAAc,GAAG,SAAS,CAAA;IACpD,OAAO,CAAC,YAAY,CAAS;gBAEjB,OAAO,EAAE,YAAY;IAqCjC;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAOjC;;;;;;OAMG;IACG,MAAM,CAAC,KAAK,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IASlG;;;;;;OAMG;IACH,OAAO,CAAC,CAAC,SAAS,aAAa,EAAE,SAAS,EAAE,wBAAwB,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,GAAG,WAAW;IAIhH;;;;;;;OAOG;IACI,MAAM,CACX,KAAK,EAAE,eAAe,EACtB,OAAO,CAAC,EAAE,uBAAuB,GAChC,cAAc,CAAC,qBAAqB,EAAE,gBAAgB,EAAE,SAAS,CAAC;YAuBtD,OAAO;IA2GtB;;OAEG;YACW,WAAW;IAsGzB,OAAO,CAAC,eAAe;IAevB,OAAO,CAAC,gBAAgB;IA8BxB,OAAO,CAAC,aAAa;IA2BrB,OAAO,CAAC,aAAa;IAoBrB,OAAO,CAAC,eAAe;IAiBvB;;;;OAIG;IACH,OAAO,CAAC,eAAe;IAavB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAkBzB;;;;;;;;;;;;;;OAcG;YACW,kBAAkB;IA6BhC;;OAEG;YACW,yBAAyB;IAQvC,OAAO,CAAC,WAAW;IAMnB;;;;;;;;;OASG;YACW,UAAU;CAsBzB"}

package/dist/src/multiagent/graph.js

@@ -1,5 +1,6 @@
 import { TextBlock, contentBlockFromData } from '../types/messages.js';
 import { logger } from '../logging/logger.js';
+import { warnOnce } from '../logging/warn-once.js';
 import { HookableEvent } from '../hooks/events.js';
 import { HookRegistryImplementation } from '../hooks/registry.js';
 import { MultiAgentPluginRegistry } from './plugins.js';
@@ -62,8 +63,13 @@ export class Graph {
         this.config = {
             maxConcurrency: config.maxConcurrency ?? Infinity,
             maxSteps: config.maxSteps ?? Infinity,
+            timeout: config.timeout ?? Infinity,
+            nodeTimeout: config.nodeTimeout ?? Infinity,
         };
         this._validateConfig();
+        if (this.config.maxSteps === Infinity && this.config.timeout === Infinity) {
+            warnOnce(logger, 'graph has no maxSteps or timeout set; execution is unbounded');
+        }
         this.nodes = this._resolveNodes(nodes);
         this.edges = this._resolveEdges(edges);
         this._sources = this._resolveSources(sources);
@@ -95,10 +101,11 @@ export class Graph {
      * Invoke graph and return final result (consumes stream).
      *
      * @param input - The input to pass to entry point nodes
+     * @param options - Optional per-invocation options (e.g., {@link InvocationState})
      * @returns Promise resolving to the final MultiAgentResult
      */
-    async invoke(input) {
-        const gen = this.stream(input);
+    async invoke(input, options) {
+        const gen = this.stream(input, options);
         let next = await gen.next();
         while (!next.done) {
             next = await gen.next();
@@ -120,11 +127,15 @@ export class Graph {
      * Invokes hook callbacks for each event before yielding.
      *
      * @param input - The input to pass to entry nodes
+     * @param options - Optional per-invocation options (e.g., {@link InvocationState})
      * @returns Async generator yielding streaming events and returning a MultiAgentResult
      */
-    async *stream(input) {
+    async *stream(input, options) {
         await this.initialize();
-        const gen = this._stream(input);
+        // Resolve invocationState once; the same object is threaded to every node's
+        // child agent so mutations in one node are visible in the next.
+        const invocationState = options?.invocationState ?? {};
+        const gen = this._stream(input, invocationState);
         try {
             let next = await gen.next();
             while (!next.done) {
@@ -140,7 +151,7 @@ export class Graph {
             await gen.return(undefined);
         }
     }
-    async *_stream(input) {
+    async *_stream(input, invocationState) {
         const state = new MultiAgentState({ nodeIds: [...this.nodes.keys()] });
         const queue = new Queue();
         const streams = new Map();
@@ -150,18 +161,26 @@ export class Graph {
             input,
         });
         // SessionManager (or plugins) may restore state.results here via the hook
-        yield new BeforeMultiAgentInvocationEvent({ orchestrator: this, state });
+        yield new BeforeMultiAgentInvocationEvent({ orchestrator: this, state, invocationState });
         // Resume: if state was restored, find nodes that are ready but haven't completed otherwise start from source nodes
         const targets = (await this._findResumeTargets(state)) ?? [...this._sources];
+        // Execution timeout: when fired, cancels all in-flight node invocations via their
+        // composed cancel signal. In-flight nodes return `stopReason: 'cancelled'`; the queue
+        // drains and the outer loop throws below when it sees the signal aborted.
+        const execController = Number.isFinite(this.config.timeout) ? new AbortController() : undefined;
+        const execTimeoutHandle = execController ? setTimeout(() => execController.abort(), this.config.timeout) : undefined;
         let caughtError;
         let result;
         try {
             while (targets.length > 0 || streams.size > 0) {
+                if (execController?.signal.aborted) {
+                    throw new Error(`timeout=<${this.config.timeout}>, graph_id=<${this.id}> | graph exceeded wall-clock budget`);
+                }
                 while (targets.length > 0 && streams.size < this.config.maxConcurrency) {
                     const node = targets.shift();
                     this._checkSteps(state);
                     state.steps++;
-                    streams.set(node.id, this._streamNode(node, input, state, queue, multiAgentSpan));
+                    streams.set(node.id, this._streamNode(node, input, state, queue, multiAgentSpan, invocationState, execController?.signal));
                 }
                 await queue.wait();
                 while (queue.size > 0) {
@@ -186,6 +205,7 @@ export class Graph {
                             source: node.id,
                             targets: ready.map((n) => n.id),
                             state,
+                            invocationState,
                         });
                         targets.push(...ready);
                     }
@@ -202,6 +222,8 @@ export class Graph {
             throw caughtError;
         }
         finally {
+            if (execTimeoutHandle !== undefined)
+                clearTimeout(execTimeoutHandle);
             queue.dispose();
             await Promise.allSettled(streams.values());
             this._tracer.endMultiAgentSpan(multiAgentSpan, {
@@ -209,18 +231,18 @@ export class Graph {
                 ...(result && { usage: result.usage }),
                 ...(caughtError && { error: caughtError }),
             });
-            yield new AfterMultiAgentInvocationEvent({ orchestrator: this, state });
+            yield new AfterMultiAgentInvocationEvent({ orchestrator: this, state, invocationState });
         }
-        yield new MultiAgentResultEvent({ result });
+        yield new MultiAgentResultEvent({ result, invocationState });
         return result;
     }
     /**
      * Executes a single node, pushing streaming events to the shared queue in real-time.
      */
-    async _streamNode(node, input, state, queue, multiAgentSpan) {
+    async _streamNode(node, input, state, queue, multiAgentSpan, invocationState, executionSignal) {
         const nodeState = state.node(node.id);
         const nodeSpan = this._tracer.withSpanContext(multiAgentSpan, () => this._tracer.startNodeSpan({ nodeId: node.id, nodeType: node.type }));
-        const beforeEvent = new BeforeNodeCallEvent({ orchestrator: this, state, nodeId: node.id });
+        const beforeEvent = new BeforeNodeCallEvent({ orchestrator: this, state, nodeId: node.id, invocationState });
         await queue.send({ type: 'event', node, event: beforeEvent });
         if (beforeEvent.cancel) {
             const message = typeof beforeEvent.cancel === 'string' ? beforeEvent.cancel : 'node cancelled by hook';
@@ -230,32 +252,45 @@ export class Graph {
             await queue.send({
                 type: 'event',
                 node,
-                event: new NodeCancelEvent({ nodeId: node.id, state, message }),
+                event: new NodeCancelEvent({ nodeId: node.id, state, message, invocationState }),
             });
             await queue.send({
                 type: 'event',
                 node,
-                event: new AfterNodeCallEvent({ orchestrator: this, state, nodeId: node.id }),
+                event: new AfterNodeCallEvent({ orchestrator: this, state, nodeId: node.id, invocationState }),
             });
             this._tracer.endNodeSpan(nodeSpan, { status: Status.CANCELLED, duration: 0 });
             queue.push({ type: 'result', node, result });
             return;
         }
+        // Per-node timeout applies only to AgentNode. MultiAgentNode wraps a nested Swarm/Graph
+        // that has its own timeout config; cancelSignal isn't yet plumbed into nested orchestrators
+        // so bounding belongs on the child.
+        const nodeTimeout = node instanceof AgentNode ? (node.timeout ?? this.config.nodeTimeout) : Infinity;
+        const nodeTimeoutController = Number.isFinite(nodeTimeout) ? new AbortController() : undefined;
+        const nodeTimeoutHandle = nodeTimeoutController
+            ? setTimeout(() => nodeTimeoutController.abort(), nodeTimeout)
+            : undefined;
+        const signals = [executionSignal, nodeTimeoutController?.signal].filter((s) => s !== undefined);
+        const cancelSignal = signals.length > 0 ? AbortSignal.any(signals) : undefined;
         try {
             const nodeInput = this._resolveNodeInput(node, input, state);
-            const gen = this._tracer.withSpanContext(nodeSpan, () => node.stream(nodeInput, state));
+            const gen = this._tracer.withSpanContext(nodeSpan, () => node.stream(nodeInput, state, { invocationState, ...(cancelSignal && { cancelSignal }) }));
             let next = await this._tracer.withSpanContext(nodeSpan, () => gen.next());
             while (!next.done) {
                 await queue.send({ type: 'event', node, event: next.value });
                 next = await this._tracer.withSpanContext(nodeSpan, () => gen.next());
             }
+            if (nodeTimeoutController?.signal.aborted) {
+                throw new Error(`node_timeout=<${nodeTimeout}>, node_id=<${node.id}>, graph_id=<${this.id}> | node exceeded wall-clock budget`);
+            }
             const result = next.value;
             this._tracer.endNodeSpan(nodeSpan, { status: result.status, duration: result.duration, usage: result.usage });
             queue.push({ type: 'result', node, result });
             await queue.send({
                 type: 'event',
                 node,
-                event: new AfterNodeCallEvent({ orchestrator: this, state, nodeId: node.id }),
+                event: new AfterNodeCallEvent({ orchestrator: this, state, nodeId: node.id, invocationState }),
             });
         }
         catch (error) {
@@ -268,6 +303,7 @@ export class Graph {
                     orchestrator: this,
                     state,
                     nodeId: node.id,
+                    invocationState,
                     error: nodeError,
                 }),
             });
@@ -277,6 +313,10 @@ export class Graph {
                 error: nodeError,
             });
         }
+        finally {
+            if (nodeTimeoutHandle !== undefined)
+                clearTimeout(nodeTimeoutHandle);
+        }
     }
     _validateConfig() {
         if (this.config.maxConcurrency < 1) {
@@ -285,6 +325,12 @@ export class Graph {
         if (this.config.maxSteps < 1) {
             throw new Error(`max_steps=<${this.config.maxSteps}> | must be at least 1`);
         }
+        if (this.config.timeout < 1) {
+            throw new Error(`timeout=<${this.config.timeout}> | must be at least 1`);
+        }
+        if (this.config.nodeTimeout < 1) {
+            throw new Error(`node_timeout=<${this.config.nodeTimeout}> | must be at least 1`);
+        }
     }
     _validateSources() {
         if (this._sources.length === 0) {