@illuma-ai/agents 1.1.23 → 1.1.25

package/dist/cjs/graphs/MultiAgentGraph.cjs

@@ -552,6 +552,20 @@ class MultiAgentGraph extends Graph.StandardGraph {
                 console.debug(`[MultiAgentGraph] Handoff "${sourceAgentId}" -> "${destination}" START ` +
                     `(messages: ${childMessages.length})`);
                 try {
+                    /**
+                     * Dispatch transition BEFORE invoking the child subgraph so that
+                     * callbacks.js sets multiAgentTrace.isMultiAgent = true before the
+                     * child's ON_RUN_STEP events fire.  This ensures child tool calls
+                     * are attributed to the correct agent in admin traces.
+                     */
+                    await events.safeDispatchCustomEvent(_enum.GraphEvents.ON_AGENT_TRANSITION, {
+                        sourceAgentId: sourceAgentId,
+                        sourceAgentName: this.agentContexts.get(sourceAgentId)?.name ?? sourceAgentId,
+                        destinationAgentId: destination,
+                        destinationAgentName: destContext?.name ?? destination,
+                        edgeType: _enum.EdgeType.HANDOFF,
+                        timestamp: Date.now(),
+                    }, config);
                     /**
                      * Invoke the child subgraph with config propagation.
                      * Config carries callbacks (for SSE streaming), abort signal,
@@ -563,14 +577,6 @@ class MultiAgentGraph extends Graph.StandardGraph {
                     console.debug(`[MultiAgentGraph] Handoff "${sourceAgentId}" -> "${destination}" DONE ` +
                         `(result: ${resultText.length} chars` +
                         `${truncatedResult.length < resultText.length ? `, truncated to ${truncatedResult.length}` : ''})`);
-                    await events.safeDispatchCustomEvent(_enum.GraphEvents.ON_AGENT_TRANSITION, {
-                        sourceAgentId: sourceAgentId,
-                        sourceAgentName: this.agentContexts.get(sourceAgentId)?.name ?? sourceAgentId,
-                        destinationAgentId: destination,
-                        destinationAgentName: destContext?.name ?? destination,
-                        edgeType: _enum.EdgeType.HANDOFF,
-                        timestamp: Date.now(),
-                    }, config);
                     return truncatedResult;
                 }
                 catch (err) {