@dudousxd/nestjs-agent 1.35.1 → 1.36.1

package/dist/durable/index.cjs

@@ -227,11 +227,13 @@ var AgentRunSteps = class {
     });
     const buffer = input.bufferOutput === true ? (0, import_nestjs_agent_core3.createFrameBuffer)() : void 0;
     let writer;
+    let firstText;
     if (buffer !== void 0) {
       writer = buffer.writer;
     } else {
       const opened = await deps.sink.open(input.sinkRunId);
-      writer = input.childSink ? childSinkWriter(opened) : opened;
+      firstText = (0, import_nestjs_agent_core3.observeFirstText)(input.childSink ? childSinkWriter(opened) : opened);
+      writer = firstText.writer;
     }
     const announced = (0, import_nestjs_agent_core3.announcedToolCalls)(writer);
     const previews = (0, import_nestjs_agent_core3.previewToolInputs)(announced.writer, (0, import_nestjs_agent_core3.registryInputPreviews)(deps.registry, tools, {
@@ -325,9 +327,16 @@ var AgentRunSteps = class {
       ...stamped,
       previews: shown
     } : stamped;
-    return buffer === void 0 ? previewed : {
+    const firstTextAt = firstText?.firstTextAt();
+    if (buffer !== void 0) {
+      return {
+        ...previewed,
+        bufferedFrames: buffer.frames()
+      };
+    }
+    return firstTextAt === void 0 ? previewed : {
       ...previewed,
-      bufferedFrames: buffer.frames()
+      firstTextAt
     };
   }
   /**
@@ -370,6 +379,7 @@ var AgentRunSteps = class {
       ...input.allowedTools
     ]);
     let output;
+    const toolStartedAt = Date.now();
     try {
       output = await (0, import_nestjs_agent_core3.traceToolExecution)(input.ctx.runId, {
         toolCallId: input.toolCallId,
@@ -397,8 +407,9 @@ var AgentRunSteps = class {
       }
       throw error;
     }
+    const executionMs = Date.now() - toolStartedAt;
     output = await (0, import_nestjs_agent_core3.withChannelDelivery)(deps, input.ctx, output, ui.components());
-    return input.collectUi === true ? (0, import_nestjs_agent_core3.wrapToolStepOutput)(output, ui.components(), ui.text()) : output;
+    return input.collectUi === true ? (0, import_nestjs_agent_core3.wrapToolStepOutput)(output, ui.components(), ui.text(), executionMs) : output;
   }
 };
 _ts_decorate([
@@ -798,21 +809,29 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
         });
         throw error;
       }
+      const durationMs = (0, import_nestjs_agent_core4.runDurationMs)(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
       if ((0, import_nestjs_agent_core4.isReplayIntegrityError)(error)) {
         const detail = error.message;
         const code2 = (0, import_nestjs_agent_core4.agentFailureCode)(error);
         this.logger.error(`agent run ${ctx.runId} failed (${code2}): ${detail}`);
         (0, import_nestjs_agent_core4.publishAgentRunFailed)({
           runId: ctx.runId,
+          threadId: input.threadId,
           code: code2,
-          message: detail
+          message: detail,
+          ...timed,
+          context: (0, import_nestjs_agent_core4.agentRunContext)(input)
         });
         await (0, import_nestjs_agent_core4.settleDeadRun)(this.store, {
           runId: ctx.runId,
           threadId: input.threadId,
           failure: {
             code: code2,
-            message: detail
+            message: detail,
+            ...timed
           }
         });
         await (await hooks.openSink()).fail((0, import_nestjs_agent_core4.streamFailure)(error));
@@ -823,13 +842,17 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
       this.logger.error(`agent run ${ctx.runId} failed (${code}): ${message}`);
       (0, import_nestjs_agent_core4.publishAgentRunFailed)({
         runId: ctx.runId,
+        threadId: input.threadId,
         code,
-        message
+        message,
+        ...timed,
+        context: (0, import_nestjs_agent_core4.agentRunContext)(input)
       });
       await stepOf(ctx)("persist:run:fail", async () => {
         await this.store.recordRunEnd?.({
           runId: ctx.runId,
           status: "failed",
+          ...timed,
           errorCode: code,
           errorMessage: message
         });
@@ -1004,6 +1027,57 @@ function deliveryOf(input) {
   };
 }
 __name(deliveryOf, "deliveryOf");
+function runContextOf(input) {
+  if (typeof input !== "object" || input === null) {
+    return void 0;
+  }
+  const { actor, agentName, parentRunId } = input;
+  if (typeof actor?.id !== "string") {
+    return void 0;
+  }
+  return {
+    actorId: actor.id,
+    ...typeof agentName === "string" ? {
+      agentName
+    } : {},
+    ...typeof actor.tenantRef === "string" ? {
+      tenantRef: actor.tenantRef
+    } : {},
+    ...typeof parentRunId === "string" ? {
+      parentRunId
+    } : {}
+  };
+}
+__name(runContextOf, "runContextOf");
+function cancelledAfterMs(detail) {
+  const createdAt = detail?.run.createdAt;
+  if (createdAt === void 0) {
+    return {};
+  }
+  const started = new Date(createdAt).getTime();
+  return Number.isFinite(started) ? {
+    durationMs: Math.max(0, Date.now() - started)
+  } : {};
+}
+__name(cancelledAfterMs, "cancelledAfterMs");
+function publishCancelled(runId, detail) {
+  const input = detail?.run.input;
+  const threadId = threadOfRun(input);
+  const context = runContextOf(input);
+  const timed = cancelledAfterMs(detail);
+  (0, import_nestjs_agent_core6.publishAgentRunCancelled)({
+    runId,
+    ...threadId !== void 0 ? {
+      threadId
+    } : {},
+    ...timed,
+    ...context !== void 0 ? {
+      context
+    } : {}
+  });
+  return timed;
+}
+__name(publishCancelled, "publishCancelled");
 var DurableAgentRunner = class _DurableAgentRunner {
   static {
     __name(this, "DurableAgentRunner");
@@ -1117,9 +1191,11 @@ var DurableAgentRunner = class _DurableAgentRunner {
       kind: "cancelled"
     }));
     await writer.end();
+    const timed = publishCancelled(runId, detail);
     await this.store.recordRunEnd?.({
       runId,
-      status: "cancelled"
+      status: "cancelled",
+      ...timed
     });
   }
   /**
@@ -1167,9 +1243,11 @@ var DurableAgentRunner = class _DurableAgentRunner {
       kind: "cancelled"
     }));
     await writer.end();
+    const timed = publishCancelled(childId, child);
     await this.store.recordRunEnd?.({
       runId: childId,
-      status: "cancelled"
+      status: "cancelled",
+      ...timed
     });
   }
   /**
@@ -6980,8 +7058,12 @@ var InlineAgentRunner = class _InlineAgentRunner {
       ...deps,
       day
     }, input, hooks).then(() => (0, import_nestjs_agent_core38.releaseThreadRun)(this.store, input.threadId, runId)).catch(async (error) => {
+      const durationMs = (0, import_nestjs_agent_core38.runDurationMs)(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
       if (error instanceof import_nestjs_agent_core38.RunCancelledError) {
-        await this.settleCancelled(runId, input, deps);
+        await this.settleCancelled(runId, input, deps, durationMs);
         return;
       }
       const message = error instanceof Error ? error.message : String(error);
@@ -6989,12 +7071,16 @@ var InlineAgentRunner = class _InlineAgentRunner {
       this.logger.error(`agent run ${runId} failed (${code}): ${message}`);
       (0, import_nestjs_agent_core38.publishAgentRunFailed)({
         runId,
+        threadId: input.threadId,
         code,
-        message
+        message,
+        ...timed,
+        context: (0, import_nestjs_agent_core38.agentRunContext)(input)
       });
       await this.store.recordRunEnd?.({
         runId,
         status: "failed",
+        ...timed,
         errorCode: code,
         errorMessage: message
       });
@@ -7066,8 +7152,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_core38.publishAgentRunCancelled)({
+      runId,
+      threadId: input.threadId,
+      ...timed,
+      context: (0, import_nestjs_agent_core38.agentRunContext)(input)
+    });
     const writer = await deps.sink.open(runId);
     await this.settleQueue(writer, input, runId, "cancelled");
     await (0, import_nestjs_agent_core38.releaseThreadRun)(this.store, input.threadId, runId);
@@ -7077,7 +7172,8 @@ var InlineAgentRunner = class _InlineAgentRunner {
     await writer.end();
     await this.store.recordRunEnd?.({
       runId,
-      status: "cancelled"
+      status: "cancelled",
+      ...timed
     });
   }
   /**
@@ -7263,10 +7359,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
         }
       }), "startAgent")
     };
-    void (0, import_nestjs_agent_core38.runAgentLoop)({
-      ...deps,
-      day
-    }, {
+    const childInput = {
       threadId: subThread.id,
       actor,
       userText: task,
@@ -7276,13 +7369,29 @@ var InlineAgentRunner = class _InlineAgentRunner {
       delegationPath: path,
       parentRunId,
       deliverTo
-    }, hooks).catch(async (error) => {
+    };
+    void (0, import_nestjs_agent_core38.runAgentLoop)({
+      ...deps,
+      day
+    }, childInput, hooks).catch(async (error) => {
       const cancelled = error instanceof import_nestjs_agent_core38.RunCancelledError;
       const message = error instanceof Error ? error.message : String(error);
-      if (!cancelled) {
+      const durationMs = (0, import_nestjs_agent_core38.runDurationMs)(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
+      const settled = {
+        runId,
+        threadId: subThread.id,
+        ...timed,
+        context: (0, import_nestjs_agent_core38.agentRunContext)(childInput)
+      };
+      if (cancelled) {
+        (0, import_nestjs_agent_core38.publishAgentRunCancelled)(settled);
+      } else {
         this.logger.error(`detached agent run ${runId} failed: ${message}`);
         (0, import_nestjs_agent_core38.publishAgentRunFailed)({
-          runId,
+          ...settled,
           code: (0, import_nestjs_agent_core38.agentFailureCode)(error),
           message
         });
@@ -7290,6 +7399,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
       await this.store.recordRunEnd?.({
         runId,
         status: cancelled ? "cancelled" : "failed",
+        ...timed,
         ...cancelled ? {} : {
           errorCode: (0, import_nestjs_agent_core38.agentFailureCode)(error),
           errorMessage: message