@strands-agents/sdk 0.5.0 → 0.7.0

package/dist/src/multiagent/graph.js

@@ -0,0 +1,400 @@
+import { TextBlock, contentBlockFromData } from '../types/messages.js';
+import { HookableEvent } from '../hooks/events.js';
+import { HookRegistryImplementation } from '../hooks/registry.js';
+import { MultiAgentPluginRegistry } from './plugins.js';
+import { AgentNode, MultiAgentNode, Node } from './nodes.js';
+import { MultiAgentState, MultiAgentResult, NodeResult, Status } from './state.js';
+import { Swarm } from './swarm.js';
+import { AfterMultiAgentInvocationEvent, AfterNodeCallEvent, BeforeMultiAgentInvocationEvent, BeforeNodeCallEvent, MultiAgentHandoffEvent, MultiAgentInitializedEvent, MultiAgentResultEvent, NodeCancelEvent, } from './events.js';
+import { Edge } from './edge.js';
+import { Queue } from './queue.js';
+/**
+ * Directed graph orchestration pattern.
+ *
+ * Agents execute as nodes in a dependency graph, with edges defining execution order
+ * and optional conditions controlling routing. Source nodes (those with no incoming edges)
+ * run first, and downstream nodes execute once all their dependencies complete. Parallel
+ * execution is supported up to a configurable concurrency limit.
+ *
+ * Key design choices vs the Python SDK:
+ * - Construction uses a declarative options object rather than a mutable GraphBuilder.
+ *   Nodes and edges are passed directly to the constructor.
+ * - Dependency resolution uses AND semantics: a node runs only when all incoming edges
+ *   are satisfied. Python uses OR semantics, firing a node when any single incoming
+ *   edge from the completed batch is satisfied.
+ * - Nodes are launched individually as they become ready (up to maxConcurrency). Python
+ *   executes in discrete batches, waiting for the entire batch to complete before
+ *   scheduling the next set of nodes.
+ * - Agent nodes are stateless by default (snapshot/restore on each execution). Python
+ *   accumulates agent state across executions unless `reset_on_revisit` is enabled.
+ * - Node failures produce a FAILED result, allowing parallel paths to continue.
+ *   MultiAgent-level limits (maxSteps) throw exceptions. Python does the inverse:
+ *   node failures throw exceptions (fail-fast), while limit violations return a
+ *   FAILED result.
+ *
+ * @example
+ * ```typescript
+ * const graph = new Graph({
+ *   nodes: [researcher, writer],
+ *   edges: [['researcher', 'writer']],
+ * })
+ *
+ * const result = await graph.invoke('Explain quantum computing')
+ * ```
+ */
+export class Graph {
+    id;
+    nodes;
+    edges;
+    config;
+    _pluginRegistry;
+    _hookRegistry;
+    _sources;
+    _initialized;
+    constructor(options) {
+        const { id, nodes, edges, sources, plugins, ...config } = options;
+        this.id = id ?? 'graph';
+        this.config = {
+            maxConcurrency: config.maxConcurrency ?? Infinity,
+            maxSteps: config.maxSteps ?? Infinity,
+        };
+        this._validateConfig();
+        this.nodes = this._resolveNodes(nodes);
+        this.edges = this._resolveEdges(edges);
+        this._sources = this._resolveSources(sources);
+        this._validateSources();
+        this._hookRegistry = new HookRegistryImplementation();
+        this._pluginRegistry = new MultiAgentPluginRegistry(plugins);
+        this._initialized = false;
+    }
+    /**
+     * Initialize the graph. Invokes the {@link MultiAgentInitializedEvent} callback.
+     * Called automatically on first invocation.
+     */
+    async initialize() {
+        if (this._initialized)
+            return;
+        await this._pluginRegistry.initialize(this);
+        await this._hookRegistry.invokeCallbacks(new MultiAgentInitializedEvent({ orchestrator: this }));
+        this._initialized = true;
+    }
+    /**
+     * Invoke graph and return final result (consumes stream).
+     *
+     * @param input - The input to pass to entry point nodes
+     * @returns Promise resolving to the final MultiAgentResult
+     */
+    async invoke(input) {
+        const gen = this.stream(input);
+        let next = await gen.next();
+        while (!next.done) {
+            next = await gen.next();
+        }
+        return next.value;
+    }
+    /**
+     * Register a hook callback for a specific graph event type.
+     *
+     * @param eventType - The event class constructor to register the callback for
+     * @param callback - The callback function to invoke when the event occurs
+     * @returns Cleanup function that removes the callback when invoked
+     */
+    addHook(eventType, callback) {
+        return this._hookRegistry.addCallback(eventType, callback);
+    }
+    /**
+     * Stream graph execution, yielding events as nodes execute.
+     * Invokes hook callbacks for each event before yielding.
+     *
+     * @param input - The input to pass to entry nodes
+     * @returns Async generator yielding streaming events and returning a MultiAgentResult
+     */
+    async *stream(input) {
+        await this.initialize();
+        const gen = this._stream(input);
+        try {
+            let next = await gen.next();
+            while (!next.done) {
+                if (next.value instanceof HookableEvent) {
+                    await this._hookRegistry.invokeCallbacks(next.value);
+                }
+                yield next.value;
+                next = await gen.next();
+            }
+            return next.value;
+        }
+        finally {
+            await gen.return(undefined);
+        }
+    }
+    async *_stream(input) {
+        const state = new MultiAgentState({ nodeIds: [...this.nodes.keys()] });
+        const queue = new Queue();
+        const targets = [...this._sources];
+        const streams = new Map();
+        yield new BeforeMultiAgentInvocationEvent({ orchestrator: this, state });
+        try {
+            while (targets.length > 0 || streams.size > 0) {
+                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));
+                }
+                await queue.wait();
+                while (queue.size > 0) {
+                    const { data, ack } = queue.shift();
+                    if (data.type === 'event') {
+                        yield data.event;
+                        ack();
+                        continue;
+                    }
+                    if (data.type === 'error') {
+                        streams.delete(data.node.id);
+                        ack();
+                        throw data.error;
+                    }
+                    const { node, result } = data;
+                    streams.delete(node.id);
+                    ack();
+                    state.results.push(result);
+                    const ready = await this._findReady(node, state, streams, targets);
+                    if (ready.length > 0) {
+                        yield new MultiAgentHandoffEvent({
+                            source: node.id,
+                            targets: ready.map((n) => n.id),
+                            state,
+                        });
+                        targets.push(...ready);
+                    }
+                }
+            }
+        }
+        finally {
+            queue.dispose();
+            await Promise.allSettled(streams.values());
+            yield new AfterMultiAgentInvocationEvent({ orchestrator: this, state });
+        }
+        const result = new MultiAgentResult({
+            results: state.results,
+            content: this._resolveContent(state),
+            duration: Date.now() - state.startTime,
+        });
+        yield new MultiAgentResultEvent({ result });
+        return result;
+    }
+    /**
+     * Executes a single node, pushing streaming events to the shared queue in real-time.
+     */
+    async _streamNode(node, input, state, queue) {
+        const nodeState = state.node(node.id);
+        const beforeEvent = new BeforeNodeCallEvent({ orchestrator: this, state, nodeId: node.id });
+        await queue.send({ type: 'event', node, event: beforeEvent });
+        if (beforeEvent.cancel) {
+            const message = typeof beforeEvent.cancel === 'string' ? beforeEvent.cancel : 'node cancelled by hook';
+            const result = new NodeResult({ nodeId: node.id, status: Status.CANCELLED, duration: 0 });
+            nodeState.status = Status.CANCELLED;
+            nodeState.results.push(result);
+            await queue.send({
+                type: 'event',
+                node,
+                event: new NodeCancelEvent({ nodeId: node.id, state, message }),
+            });
+            await queue.send({
+                type: 'event',
+                node,
+                event: new AfterNodeCallEvent({ orchestrator: this, state, nodeId: node.id }),
+            });
+            queue.push({ type: 'result', node, result });
+            return;
+        }
+        try {
+            const nodeInput = this._resolveNodeInput(node, input, state);
+            const gen = node.stream(nodeInput, state);
+            let next = await gen.next();
+            while (!next.done) {
+                await queue.send({ type: 'event', node, event: next.value });
+                next = await gen.next();
+            }
+            queue.push({ type: 'result', node, result: next.value });
+            await queue.send({
+                type: 'event',
+                node,
+                event: new AfterNodeCallEvent({ orchestrator: this, state, nodeId: node.id }),
+            });
+        }
+        catch (error) {
+            await queue.send({
+                type: 'event',
+                node,
+                event: new AfterNodeCallEvent({
+                    orchestrator: this,
+                    state,
+                    nodeId: node.id,
+                    error: error instanceof Error ? error : new Error(String(error)),
+                }),
+            });
+            queue.push({
+                type: 'error',
+                node,
+                error: error instanceof Error ? error : new Error(String(error)),
+            });
+        }
+    }
+    _validateConfig() {
+        if (this.config.maxConcurrency < 1) {
+            throw new Error(`max_concurrency=<${this.config.maxConcurrency}> | must be at least 1`);
+        }
+        if (this.config.maxSteps < 1) {
+            throw new Error(`max_steps=<${this.config.maxSteps}> | must be at least 1`);
+        }
+    }
+    _validateSources() {
+        if (this._sources.length === 0) {
+            throw new Error('graph has no source nodes');
+        }
+        const visited = new Set();
+        const adjacency = new Map();
+        for (const edge of this.edges) {
+            const targets = adjacency.get(edge.source.id) ?? [];
+            targets.push(edge.target.id);
+            adjacency.set(edge.source.id, targets);
+        }
+        const queue = this._sources.map((n) => n.id);
+        while (queue.length > 0) {
+            const id = queue.shift();
+            if (visited.has(id))
+                continue;
+            visited.add(id);
+            for (const target of adjacency.get(id) ?? []) {
+                queue.push(target);
+            }
+        }
+        for (const id of this.nodes.keys()) {
+            if (!visited.has(id)) {
+                throw new Error(`node_id=<${id}> | unreachable from any source node`);
+            }
+        }
+    }
+    _resolveNodes(definitions) {
+        const nodes = new Map();
+        for (const definition of definitions) {
+            let node;
+            if (definition instanceof Node) {
+                node = definition;
+            }
+            else if ('orchestrator' in definition) {
+                node = new MultiAgentNode(definition);
+            }
+            else if ('agent' in definition) {
+                node = new AgentNode(definition);
+            }
+            else if (definition instanceof Graph || definition instanceof Swarm) {
+                node = new MultiAgentNode({ orchestrator: definition });
+            }
+            else {
+                node = new AgentNode({ agent: definition });
+            }
+            if (nodes.has(node.id)) {
+                throw new Error(`node_id=<${node.id}> | duplicate node id`);
+            }
+            nodes.set(node.id, node);
+        }
+        return nodes;
+    }
+    _resolveEdges(definitions) {
+        const edges = [];
+        for (const definition of definitions) {
+            const [sourceId, targetId, handler] = Array.isArray(definition)
+                ? [definition[0], definition[1], undefined]
+                : [definition.source, definition.target, definition.handler];
+            const source = this.nodes.get(sourceId);
+            const target = this.nodes.get(targetId);
+            if (!source) {
+                throw new Error(`source=<${sourceId}> | edge references unknown source node`);
+            }
+            if (!target) {
+                throw new Error(`target=<${targetId}> | edge references unknown target node`);
+            }
+            edges.push(new Edge({ source, target, ...(handler && { handler }) }));
+        }
+        return edges;
+    }
+    _resolveSources(sourceIds) {
+        if (sourceIds) {
+            const sources = [];
+            for (const id of sourceIds) {
+                const node = this.nodes.get(id);
+                if (!node) {
+                    throw new Error(`source=<${id}> | source references unknown node`);
+                }
+                sources.push(node);
+            }
+            return sources;
+        }
+        const targetIds = new Set(this.edges.map((e) => e.target.id));
+        return [...this.nodes.values()].filter((node) => !targetIds.has(node.id));
+    }
+    /**
+     * Identifies terminus nodes and returns their combined content.
+     * A terminus node is where an execution path ended: completed with no
+     * downstream progress, or failed/cancelled.
+     */
+    _resolveContent(state) {
+        for (const [id, ns] of state.nodes.entries()) {
+            if (ns.status === Status.FAILED || ns.status === Status.CANCELLED) {
+                ns.terminus = true;
+            }
+            else if (ns.status === Status.COMPLETED) {
+                ns.terminus = !this.edges
+                    .filter((e) => e.source.id === id)
+                    .some((e) => state.node(e.target.id)?.status !== Status.PENDING);
+            }
+        }
+        return [...state.nodes.values()].filter((ns) => ns.terminus).flatMap((ns) => ns.content);
+    }
+    /**
+     * Builds the input for a node by combining the original task with dependency outputs.
+     */
+    _resolveNodeInput(node, input, state) {
+        const deps = [];
+        for (const edge of this.edges.filter((e) => e.target.id === node.id)) {
+            const ns = state.node(edge.source.id);
+            if (ns.content.length > 0) {
+                deps.push(new TextBlock(`[node: ${edge.source.id}]`), ...ns.content);
+            }
+        }
+        if (deps.length === 0)
+            return input;
+        const blocks = typeof input === 'string'
+            ? [new TextBlock(input)]
+            : input.map((b) => ('type' in b ? b : contentBlockFromData(b)));
+        return [...blocks, ...deps];
+    }
+    _checkSteps(state) {
+        if (state.steps >= this.config.maxSteps) {
+            throw new Error(`steps=<${state.steps}> | max steps reached`);
+        }
+    }
+    /**
+     * Finds downstream nodes that are ready to execute after a node completes.
+     * A target is ready when all its incoming edge sources are COMPLETED.
+     */
+    async _findReady(node, state, streams, targets) {
+        if (state.node(node.id)?.status !== Status.COMPLETED)
+            return [];
+        const ready = [];
+        for (const edge of this.edges.filter((e) => e.source.id === node.id)) {
+            if (!(await edge.handler(state)))
+                continue;
+            if (streams.has(edge.target.id) || targets.some((n) => n.id === edge.target.id))
+                continue;
+            const deps = this.edges.filter((e) => e.target.id === edge.target.id);
+            if (deps.every((e) => state.node(e.source.id)?.status === Status.COMPLETED)) {
+                ready.push(edge.target);
+            }
+        }
+        return ready;
+    }
+}
+//# sourceMappingURL=graph.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"graph.js","sourceRoot":"","sources":["../../../src/multiagent/graph.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAA;AACtE,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAClD,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAA;AAGjE,OAAO,EAAE,wBAAwB,EAAE,MAAM,cAAc,CAAA;AAEvD,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,YAAY,CAAA;AAC5D,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,YAAY,CAAA;AAElF,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAA;AAElC,OAAO,EACL,8BAA8B,EAC9B,kBAAkB,EAClB,+BAA+B,EAC/B,mBAAmB,EACnB,sBAAsB,EACtB,0BAA0B,EAC1B,qBAAqB,EACrB,eAAe,GAChB,MAAM,aAAa,CAAA;AAEpB,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAA;AA4BlC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,MAAM,OAAO,KAAK;IACP,EAAE,CAAQ;IACV,KAAK,CAA2B;IAChC,KAAK,CAAiB;IACtB,MAAM,CAAuB;IACrB,eAAe,CAA0B;IACzC,aAAa,CAA4B;IACzC,QAAQ,CAAQ;IACzB,YAAY,CAAS;IAE7B,YAAY,OAAqB;QAC/B,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,MAAM,EAAE,GAAG,OAAO,CAAA;QAEjE,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,OAAO,CAAA;QAEvB,IAAI,CAAC,MAAM,GAAG;YACZ,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,QAAQ;YACjD,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,QAAQ;SACtC,CAAA;QACD,IAAI,CAAC,eAAe,EAAE,CAAA;QAEtB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAA;QACtC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAA;QACtC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;QAC7C,IAAI,CAAC,gBAAgB,EAAE,CAAA;QAEvB,IAAI,CAAC,aAAa,GAAG,IAAI,0BAA0B,EAAE,CAAA;QACrD,IAAI,CAAC,eAAe,GAAG,IAAI,wBAAwB,CAAC,OAAO,CAAC,CAAA;QAC5D,IAAI,CAAC,YAAY,GAAG,KAAK,CAAA;IAC3B,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,YAAY;YAAE,OAAM;QAC7B,MAAM,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,IAAI,CAAC,CAAA;QAC3C,MAAM,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,0BAA0B,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;QAChG,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;IAC1B,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,MAAM,CAAC,KAAsB;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC9B,IAAI,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;QAC3B,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;QACzB,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAA;IACnB,CAAC;IAED;;;;;;OAMG;IACH,OAAO,CAA0B,SAAsC,EAAE,QAAyB;QAChG,OAAO,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;IAC5D,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,CAAC,MAAM,CAAC,KAAsB;QAClC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAA;QAEvB,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAC/B,IAAI,CAAC;YACH,IAAI,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;YAC3B,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,IAAI,IAAI,CAAC,KAAK,YAAY,aAAa,EAAE,CAAC;oBACxC,MAAM,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;gBACtD,CAAC;gBACD,MAAM,IAAI,CAAC,KAAK,CAAA;gBAChB,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;YACzB,CAAC;YACD,OAAO,IAAI,CAAC,KAAK,CAAA;QACnB,CAAC;gBAAS,CAAC;YACT,MAAM,GAAG,CAAC,MAAM,CAAC,SAAkB,CAAC,CAAA;QACtC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,OAAO,CAAC,KAAsB;QAC3C,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAA;QAEtE,MAAM,KAAK,GAAG,IAAI,KAAK,EAAE,CAAA;QACzB,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAA;QAClC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAyB,CAAA;QAEhD,MAAM,IAAI,+BAA+B,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;QAExE,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC9C,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;oBACvE,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,EAAG,CAAA;oBAE7B,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAA;oBACvB,KAAK,CAAC,KAAK,EAAE,CAAA;oBAEb,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAA;gBACnE,CAAC;gBAED,MAAM,KAAK,CAAC,IAAI,EAAE,CAAA;gBAClB,OAAO,KAAK,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;oBACtB,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,KAAK,EAAG,CAAA;oBAEpC,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;wBAC1B,MAAM,IAAI,CAAC,KAAK,CAAA;wBAChB,GAAG,EAAE,CAAA;wBACL,SAAQ;oBACV,CAAC;oBAED,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;wBAC1B,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;wBAC5B,GAAG,EAAE,CAAA;wBACL,MAAM,IAAI,CAAC,KAAK,CAAA;oBAClB,CAAC;oBAED,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,IAAI,CAAA;oBAC7B,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;oBACvB,GAAG,EAAE,CAAA;oBAEL,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;oBAE1B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;oBAClE,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACrB,MAAM,IAAI,sBAAsB,CAAC;4BAC/B,MAAM,EAAE,IAAI,CAAC,EAAE;4BACf,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;4BAC/B,KAAK;yBACN,CAAC,CAAA;wBACF,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAA;oBACxB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,KAAK,CAAC,OAAO,EAAE,CAAA;YACf,MAAM,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;YAC1C,MAAM,IAAI,8BAA8B,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAA;QACzE,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC;YAClC,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,OAAO,EAAE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC;YACpC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS;SACvC,CAAC,CAAA;QACF,MAAM,IAAI,qBAAqB,CAAC,EAAE,MAAM,EAAE,CAAC,CAAA;QAC3C,OAAO,MAAM,CAAA;IACf,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,IAAU,EAAE,KAAsB,EAAE,KAAsB,EAAE,KAAY;QAChG,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAE,CAAA;QAEtC,MAAM,WAAW,GAAG,IAAI,mBAAmB,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAA;QAC3F,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAA;QAE7D,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;YACvB,MAAM,OAAO,GAAG,OAAO,WAAW,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,wBAAwB,CAAA;YACtG,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAA;YACzF,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC,SAAS,CAAA;YACnC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAE9B,MAAM,KAAK,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,OAAO;gBACb,IAAI;gBACJ,KAAK,EAAE,IAAI,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;aAChE,CAAC,CAAA;YACF,MAAM,KAAK,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,OAAO;gBACb,IAAI;gBACJ,KAAK,EAAE,IAAI,kBAAkB,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;aAC9E,CAAC,CAAA;YACF,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;YAC5C,OAAM;QACR,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAA;YAE5D,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAA;YACzC,IAAI,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;YAC3B,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAA;gBAC5D,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;YACzB,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAA;YAExD,MAAM,KAAK,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,OAAO;gBACb,IAAI;gBACJ,KAAK,EAAE,IAAI,kBAAkB,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;aAC9E,CAAC,CAAA;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,OAAO;gBACb,IAAI;gBACJ,KAAK,EAAE,IAAI,kBAAkB,CAAC;oBAC5B,YAAY,EAAE,IAAI;oBAClB,KAAK;oBACL,MAAM,EAAE,IAAI,CAAC,EAAE;oBACf,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;iBACjE,CAAC;aACH,CAAC,CAAA;YACF,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,OAAO;gBACb,IAAI;gBACJ,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;aACjE,CAAC,CAAA;QACJ,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,GAAG,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,oBAAoB,IAAI,CAAC,MAAM,CAAC,cAAc,wBAAwB,CAAC,CAAA;QACzF,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,CAAC,MAAM,CAAC,QAAQ,wBAAwB,CAAC,CAAA;QAC7E,CAAC;IACH,CAAC;IAEO,gBAAgB;QACtB,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAA;QAC9C,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAA;QACjC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAoB,CAAA;QAC7C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,CAAA;YACnD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YAC5B,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,CAAA;QACxC,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;QAC5C,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,EAAE,GAAG,KAAK,CAAC,KAAK,EAAG,CAAA;YACzB,IAAI,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBAAE,SAAQ;YAC7B,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YACf,KAAK,MAAM,MAAM,IAAI,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC7C,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YACpB,CAAC;QACH,CAAC;QAED,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;YACnC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CAAC,YAAY,EAAE,sCAAsC,CAAC,CAAA;YACvE,CAAC;QACH,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,WAA6B;QACjD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAgB,CAAA;QAErC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,IAAI,IAAU,CAAA;YAEd,IAAI,UAAU,YAAY,IAAI,EAAE,CAAC;gBAC/B,IAAI,GAAG,UAAU,CAAA;YACnB,CAAC;iBAAM,IAAI,cAAc,IAAI,UAAU,EAAE,CAAC;gBACxC,IAAI,GAAG,IAAI,cAAc,CAAC,UAAU,CAAC,CAAA;YACvC,CAAC;iBAAM,IAAI,OAAO,IAAI,UAAU,EAAE,CAAC;gBACjC,IAAI,GAAG,IAAI,SAAS,CAAC,UAAU,CAAC,CAAA;YAClC,CAAC;iBAAM,IAAI,UAAU,YAAY,KAAK,IAAI,UAAU,YAAY,KAAK,EAAE,CAAC;gBACtE,IAAI,GAAG,IAAI,cAAc,CAAC,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,CAAA;YACzD,CAAC;iBAAM,CAAC;gBACN,IAAI,GAAG,IAAI,SAAS,CAAC,EAAE,KAAK,EAAE,UAA4B,EAAE,CAAC,CAAA;YAC/D,CAAC;YAED,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;gBACvB,MAAM,IAAI,KAAK,CAAC,YAAY,IAAI,CAAC,EAAE,uBAAuB,CAAC,CAAA;YAC7D,CAAC;YACD,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;QAC1B,CAAC;QAED,OAAO,KAAK,CAAA;IACd,CAAC;IAEO,aAAa,CAAC,WAA6B;QACjD,MAAM,KAAK,GAAW,EAAE,CAAA;QACxB,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC;gBAC7D,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC;gBAC3C,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC,CAAA;YAE9D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;YACvC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;YACvC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,WAAW,QAAQ,yCAAyC,CAAC,CAAA;YAC/E,CAAC;YACD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,WAAW,QAAQ,yCAAyC,CAAC,CAAA;YAC/E,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;QACvE,CAAC;QACD,OAAO,KAAK,CAAA;IACd,CAAC;IAEO,eAAe,CAAC,SAAoB;QAC1C,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,OAAO,GAAW,EAAE,CAAA;YAC1B,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;gBAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;gBAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,MAAM,IAAI,KAAK,CAAC,WAAW,EAAE,oCAAoC,CAAC,CAAA;gBACpE,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACpB,CAAC;YACD,OAAO,OAAO,CAAA;QAChB,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAA;QAC7D,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAA;IAC3E,CAAC;IAED;;;;OAIG;IACK,eAAe,CAAC,KAAsB;QAC5C,KAAK,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YAC7C,IAAI,EAAE,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,MAAM,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;gBAClE,EAAE,CAAC,QAAQ,GAAG,IAAI,CAAA;YACpB,CAAC;iBAAM,IAAI,EAAE,CAAC,MAAM,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;gBAC1C,EAAE,CAAC,QAAQ,GAAG,CAAC,IAAI,CAAC,KAAK;qBACtB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;qBACjC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC,OAAO,CAAC,CAAA;YACpE,CAAC;QACH,CAAC;QACD,OAAO,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAA;IAC1F,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,IAAU,EAAE,KAAsB,EAAE,KAAsB;QAClF,MAAM,IAAI,GAAmB,EAAE,CAAA;QAC/B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YACrE,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAE,CAAA;YACtC,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,IAAI,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC,UAAU,IAAI,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,CAAA;YACtE,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,KAAK,CAAA;QAEnC,MAAM,MAAM,GACV,OAAO,KAAK,KAAK,QAAQ;YACvB,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAE,CAAkB,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;QACrF,OAAO,CAAC,GAAG,MAAM,EAAE,GAAG,IAAI,CAAC,CAAA;IAC7B,CAAC;IAEO,WAAW,CAAC,KAAsB;QACxC,IAAI,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,UAAU,KAAK,CAAC,KAAK,uBAAuB,CAAC,CAAA;QAC/D,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,UAAU,CACtB,IAAU,EACV,KAAsB,EACtB,OAA2C,EAC3C,OAAwB;QAExB,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC,SAAS;YAAE,OAAO,EAAE,CAAA;QAE/D,MAAM,KAAK,GAAW,EAAE,CAAA;QAExB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YACrE,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAAE,SAAQ;YAE1C,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;gBAAE,SAAQ;YAEzF,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YACrE,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC5E,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YACzB,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAA;IACd,CAAC;CACF"}

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

@@ -5,9 +5,14 @@ export { MultiAgentState, NodeState, Status, NodeResult, MultiAgentResult } from
 export type { NodeResultUpdate, ResultStatus } from './state.js';
 export { Node, AgentNode, MultiAgentNode } from './nodes.js';
 export type { NodeConfig, AgentNodeOptions, MultiAgentNodeOptions, NodeDefinition, NodeType } from './nodes.js';
-export type { MultiAgentBase } from './base.js';
-export { MultiAgentInitializedEvent, BeforeMultiAgentInvocationEvent, AfterMultiAgentInvocationEvent, BeforeNodeCallEvent, AfterNodeCallEvent, NodeStreamUpdateEvent, NodeResultEvent, MultiAgentHandoffEvent, MultiAgentResultEvent, } from './events.js';
-export type { MultiAgentStreamEvent } from './events.js';
+export { MultiAgentInitializedEvent, BeforeMultiAgentInvocationEvent, AfterMultiAgentInvocationEvent, BeforeNodeCallEvent, AfterNodeCallEvent, NodeStreamUpdateEvent, NodeResultEvent, NodeCancelEvent, MultiAgentHandoffEvent, MultiAgentResultEvent, } from './events.js';
+export type { MultiAgentStreamEvent, NodeStreamUpdateInnerEvent } from './events.js';
 export { Edge } from './edge.js';
 export type { EdgeHandler, EdgeDefinition } from './edge.js';
+export { Graph } from './graph.js';
+export type { GraphConfig, GraphOptions } from './graph.js';
+export { Swarm } from './swarm.js';
+export type { SwarmConfig, SwarmNodeDefinition, SwarmOptions } from './swarm.js';
+export type { MultiAgentPlugin } from './plugins.js';
+export type { MultiAgent, MultiAgentInput } from './multiagent.js';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/multiagent/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAA;AAC7F,YAAY,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AAEhE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC5D,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAE/G,YAAY,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAE/C,OAAO,EACL,0BAA0B,EAC1B,+BAA+B,EAC/B,8BAA8B,EAC9B,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,aAAa,CAAA;AACpB,YAAY,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAA;AAExD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/multiagent/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAA;AAC7F,YAAY,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AAEhE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC5D,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAE/G,OAAO,EACL,0BAA0B,EAC1B,+BAA+B,EAC/B,8BAA8B,EAC9B,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,eAAe,EACf,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,aAAa,CAAA;AACpB,YAAY,EAAE,qBAAqB,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAA;AAEpF,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAE5D,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAA;AAClC,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AAE3D,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAA;AAClC,YAAY,EAAE,WAAW,EAAE,mBAAmB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AAEhF,YAAY,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAA;AAEpD,YAAY,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAA"}

package/dist/src/multiagent/index.js

@@ -3,6 +3,8 @@
  */
 export { MultiAgentState, NodeState, Status, NodeResult, MultiAgentResult } from './state.js';
 export { Node, AgentNode, MultiAgentNode } from './nodes.js';
-export { MultiAgentInitializedEvent, BeforeMultiAgentInvocationEvent, AfterMultiAgentInvocationEvent, BeforeNodeCallEvent, AfterNodeCallEvent, NodeStreamUpdateEvent, NodeResultEvent, MultiAgentHandoffEvent, MultiAgentResultEvent, } from './events.js';
+export { MultiAgentInitializedEvent, BeforeMultiAgentInvocationEvent, AfterMultiAgentInvocationEvent, BeforeNodeCallEvent, AfterNodeCallEvent, NodeStreamUpdateEvent, NodeResultEvent, NodeCancelEvent, MultiAgentHandoffEvent, MultiAgentResultEvent, } from './events.js';
 export { Edge } from './edge.js';
+export { Graph } from './graph.js';
+export { Swarm } from './swarm.js';
 //# sourceMappingURL=index.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/multiagent/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAA;AAG7F,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAK5D,OAAO,EACL,0BAA0B,EAC1B,+BAA+B,EAC/B,8BAA8B,EAC9B,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,aAAa,CAAA;AAGpB,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/multiagent/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAA;AAG7F,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAG5D,OAAO,EACL,0BAA0B,EAC1B,+BAA+B,EAC/B,8BAA8B,EAC9B,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,eAAe,EACf,sBAAsB,EACtB,qBAAqB,GACtB,MAAM,aAAa,CAAA;AAGpB,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAGhC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAA;AAGlC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAA"}

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

@@ -0,0 +1,41 @@
+import type { InvokeArgs } from '../types/agent.js';
+import type { Message, MessageData } from '../types/messages.js';
+import type { HookableEvent } from '../hooks/events.js';
+import type { HookCallback, HookableEventConstructor, HookCleanup } from '../hooks/types.js';
+import type { MultiAgentStreamEvent } from './events.js';
+import type { MultiAgentResult } from './state.js';
+/**
+ * Input type for multi-agent orchestrators. Excludes `Message[]` and `MessageData[]`
+ * from {@link InvokeArgs} since orchestrators route content blocks between nodes.
+ */
+export type MultiAgentInput = Exclude<InvokeArgs, Message[] | MessageData[]>;
+/**
+ * Interface for any multi-agent orchestrator that can stream execution.
+ * Implement this interface to create custom orchestration patterns that can be
+ * composed as nodes within other orchestrators via {@link MultiAgentNode}.
+ */
+export interface MultiAgent {
+    /** Unique identifier for this orchestrator. */
+    readonly id: string;
+    /**
+     * Execute the orchestrator and return the final result.
+     * @param input - Input to pass to the orchestrator
+     * @returns The aggregate result from all executed nodes
+     */
+    invoke(input: MultiAgentInput): Promise<MultiAgentResult>;
+    /**
+     * Execute the orchestrator and stream events as they occur.
+     * @param input - Input to pass to the orchestrator
+     * @returns Async generator yielding events and returning the final result
+     */
+    stream(input: MultiAgentInput): AsyncGenerator<MultiAgentStreamEvent, MultiAgentResult, undefined>;
+    /**
+     * Register a hook callback for a specific orchestrator event type.
+     *
+     * @param eventType - The event class constructor to register the callback for
+     * @param callback - The callback function to invoke when the event occurs
+     * @returns Cleanup function that removes the callback when invoked
+     */
+    addHook<T extends HookableEvent>(eventType: HookableEventConstructor<T>, callback: HookCallback<T>): HookCleanup;
+}
+//# sourceMappingURL=multiagent.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"multiagent.d.ts","sourceRoot":"","sources":["../../../src/multiagent/multiagent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AACnD,OAAO,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAA;AAChE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,wBAAwB,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAC5F,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAA;AACxD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAA;AAElD;;;GAGG;AACH,MAAM,MAAM,eAAe,GAAG,OAAO,CAAC,UAAU,EAAE,OAAO,EAAE,GAAG,WAAW,EAAE,CAAC,CAAA;AAE5E;;;;GAIG;AACH,MAAM,WAAW,UAAU;IACzB,+CAA+C;IAC/C,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAA;IAEnB;;;;OAIG;IACH,MAAM,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAA;IAEzD;;;;OAIG;IACH,MAAM,CAAC,KAAK,EAAE,eAAe,GAAG,cAAc,CAAC,qBAAqB,EAAE,gBAAgB,EAAE,SAAS,CAAC,CAAA;IAElG;;;;;;OAMG;IACH,OAAO,CAAC,CAAC,SAAS,aAAa,EAAE,SAAS,EAAE,wBAAwB,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAC,GAAG,WAAW,CAAA;CACjH"}

package/dist/src/multiagent/multiagent.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=multiagent.js.map

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

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

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

@@ -1,8 +1,9 @@
-import type { Agent, InvokeArgs } from '../agent/agent.js';
+import type { InvokableAgent } from '../types/agent.js';
+import type { MultiAgentInput } from './multiagent.js';
 import type { MultiAgentStreamEvent } from './events.js';
 import { NodeResult } from './state.js';
 import type { MultiAgentState, NodeResultUpdate } from './state.js';
-import type { MultiAgentBase } from './base.js';
+import type { MultiAgent } from './multiagent.js';
 /**
  * Known node type identifiers with extensibility for custom nodes.
  */
@@ -12,9 +13,9 @@ export type NodeType = 'agentNode' | 'multiAgentNode' | (string & {});
  */
 export interface NodeConfig {
     /**
-     * Maximum execution time for this node in milliseconds.
+     * Optional description of what this node does.
      */
-    timeout?: number;
+    description?: string;
 }
 /**
  * Abstract base class for all multi-agent orchestration nodes.
@@ -38,40 +39,38 @@ export declare abstract class Node {
      * Execute the node. Handles duration measurement, error capture,
      * and delegates to handle() for node-specific logic.
      *
-     * @param args - Input to pass to the node (string, content blocks, or messages)
+     * @param args - Input to pass to the node (string or content blocks)
      * @param state - The current multi-agent state
      * @returns Async generator yielding streaming events and returning a NodeResult
      */
-    stream(args: InvokeArgs, state: MultiAgentState): AsyncGenerator<MultiAgentStreamEvent, NodeResult, undefined>;
+    stream(args: MultiAgentInput, state: MultiAgentState): AsyncGenerator<MultiAgentStreamEvent, NodeResult, undefined>;
     /**
      * Node-specific execution logic implemented by subclasses.
      *
-     * @param args - Input to process (string, content blocks, or messages)
+     * @param args - Input to process (string or content blocks)
      * @param state - The current multi-agent state
      * @returns Async generator yielding streaming events and returning a partial result
      */
-    abstract handle(args: InvokeArgs, state: MultiAgentState): AsyncGenerator<MultiAgentStreamEvent, NodeResultUpdate, undefined>;
+    abstract handle(args: MultiAgentInput, state: MultiAgentState): AsyncGenerator<MultiAgentStreamEvent, NodeResultUpdate, undefined>;
 }
 /**
  * Options for creating an {@link AgentNode}.
  */
-export interface AgentNodeOptions extends NodeConfig {
-    /** Unique node identifier. */
-    id: string;
+export interface AgentNodeOptions {
     /** The agent to wrap as a node. */
-    agent: Agent;
+    agent: InvokableAgent;
 }
 /**
- * Node that wraps an Agent instance for multi-agent orchestration.
+ * Node that wraps an {@link InvokableAgent} instance for multi-agent orchestration.
  *
- * Each execution is isolated — the wrapped agent's internal state
- * is unchanged after the node completes.
+ * Each execution is isolated. When the wrapped agent is an {@link Agent} instance,
+ * its internal state is snapshot/restored so it remains unchanged after the node completes.
  */
 export declare class AgentNode extends Node {
     readonly type: "agentNode";
     private readonly _agent;
     constructor(options: AgentNodeOptions);
-    get agent(): Agent;
+    get agent(): InvokableAgent;
     /**
      * Executes the wrapped agent, yielding each agent streaming event
      * wrapped in a {@link NodeStreamUpdateEvent}.
@@ -80,14 +79,14 @@ export declare class AgentNode extends Node {
      * @param state - The current multi-agent state
      * @returns Async generator yielding streaming events and returning the agent's content blocks
      */
-    handle(args: InvokeArgs, state: MultiAgentState): AsyncGenerator<MultiAgentStreamEvent, NodeResultUpdate, undefined>;
+    handle(args: MultiAgentInput, state: MultiAgentState): AsyncGenerator<MultiAgentStreamEvent, NodeResultUpdate, undefined>;
 }
 /**
  * Options for creating a {@link MultiAgentNode}.
  */
 export interface MultiAgentNodeOptions extends NodeConfig {
     /** The orchestrator to wrap as a node. */
-    orchestrator: MultiAgentBase;
+    orchestrator: MultiAgent;
 }
 /**
  * Node that wraps a multi-agent orchestrator (e.g. Graph) for nested composition.
@@ -100,7 +99,7 @@ export declare class MultiAgentNode extends Node {
     readonly type: "multiAgentNode";
     private readonly _orchestrator;
     constructor(options: MultiAgentNodeOptions);
-    get orchestrator(): MultiAgentBase;
+    get orchestrator(): MultiAgent;
     /**
      * Executes the wrapped orchestrator. Inner {@link NodeStreamUpdateEvent}s
      * pass through as-is; all other events are wrapped in a new
@@ -110,14 +109,14 @@ export declare class MultiAgentNode extends Node {
      * @param _state - The current multi-agent state (unused)
      * @returns Async generator yielding streaming events and returning the orchestrator's content
      */
-    handle(args: InvokeArgs, _state: MultiAgentState): AsyncGenerator<MultiAgentStreamEvent, NodeResultUpdate, undefined>;
+    handle(args: MultiAgentInput, state: MultiAgentState): AsyncGenerator<MultiAgentStreamEvent, NodeResultUpdate, undefined>;
 }
 /**
  * A node definition accepted by orchestration constructors.
+ *
+ * Pass an {@link InvokableAgent} or {@link MultiAgent} directly for the simple case,
+ * use typed options objects for per-node configuration, or provide pre-built
+ * {@link Node} instances for full control.
  */
-export type NodeDefinition = Node | (AgentNodeOptions & {
-    type: 'agent';
-}) | (MultiAgentNodeOptions & {
-    type: 'multiAgent';
-});
+export type NodeDefinition = InvokableAgent | MultiAgent | Node | AgentNodeOptions | MultiAgentNodeOptions;
 //# sourceMappingURL=nodes.d.ts.map