llmist 12.2.2 → 12.2.4

package/dist/index.js

@@ -10290,6 +10290,137 @@ var init_executor = __esm({
   }
 });
 
+// src/agent/tree-hook-bridge.ts
+function findParentGadgetInvocationId(tree, nodeId) {
+  let currentId = nodeId;
+  while (currentId) {
+    const node = tree.getNode(currentId);
+    if (!node) break;
+    currentId = node.parentId;
+    if (!currentId) break;
+    const parentNode = tree.getNode(currentId);
+    if (parentNode?.type === "gadget") {
+      return parentNode.invocationId;
+    }
+  }
+  return void 0;
+}
+function buildSubagentContext(tree, event) {
+  const parentGadgetInvocationId = findParentGadgetInvocationId(tree, event.nodeId);
+  if (!parentGadgetInvocationId) {
+    return void 0;
+  }
+  return {
+    parentGadgetInvocationId,
+    depth: event.depth
+  };
+}
+function getSubagentContextForNode(tree, nodeId) {
+  const node = tree.getNode(nodeId);
+  if (!node) return void 0;
+  const parentGadgetInvocationId = findParentGadgetInvocationId(tree, nodeId);
+  if (!parentGadgetInvocationId) {
+    return void 0;
+  }
+  return {
+    parentGadgetInvocationId,
+    depth: node.depth
+  };
+}
+async function safeObserve(fn, logger, eventType) {
+  try {
+    await fn();
+  } catch (error) {
+    logger.warn(`Observer error in ${eventType}:`, error);
+  }
+}
+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
+      case "llm_call_start": {
+        if (subagentContext && hooks.observers?.onLLMCallStart) {
+          const llmEvent = event;
+          const context = {
+            iteration: llmEvent.iteration,
+            options: {
+              model: llmEvent.model,
+              messages: llmEvent.request ?? [],
+              // These fields are not available from tree events, use defaults
+              temperature: void 0,
+              maxTokens: void 0
+            },
+            logger,
+            subagentContext
+          };
+          safeObserve(() => hooks.observers.onLLMCallStart(context), logger, "onLLMCallStart");
+        }
+        break;
+      }
+      case "llm_call_complete": {
+        if (subagentContext && hooks.observers?.onLLMCallComplete) {
+          const llmEvent = event;
+          const llmNode = tree.getNode(event.nodeId);
+          const context = {
+            iteration: llmNode?.iteration ?? 1,
+            options: {
+              model: llmNode?.model ?? "unknown",
+              messages: llmNode?.request ?? [],
+              temperature: void 0,
+              maxTokens: void 0
+            },
+            finishReason: llmEvent.finishReason ?? null,
+            usage: llmEvent.usage,
+            rawResponse: llmEvent.response,
+            // Use rawResponse as finalMessage since interceptor modifications aren't available from tree events
+            finalMessage: llmEvent.response,
+            logger,
+            subagentContext
+          };
+          safeObserve(
+            () => hooks.observers.onLLMCallComplete(context),
+            logger,
+            "onLLMCallComplete"
+          );
+        }
+        break;
+      }
+      case "llm_call_error": {
+        if (subagentContext && hooks.observers?.onLLMCallError) {
+          const llmEvent = event;
+          const llmNode = tree.getNode(event.nodeId);
+          const context = {
+            iteration: llmNode?.iteration ?? 1,
+            options: {
+              model: llmNode?.model ?? "unknown",
+              messages: llmNode?.request ?? [],
+              temperature: void 0,
+              maxTokens: void 0
+            },
+            error: llmEvent.error,
+            recovered: llmEvent.recovered,
+            logger,
+            subagentContext
+          };
+          safeObserve(() => hooks.observers.onLLMCallError(context), logger, "onLLMCallError");
+        }
+        break;
+      }
+    }
+  });
+}
+var init_tree_hook_bridge = __esm({
+  "src/agent/tree-hook-bridge.ts"() {
+    "use strict";
+  }
+});
+
 // src/agent/stream-processor.ts
 var StreamProcessor;
 var init_stream_processor = __esm({
@@ -10299,6 +10430,7 @@ var init_stream_processor = __esm({
     init_parser();
     init_logger();
     init_hook_validators();
+    init_tree_hook_bridge();
     StreamProcessor = class {
       iteration;
       registry;
@@ -10560,15 +10692,31 @@ var init_stream_processor = __esm({
               invocationId: call.invocationId
             });
             this.failedInvocations.add(call.invocationId);
+            const errorMessage = `Gadget "${call.invocationId}" cannot depend on itself (self-referential dependency)`;
             const skipEvent = {
               type: "gadget_skipped",
               gadgetName: call.gadgetName,
               invocationId: call.invocationId,
               parameters: call.parameters ?? {},
               failedDependency: call.invocationId,
-              failedDependencyError: `Gadget "${call.invocationId}" cannot depend on itself (self-referential dependency)`
+              failedDependencyError: errorMessage
             };
             yield skipEvent;
+            if (this.hooks.observers?.onGadgetSkipped) {
+              const gadgetNode = this.tree?.getNodeByInvocationId(call.invocationId);
+              const subagentContext = this.tree && gadgetNode ? getSubagentContextForNode(this.tree, gadgetNode.id) : void 0;
+              const context = {
+                iteration: this.iteration,
+                gadgetName: call.gadgetName,
+                invocationId: call.invocationId,
+                parameters: call.parameters ?? {},
+                failedDependency: call.invocationId,
+                failedDependencyError: errorMessage,
+                logger: this.logger,
+                subagentContext
+              };
+              await this.safeObserve(() => this.hooks.observers.onGadgetSkipped(context));
+            }
             return;
           }
           const failedDep = call.dependencies.find(
@@ -10711,6 +10859,19 @@ var init_stream_processor = __esm({
             this.tree.startGadget(gadgetNode.id);
           }
         }
+        if (this.hooks.observers?.onGadgetExecutionStart) {
+          const gadgetNode = this.tree?.getNodeByInvocationId(call.invocationId);
+          const subagentContext = this.tree && gadgetNode ? getSubagentContextForNode(this.tree, gadgetNode.id) : void 0;
+          const context = {
+            iteration: this.iteration,
+            gadgetName: call.gadgetName,
+            invocationId: call.invocationId,
+            parameters,
+            logger: this.logger,
+            subagentContext
+          };
+          await this.safeObserve(() => this.hooks.observers.onGadgetExecutionStart(context));
+        }
         let result;
         if (shouldSkip) {
           result = {
@@ -10778,6 +10939,23 @@ var init_stream_processor = __esm({
             }
           }
         }
+        if (this.hooks.observers?.onGadgetExecutionComplete) {
+          const gadgetNode = this.tree?.getNodeByInvocationId(result.invocationId);
+          const subagentContext = this.tree && gadgetNode ? getSubagentContextForNode(this.tree, gadgetNode.id) : void 0;
+          const context = {
+            iteration: this.iteration,
+            gadgetName: result.gadgetName,
+            invocationId: result.invocationId,
+            parameters,
+            finalResult: result.result,
+            error: result.error,
+            executionTimeMs: result.executionTimeMs,
+            cost: result.cost,
+            logger: this.logger,
+            subagentContext
+          };
+          await this.safeObserve(() => this.hooks.observers.onGadgetExecutionComplete(context));
+        }
         this.completedResults.set(result.invocationId, result);
         if (result.error) {
           this.failedInvocations.add(result.invocationId);
@@ -10903,6 +11081,21 @@ var init_stream_processor = __esm({
             failedDependencyError: depError
           };
           events.push(skipEvent);
+          if (this.hooks.observers?.onGadgetSkipped) {
+            const gadgetNode = this.tree?.getNodeByInvocationId(call.invocationId);
+            const subagentContext = this.tree && gadgetNode ? getSubagentContextForNode(this.tree, gadgetNode.id) : void 0;
+            const context = {
+              iteration: this.iteration,
+              gadgetName: call.gadgetName,
+              invocationId: call.invocationId,
+              parameters: call.parameters ?? {},
+              failedDependency: failedDep,
+              failedDependencyError: depError,
+              logger: this.logger,
+              subagentContext
+            };
+            await this.safeObserve(() => this.hooks.observers.onGadgetSkipped(context));
+          }
           this.logger.info("Gadget skipped due to failed dependency", {
             gadgetName: call.gadgetName,
             invocationId: call.invocationId,
@@ -11030,6 +11223,21 @@ var init_stream_processor = __esm({
               failedDependencyError: errorMessage
             };
             yield skipEvent;
+            if (this.hooks.observers?.onGadgetSkipped) {
+              const gadgetNode = this.tree?.getNodeByInvocationId(invocationId);
+              const subagentContext = this.tree && gadgetNode ? getSubagentContextForNode(this.tree, gadgetNode.id) : void 0;
+              const context = {
+                iteration: this.iteration,
+                gadgetName: call.gadgetName,
+                invocationId,
+                parameters: call.parameters ?? {},
+                failedDependency: missingDeps[0],
+                failedDependencyError: errorMessage,
+                logger: this.logger,
+                subagentContext
+              };
+              await this.safeObserve(() => this.hooks.observers.onGadgetSkipped(context));
+            }
           }
           this.gadgetsAwaitingDependencies.clear();
         }
@@ -11055,9 +11263,7 @@ var init_stream_processor = __esm({
        */
       async runObserversInParallel(observers) {
         if (observers.length === 0) return;
-        await Promise.allSettled(
-          observers.map((observer) => this.safeObserve(observer))
-        );
+        await Promise.allSettled(observers.map((observer) => this.safeObserve(observer)));
       }
       // ==========================================================================
       // Public accessors for cross-iteration dependency tracking
@@ -11080,239 +11286,6 @@ var init_stream_processor = __esm({
   }
 });
 
-// src/agent/tree-hook-bridge.ts
-function findParentGadgetInvocationId(tree, nodeId) {
-  let currentId = nodeId;
-  while (currentId) {
-    const node = tree.getNode(currentId);
-    if (!node) break;
-    currentId = node.parentId;
-    if (!currentId) break;
-    const parentNode = tree.getNode(currentId);
-    if (parentNode?.type === "gadget") {
-      return parentNode.invocationId;
-    }
-  }
-  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) {
-    return void 0;
-  }
-  return {
-    parentGadgetInvocationId,
-    depth: event.depth
-  };
-}
-function getSubagentContextForNode(tree, nodeId) {
-  const node = tree.getNode(nodeId);
-  if (!node) return void 0;
-  const parentGadgetInvocationId = findParentGadgetInvocationId(tree, nodeId);
-  if (!parentGadgetInvocationId) {
-    return void 0;
-  }
-  return {
-    parentGadgetInvocationId,
-    depth: node.depth
-  };
-}
-async function safeObserve(fn, logger, eventType) {
-  try {
-    await fn();
-  } catch (error) {
-    logger.warn(`Observer error in ${eventType}:`, error);
-  }
-}
-function bridgeTreeToHooks(tree, hooks, logger) {
-  return tree.onAll((event) => {
-    const subagentContext = buildSubagentContext(tree, event);
-    switch (event.type) {
-      case "gadget_start": {
-        if (hooks.observers?.onGadgetExecutionStart) {
-          const gadgetEvent = event;
-          const gadgetNode = tree.getNodeByInvocationId(gadgetEvent.invocationId);
-          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 (hooks.observers?.onGadgetExecutionComplete) {
-          const gadgetEvent = event;
-          const gadgetNode = tree.getNodeByInvocationId(gadgetEvent.invocationId);
-          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 (hooks.observers?.onGadgetExecutionComplete) {
-          const gadgetNode = tree.getNodeByInvocationId(event.invocationId);
-          const context = {
-            iteration: getIterationFromTree(tree, event.nodeId),
-            gadgetName: event.name,
-            invocationId: event.invocationId,
-            parameters: gadgetNode?.parameters ?? {},
-            error: event.error,
-            executionTimeMs: event.executionTimeMs,
-            logger,
-            subagentContext
-          };
-          safeObserve(
-            () => hooks.observers.onGadgetExecutionComplete(context),
-            logger,
-            "onGadgetExecutionComplete"
-          );
-        }
-        break;
-      }
-      case "gadget_skipped": {
-        if (hooks.observers?.onGadgetSkipped) {
-          const gadgetNode = tree.getNodeByInvocationId(event.invocationId);
-          const context = {
-            iteration: getIterationFromTree(tree, event.nodeId),
-            gadgetName: event.name,
-            invocationId: event.invocationId,
-            parameters: gadgetNode?.parameters ?? {},
-            failedDependency: event.failedDependency ?? "unknown",
-            failedDependencyError: event.failedDependencyError ?? event.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, enabling unified monitoring of all LLM activity
-      case "llm_call_start": {
-        if (subagentContext && hooks.observers?.onLLMCallStart) {
-          const llmEvent = event;
-          const context = {
-            iteration: llmEvent.iteration,
-            options: {
-              model: llmEvent.model,
-              messages: llmEvent.request ?? [],
-              // These fields are not available from tree events, use defaults
-              temperature: void 0,
-              maxTokens: void 0
-            },
-            logger,
-            subagentContext
-          };
-          safeObserve(
-            () => hooks.observers.onLLMCallStart(context),
-            logger,
-            "onLLMCallStart"
-          );
-        }
-        break;
-      }
-      case "llm_call_complete": {
-        if (subagentContext && hooks.observers?.onLLMCallComplete) {
-          const llmEvent = event;
-          const llmNode = tree.getNode(event.nodeId);
-          const context = {
-            iteration: llmNode?.iteration ?? 1,
-            options: {
-              model: llmNode?.model ?? "unknown",
-              messages: llmNode?.request ?? [],
-              temperature: void 0,
-              maxTokens: void 0
-            },
-            finishReason: llmEvent.finishReason ?? null,
-            usage: llmEvent.usage,
-            rawResponse: llmEvent.response,
-            // Use rawResponse as finalMessage since interceptor modifications aren't available from tree events
-            finalMessage: llmEvent.response,
-            logger,
-            subagentContext
-          };
-          safeObserve(
-            () => hooks.observers.onLLMCallComplete(context),
-            logger,
-            "onLLMCallComplete"
-          );
-        }
-        break;
-      }
-      case "llm_call_error": {
-        if (subagentContext && hooks.observers?.onLLMCallError) {
-          const llmEvent = event;
-          const llmNode = tree.getNode(event.nodeId);
-          const context = {
-            iteration: llmNode?.iteration ?? 1,
-            options: {
-              model: llmNode?.model ?? "unknown",
-              messages: llmNode?.request ?? [],
-              temperature: void 0,
-              maxTokens: void 0
-            },
-            error: llmEvent.error,
-            recovered: llmEvent.recovered,
-            logger,
-            subagentContext
-          };
-          safeObserve(
-            () => hooks.observers.onLLMCallError(context),
-            logger,
-            "onLLMCallError"
-          );
-        }
-        break;
-      }
-    }
-  });
-}
-var init_tree_hook_bridge = __esm({
-  "src/agent/tree-hook-bridge.ts"() {
-    "use strict";
-  }
-});
-
 // src/agent/agent.ts
 import pRetry from "p-retry";
 var Agent;