@dudousxd/nestjs-agent 1.35.1 → 1.36.0

package/dist/index.cjs

@@ -6000,8 +6000,12 @@ var InlineAgentRunner = class _InlineAgentRunner {
       ...deps,
       day
     }, input, hooks).then(() => (0, import_nestjs_agent_core32.releaseThreadRun)(this.store, input.threadId, runId)).catch(async (error) => {
+      const durationMs = (0, import_nestjs_agent_core32.runDurationMs)(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
       if (error instanceof import_nestjs_agent_core32.RunCancelledError) {
-        await this.settleCancelled(runId, input, deps);
+        await this.settleCancelled(runId, input, deps, durationMs);
         return;
       }
       const message = error instanceof Error ? error.message : String(error);
@@ -6009,12 +6013,16 @@ var InlineAgentRunner = class _InlineAgentRunner {
       this.logger.error(`agent run ${runId} failed (${code}): ${message}`);
       (0, import_nestjs_agent_core32.publishAgentRunFailed)({
         runId,
+        threadId: input.threadId,
         code,
-        message
+        message,
+        ...timed,
+        context: (0, import_nestjs_agent_core32.agentRunContext)(input)
       });
       await this.store.recordRunEnd?.({
         runId,
         status: "failed",
+        ...timed,
         errorCode: code,
         errorMessage: message
       });
@@ -6086,8 +6094,17 @@ var InlineAgentRunner = class _InlineAgentRunner {
   * the stream ENDED after a `cancelled` frame rather than failed, so a client that retries failed
   * streams does not retry this one.
   */
-  async settleCancelled(runId, input, deps) {
+  async settleCancelled(runId, input, deps, durationMs) {
     this.logger.log(`agent run ${runId} cancelled`);
+    const timed = durationMs !== void 0 ? {
+      durationMs
+    } : {};
+    (0, import_nestjs_agent_core32.publishAgentRunCancelled)({
+      runId,
+      threadId: input.threadId,
+      ...timed,
+      context: (0, import_nestjs_agent_core32.agentRunContext)(input)
+    });
     const writer = await deps.sink.open(runId);
     await this.settleQueue(writer, input, runId, "cancelled");
     await (0, import_nestjs_agent_core32.releaseThreadRun)(this.store, input.threadId, runId);
@@ -6097,7 +6114,8 @@ var InlineAgentRunner = class _InlineAgentRunner {
     await writer.end();
     await this.store.recordRunEnd?.({
       runId,
-      status: "cancelled"
+      status: "cancelled",
+      ...timed
     });
   }
   /**
@@ -6283,10 +6301,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
         }
       }), "startAgent")
     };
-    void (0, import_nestjs_agent_core32.runAgentLoop)({
-      ...deps,
-      day
-    }, {
+    const childInput = {
       threadId: subThread.id,
       actor,
       userText: task,
@@ -6296,13 +6311,29 @@ var InlineAgentRunner = class _InlineAgentRunner {
       delegationPath: path,
       parentRunId,
       deliverTo
-    }, hooks).catch(async (error) => {
+    };
+    void (0, import_nestjs_agent_core32.runAgentLoop)({
+      ...deps,
+      day
+    }, childInput, hooks).catch(async (error) => {
       const cancelled = error instanceof import_nestjs_agent_core32.RunCancelledError;
       const message = error instanceof Error ? error.message : String(error);
-      if (!cancelled) {
+      const durationMs = (0, import_nestjs_agent_core32.runDurationMs)(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
+      const settled = {
+        runId,
+        threadId: subThread.id,
+        ...timed,
+        context: (0, import_nestjs_agent_core32.agentRunContext)(childInput)
+      };
+      if (cancelled) {
+        (0, import_nestjs_agent_core32.publishAgentRunCancelled)(settled);
+      } else {
         this.logger.error(`detached agent run ${runId} failed: ${message}`);
         (0, import_nestjs_agent_core32.publishAgentRunFailed)({
-          runId,
+          ...settled,
           code: (0, import_nestjs_agent_core32.agentFailureCode)(error),
           message
         });
@@ -6310,6 +6341,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
       await this.store.recordRunEnd?.({
         runId,
         status: cancelled ? "cancelled" : "failed",
+        ...timed,
         ...cancelled ? {} : {
           errorCode: (0, import_nestjs_agent_core32.agentFailureCode)(error),
           errorMessage: message