llmist 12.3.3 → 12.3.4

package/dist/index.cjs

@@ -10403,7 +10403,17 @@ async function safeObserve(fn, logger, eventType) {
     logger.warn(`Observer error in ${eventType}:`, error);
   }
 }
+function chainObserverCall(chainMap, key, fn, logger, eventType, cleanup = false) {
+  const previousPromise = chainMap.get(key) ?? Promise.resolve();
+  const newPromise = previousPromise.then(() => safeObserve(fn, logger, eventType));
+  chainMap.set(key, newPromise);
+  if (cleanup) {
+    newPromise.finally(() => chainMap.delete(key));
+  }
+}
 function bridgeTreeToHooks(tree, hooks, logger) {
+  const gadgetPromiseChains = /* @__PURE__ */ new Map();
+  const llmPromiseChains = /* @__PURE__ */ new Map();
   return tree.onAll((event) => {
     const subagentContext = buildSubagentContext(tree, event);
     switch (event.type) {
@@ -10426,10 +10436,14 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(
+          chainObserverCall(
+            gadgetPromiseChains,
+            gadgetEvent.invocationId,
             () => hooks.observers?.onGadgetExecutionStart?.(context),
             logger,
-            "onGadgetExecutionStart"
+            "onGadgetExecutionStart",
+            false
+            // Don't cleanup - wait for completion event
           );
         }
         break;
@@ -10449,10 +10463,14 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(
+          chainObserverCall(
+            gadgetPromiseChains,
+            gadgetEvent.invocationId,
             () => hooks.observers?.onGadgetExecutionComplete?.(context),
             logger,
-            "onGadgetExecutionComplete"
+            "onGadgetExecutionComplete",
+            true
+            // Cleanup after completion
           );
         }
         break;
@@ -10471,10 +10489,14 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(
+          chainObserverCall(
+            gadgetPromiseChains,
+            gadgetEvent.invocationId,
             () => hooks.observers?.onGadgetExecutionComplete?.(context),
             logger,
-            "onGadgetExecutionComplete"
+            "onGadgetExecutionComplete",
+            true
+            // Cleanup after error
           );
         }
         break;
@@ -10493,7 +10515,15 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(() => hooks.observers?.onGadgetSkipped?.(context), logger, "onGadgetSkipped");
+          chainObserverCall(
+            gadgetPromiseChains,
+            gadgetEvent.invocationId,
+            () => hooks.observers?.onGadgetSkipped?.(context),
+            logger,
+            "onGadgetSkipped",
+            true
+            // Cleanup after skipped
+          );
         }
         break;
       }
@@ -10517,7 +10547,15 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(() => hooks.observers?.onLLMCallStart?.(context), logger, "onLLMCallStart");
+          chainObserverCall(
+            llmPromiseChains,
+            event.nodeId,
+            () => hooks.observers?.onLLMCallStart?.(context),
+            logger,
+            "onLLMCallStart",
+            false
+            // Don't cleanup - wait for completion event
+          );
         }
         break;
       }
@@ -10541,10 +10579,14 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(
+          chainObserverCall(
+            llmPromiseChains,
+            event.nodeId,
             () => hooks.observers?.onLLMCallComplete?.(context),
             logger,
-            "onLLMCallComplete"
+            "onLLMCallComplete",
+            true
+            // Cleanup after completion
           );
         }
         break;
@@ -10566,7 +10608,15 @@ function bridgeTreeToHooks(tree, hooks, logger) {
             logger,
             subagentContext
           };
-          safeObserve(() => hooks.observers?.onLLMCallError?.(context), logger, "onLLMCallError");
+          chainObserverCall(
+            llmPromiseChains,
+            event.nodeId,
+            () => hooks.observers?.onLLMCallError?.(context),
+            logger,
+            "onLLMCallError",
+            true
+            // Cleanup after error
+          );
         }
         break;
       }