@illuma-ai/agents 1.1.23 → 1.1.24

package/src/graphs/MultiAgentGraph.ts

@@ -689,6 +689,25 @@ export class MultiAgentGraph extends StandardGraph {
             );
 
             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 safeDispatchCustomEvent(
+                GraphEvents.ON_AGENT_TRANSITION,
+                {
+                  sourceAgentId: sourceAgentId,
+                  sourceAgentName: this.agentContexts.get(sourceAgentId)?.name ?? sourceAgentId,
+                  destinationAgentId: destination,
+                  destinationAgentName: destContext?.name ?? destination,
+                  edgeType: EdgeType.HANDOFF,
+                  timestamp: Date.now(),
+                },
+                config
+              );
+
               /**
                * Invoke the child subgraph with config propagation.
                * Config carries callbacks (for SSE streaming), abort signal,
@@ -711,19 +730,6 @@ export class MultiAgentGraph extends StandardGraph {
                   `${truncatedResult.length < resultText.length ? `, truncated to ${truncatedResult.length}` : ''})`
               );
 
-              await safeDispatchCustomEvent(
-                GraphEvents.ON_AGENT_TRANSITION,
-                {
-                  sourceAgentId: sourceAgentId,
-                  sourceAgentName: this.agentContexts.get(sourceAgentId)?.name ?? sourceAgentId,
-                  destinationAgentId: destination,
-                  destinationAgentName: destContext?.name ?? destination,
-                  edgeType: EdgeType.HANDOFF,
-                  timestamp: Date.now(),
-                },
-                config
-              );
-
               return truncatedResult;
             } catch (err) {
               const errorMessage =