@illuma-ai/agents 1.1.22 → 1.1.24

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) {
@@ -766,6 +772,26 @@ class MultiAgentGraph extends Graph.StandardGraph {
                 }));
             }
         }
+        /**
+         * Ensure messages end with a HumanMessage.
+         * After stripping tool artifacts, the last message may be an AIMessage
+         * (orchestrator's reasoning before the handoff). Some providers (Bedrock,
+         * Google/VertexAI) reject conversations ending with an assistant message.
+         * Convert the trailing AIMessage to a HumanMessage to preserve any useful
+         * context (e.g., compacted tool summaries) while satisfying the API requirement.
+         */
+        if (cleaned.length > 0 && cleaned[cleaned.length - 1].getType() === 'ai') {
+            const lastAI = cleaned[cleaned.length - 1];
+            const content = typeof lastAI.content === 'string'
+                ? lastAI.content
+                : '';
+            if (content.trim()) {
+                cleaned[cleaned.length - 1] = new messages.HumanMessage(`[Context from orchestrator]: ${content}`);
+            }
+            else {
+                cleaned.pop();
+            }
+        }
         return cleaned;
     }
     processTransferReception(messages$1, agentId) {