llmist 12.3.2 → 12.3.3

package/dist/index.cjs

@@ -10362,6 +10362,18 @@ function findParentGadgetInvocationId(tree, nodeId) {
   }
   return void 0;
 }
+function getIterationFromTree(tree, nodeId) {
+  let currentId = nodeId;
+  while (currentId) {
+    const node = tree.getNode(currentId);
+    if (!node) break;
+    if (node.type === "llm_call") {
+      return node.iteration;
+    }
+    currentId = node.parentId;
+  }
+  return 0;
+}
 function buildSubagentContext(tree, event) {
   const parentGadgetInvocationId = findParentGadgetInvocationId(tree, event.nodeId);
   if (!parentGadgetInvocationId) {
@@ -10395,12 +10407,101 @@ function bridgeTreeToHooks(tree, hooks, logger) {
   return tree.onAll((event) => {
     const subagentContext = buildSubagentContext(tree, event);
     switch (event.type) {
-      // NOTE: gadget_start, gadget_complete, gadget_error, gadget_skipped are NOT
-      // handled here. They are called directly in stream-processor.ts with await
-      // to ensure proper ordering of observer commands.
-      // LLM events - bridged for subagent visibility
-      // When a subagent makes LLM calls, these events propagate through the shared tree
-      // to the parent's hooks, enabling unified monitoring of all LLM activity
+      // =================================================================
+      // GADGET EVENTS - Bridged for subagent visibility
+      // =================================================================
+      // When a subagent executes gadgets, these events propagate through
+      // the shared tree to the parent's hooks.
+      // Only bridged for subagent events (depth > 0) to avoid double-calling
+      // root agent events which are handled directly in stream-processor.ts
+      case "gadget_start": {
+        if (subagentContext && hooks.observers?.onGadgetExecutionStart) {
+          const gadgetEvent = event;
+          const gadgetNode = tree.getNode(event.nodeId);
+          const context = {
+            iteration: getIterationFromTree(tree, event.nodeId),
+            gadgetName: gadgetEvent.name,
+            invocationId: gadgetEvent.invocationId,
+            parameters: gadgetNode?.parameters ?? {},
+            logger,
+            subagentContext
+          };
+          safeObserve(
+            () => hooks.observers?.onGadgetExecutionStart?.(context),
+            logger,
+            "onGadgetExecutionStart"
+          );
+        }
+        break;
+      }
+      case "gadget_complete": {
+        if (subagentContext && hooks.observers?.onGadgetExecutionComplete) {
+          const gadgetEvent = event;
+          const gadgetNode = tree.getNode(event.nodeId);
+          const context = {
+            iteration: getIterationFromTree(tree, event.nodeId),
+            gadgetName: gadgetEvent.name,
+            invocationId: gadgetEvent.invocationId,
+            parameters: gadgetNode?.parameters ?? {},
+            finalResult: gadgetEvent.result,
+            executionTimeMs: gadgetEvent.executionTimeMs,
+            cost: gadgetEvent.cost,
+            logger,
+            subagentContext
+          };
+          safeObserve(
+            () => hooks.observers?.onGadgetExecutionComplete?.(context),
+            logger,
+            "onGadgetExecutionComplete"
+          );
+        }
+        break;
+      }
+      case "gadget_error": {
+        if (subagentContext && hooks.observers?.onGadgetExecutionComplete) {
+          const gadgetEvent = event;
+          const gadgetNode = tree.getNode(event.nodeId);
+          const context = {
+            iteration: getIterationFromTree(tree, event.nodeId),
+            gadgetName: gadgetEvent.name,
+            invocationId: gadgetEvent.invocationId,
+            parameters: gadgetNode?.parameters ?? {},
+            error: gadgetEvent.error,
+            executionTimeMs: gadgetEvent.executionTimeMs,
+            logger,
+            subagentContext
+          };
+          safeObserve(
+            () => hooks.observers?.onGadgetExecutionComplete?.(context),
+            logger,
+            "onGadgetExecutionComplete"
+          );
+        }
+        break;
+      }
+      case "gadget_skipped": {
+        if (subagentContext && hooks.observers?.onGadgetSkipped) {
+          const gadgetEvent = event;
+          const gadgetNode = tree.getNode(event.nodeId);
+          const context = {
+            iteration: getIterationFromTree(tree, event.nodeId),
+            gadgetName: gadgetEvent.name,
+            invocationId: gadgetEvent.invocationId,
+            parameters: gadgetNode?.parameters ?? {},
+            failedDependency: gadgetEvent.failedDependency ?? "",
+            failedDependencyError: gadgetEvent.failedDependencyError ?? gadgetEvent.error,
+            logger,
+            subagentContext
+          };
+          safeObserve(() => hooks.observers?.onGadgetSkipped?.(context), logger, "onGadgetSkipped");
+        }
+        break;
+      }
+      // =================================================================
+      // LLM EVENTS - Bridged for subagent visibility
+      // =================================================================
+      // When a subagent makes LLM calls, these events propagate through
+      // the shared tree to the parent's hooks.
       case "llm_call_start": {
         if (subagentContext && hooks.observers?.onLLMCallStart) {
           const llmEvent = event;
@@ -10416,7 +10517,7 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(() => hooks.observers.onLLMCallStart(context), logger, "onLLMCallStart");
+          safeObserve(() => hooks.observers?.onLLMCallStart?.(context), logger, "onLLMCallStart");
         }
         break;
       }
@@ -10441,7 +10542,7 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             subagentContext
           };
           safeObserve(
-            () => hooks.observers.onLLMCallComplete(context),
+            () => hooks.observers?.onLLMCallComplete?.(context),
             logger,
             "onLLMCallComplete"
           );
@@ -10465,7 +10566,7 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(() => hooks.observers.onLLMCallError(context), logger, "onLLMCallError");
+          safeObserve(() => hooks.observers?.onLLMCallError?.(context), logger, "onLLMCallError");
         }
         break;
       }