llmist 10.1.1 → 10.2.1

package/dist/index.js

@@ -7902,19 +7902,16 @@ var init_typed_gadget = __esm({
 import equal from "fast-deep-equal";
 import { z as z4 } from "zod";
 function getHostExportsInternal() {
-  if (!cachedHostExports) {
-    cachedHostExports = {
-      AgentBuilder,
-      Gadget,
-      createGadget,
-      ExecutionTree,
-      LLMist,
-      z: z4
-    };
-  }
-  return cachedHostExports;
+  return {
+    AgentBuilder,
+    Gadget,
+    createGadget,
+    ExecutionTree,
+    LLMist,
+    z: z4
+  };
 }
-var cachedHostExports, GadgetExecutor;
+var GadgetExecutor;
 var init_executor = __esm({
   "src/gadgets/executor.ts"() {
     "use strict";
@@ -10769,13 +10766,67 @@ var init_builder = __esm({
           };
         }
         if (ctx.onSubagentEvent && ctx.invocationId) {
+          const invocationId = ctx.invocationId;
           this.subagentEventCallback = (event) => {
             ctx.onSubagentEvent({
               ...event,
-              gadgetInvocationId: ctx.invocationId,
-              depth
+              gadgetInvocationId: invocationId,
+              // Add depths: parent's depth + event's depth (for recursive nesting)
+              depth: depth + (event.depth ?? 0)
             });
           };
+          const existingHooks = this.hooks ?? {};
+          this.hooks = {
+            ...existingHooks,
+            observers: {
+              ...existingHooks.observers,
+              onLLMCallStart: async (info) => {
+                await existingHooks.observers?.onLLMCallStart?.(info);
+                ctx.onSubagentEvent({
+                  type: "llm_call_start",
+                  event: {
+                    iteration: info.iteration,
+                    model: info.options.model
+                  },
+                  gadgetInvocationId: invocationId,
+                  depth,
+                  iteration: info.iteration
+                });
+              },
+              onGadgetExecutionStart: async (gadgetCtx) => {
+                await existingHooks.observers?.onGadgetExecutionStart?.(gadgetCtx);
+                ctx.onSubagentEvent({
+                  type: "gadget_call",
+                  event: {
+                    type: "gadget_call",
+                    call: {
+                      gadgetName: gadgetCtx.gadgetName,
+                      invocationId: gadgetCtx.invocationId,
+                      parametersRaw: JSON.stringify(gadgetCtx.parameters),
+                      parameters: gadgetCtx.parameters,
+                      dependencies: []
+                    }
+                  },
+                  gadgetInvocationId: invocationId,
+                  depth
+                });
+              },
+              onLLMCallComplete: async (info) => {
+                await existingHooks.observers?.onLLMCallComplete?.(info);
+                ctx.onSubagentEvent({
+                  type: "llm_call_end",
+                  event: {
+                    iteration: info.iteration,
+                    model: info.options.model,
+                    finishReason: info.finishReason ?? void 0,
+                    usage: info.usage
+                  },
+                  gadgetInvocationId: invocationId,
+                  depth
+                });
+              }
+            }
+          };
         }
         if (ctx.signal && !this.signal) {
           this.signal = ctx.signal;