@dudousxd/nestjs-agent 1.35.0 → 1.36.0

package/dist/durable/index.js

@@ -69,7 +69,7 @@ var init_kit = __esm({
 });
 
 // src/durable/agent-run.workflow.ts
-import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY2, AGENT_STORE, RUN_ENDED_BEFORE_TOOL_CALL, RunCancelledError, agentFailureCode, encodeStreamEvent as encodeStreamEvent2, isReplayIntegrityError as isReplayIntegrityError2, publishAgentRunFailed, releaseThreadRun, runAgentLoop, settleAll, settleDeadRun, settleUnsettledDelegation, streamFailure } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY2, AGENT_STORE, RUN_ENDED_BEFORE_TOOL_CALL, RunCancelledError, agentFailureCode, agentRunContext, encodeStreamEvent as encodeStreamEvent2, isReplayIntegrityError as isReplayIntegrityError2, publishAgentRunFailed, releaseThreadRun, runAgentLoop, runDurationMs, settleAll, settleDeadRun, settleUnsettledDelegation, streamFailure } from "@dudousxd/nestjs-agent-core";
 import { RUN_GATEWAY, Workflow, WorkflowService } from "@dudousxd/nestjs-durable";
 import { isWorkflowControlFlowSignal } from "@dudousxd/nestjs-durable-core";
 import { Inject as Inject2, Injectable as Injectable2, Logger, Optional } from "@nestjs/common";
@@ -102,7 +102,7 @@ var AGENT_CHAT_QUEUE = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:chat-q
 // src/durable/agent-run.steps.ts
 import { createNegotiatedUiCollector } from "@dudousxd/nestjs-agent-core";
 import { prepareActionProposal, resolveActionProposalApproval } from "@dudousxd/nestjs-agent-core";
-import { AGENT_DEPS_FACTORY, announcedToolCalls, createFrameBuffer, createNoopEmitUi, encodeStreamEvent, intersectAllowLists, invokeWithTransientRetry, isControlFlowSignal, isReplayIntegrityError, observeTurnFrames, previewToolInputs, publishAgentToolRetry, registryInputPreviews, stampToolKinds, toolCallContext, traceLlmTurn, traceToolExecution, withAskTool, withChannelDelivery, withMemoryTool, withSkillTool, withToolTimeout, withTurnFrames, wrapToolPreflightDenied, wrapToolStepOutput } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY, announcedToolCalls, createFrameBuffer, createNoopEmitUi, encodeStreamEvent, intersectAllowLists, invokeWithTransientRetry, isControlFlowSignal, isReplayIntegrityError, observeFirstText, observeTurnFrames, previewToolInputs, publishAgentToolRetry, registryInputPreviews, stampToolKinds, toolCallContext, traceLlmTurn, traceToolExecution, withAskTool, withChannelDelivery, withMemoryTool, withSkillTool, withToolTimeout, withTurnFrames, wrapToolPreflightDenied, wrapToolStepOutput } from "@dudousxd/nestjs-agent-core";
 import { turnChannel } from "@dudousxd/nestjs-agent-core/genui";
 import { Step } from "@dudousxd/nestjs-durable";
 import { Inject, Injectable } from "@nestjs/common";
@@ -194,11 +194,13 @@ var AgentRunSteps = class {
     });
     const buffer = input.bufferOutput === true ? 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 = observeFirstText(input.childSink ? childSinkWriter(opened) : opened);
+      writer = firstText.writer;
     }
     const announced = announcedToolCalls(writer);
     const previews = previewToolInputs(announced.writer, registryInputPreviews(deps.registry, tools, {
@@ -292,9 +294,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
     };
   }
   /**
@@ -337,6 +346,7 @@ var AgentRunSteps = class {
       ...input.allowedTools
     ]);
     let output;
+    const toolStartedAt = Date.now();
     try {
       output = await traceToolExecution(input.ctx.runId, {
         toolCallId: input.toolCallId,
@@ -364,8 +374,9 @@ var AgentRunSteps = class {
       }
       throw error;
     }
+    const executionMs = Date.now() - toolStartedAt;
     output = await withChannelDelivery(deps, input.ctx, output, ui.components());
-    return input.collectUi === true ? wrapToolStepOutput(output, ui.components(), ui.text()) : output;
+    return input.collectUi === true ? wrapToolStepOutput(output, ui.components(), ui.text(), executionMs) : output;
   }
 };
 _ts_decorate([
@@ -765,21 +776,29 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
         });
         throw error;
       }
+      const durationMs = runDurationMs(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
       if (isReplayIntegrityError2(error)) {
         const detail = error.message;
         const code2 = agentFailureCode(error);
         this.logger.error(`agent run ${ctx.runId} failed (${code2}): ${detail}`);
         publishAgentRunFailed({
           runId: ctx.runId,
+          threadId: input.threadId,
           code: code2,
-          message: detail
+          message: detail,
+          ...timed,
+          context: agentRunContext(input)
         });
         await settleDeadRun(this.store, {
           runId: ctx.runId,
           threadId: input.threadId,
           failure: {
             code: code2,
-            message: detail
+            message: detail,
+            ...timed
           }
         });
         await (await hooks.openSink()).fail(streamFailure(error));
@@ -790,13 +809,17 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
       this.logger.error(`agent run ${ctx.runId} failed (${code}): ${message}`);
       publishAgentRunFailed({
         runId: ctx.runId,
+        threadId: input.threadId,
         code,
-        message
+        message,
+        ...timed,
+        context: agentRunContext(input)
       });
       await stepOf(ctx)("persist:run:fail", async () => {
         await this.store.recordRunEnd?.({
           runId: ctx.runId,
           status: "failed",
+          ...timed,
           errorCode: code,
           errorMessage: message
         });
@@ -846,7 +869,7 @@ __name(isSignalTimeout, "isSignalTimeout");
 
 // src/durable/durable-agent-runner.ts
 import { randomUUID } from "crypto";
-import { AGENT_SINK, AGENT_STORE as AGENT_STORE2, encodeStreamEvent as encodeStreamEvent3, releaseThreadRun as releaseThreadRun2, settleUnsettledDelegation as settleUnsettledDelegation2 } from "@dudousxd/nestjs-agent-core";
+import { AGENT_SINK, AGENT_STORE as AGENT_STORE2, encodeStreamEvent as encodeStreamEvent3, publishAgentRunCancelled, releaseThreadRun as releaseThreadRun2, settleUnsettledDelegation as settleUnsettledDelegation2 } from "@dudousxd/nestjs-agent-core";
 import { RUN_GATEWAY as RUN_GATEWAY2, WorkflowService as WorkflowService2 } from "@dudousxd/nestjs-durable";
 import { isWorkflowControlFlowSignal as isWorkflowControlFlowSignal2 } from "@dudousxd/nestjs-durable-core";
 import { Inject as Inject3, Injectable as Injectable3, Logger as Logger2, Optional as Optional2 } from "@nestjs/common";
@@ -971,6 +994,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);
+  publishAgentRunCancelled({
+    runId,
+    ...threadId !== void 0 ? {
+      threadId
+    } : {},
+    ...timed,
+    ...context !== void 0 ? {
+      context
+    } : {}
+  });
+  return timed;
+}
+__name(publishCancelled, "publishCancelled");
 var DurableAgentRunner = class _DurableAgentRunner {
   static {
     __name(this, "DurableAgentRunner");
@@ -1084,9 +1158,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
     });
   }
   /**
@@ -1134,9 +1210,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
     });
   }
   /**
@@ -6854,7 +6932,7 @@ ActionProposalController = _ts_decorate29([
 ], ActionProposalController);
 
 // src/runner/inline-agent-runner.ts
-import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY8, AGENT_STORE as AGENT_STORE8, RUN_ENDED_BEFORE_TOOL_CALL as RUN_ENDED_BEFORE_TOOL_CALL2, RunCancelledError as RunCancelledError2, agentFailureCode as agentFailureCode2, encodeStreamEvent as encodeStreamEvent5, publishAgentRunFailed as publishAgentRunFailed2, releaseThreadRun as releaseThreadRun3, runAgentLoop as runAgentLoop2, settleAll as settleAll2, settleUnsettledDelegation as settleUnsettledDelegation3, streamFailure as streamFailure3 } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY8, AGENT_STORE as AGENT_STORE8, RUN_ENDED_BEFORE_TOOL_CALL as RUN_ENDED_BEFORE_TOOL_CALL2, RunCancelledError as RunCancelledError2, agentFailureCode as agentFailureCode2, agentRunContext as agentRunContext2, encodeStreamEvent as encodeStreamEvent5, publishAgentRunCancelled as publishAgentRunCancelled2, publishAgentRunFailed as publishAgentRunFailed2, releaseThreadRun as releaseThreadRun3, runAgentLoop as runAgentLoop2, runDurationMs as runDurationMs2, settleAll as settleAll2, settleUnsettledDelegation as settleUnsettledDelegation3, streamFailure as streamFailure3 } from "@dudousxd/nestjs-agent-core";
 import { Inject as Inject28, Injectable as Injectable14, Logger as Logger9, Optional as Optional12 } from "@nestjs/common";
 function _ts_decorate30(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
@@ -6947,8 +7025,12 @@ var InlineAgentRunner = class _InlineAgentRunner {
       ...deps,
       day
     }, input, hooks).then(() => releaseThreadRun3(this.store, input.threadId, runId)).catch(async (error) => {
+      const durationMs = runDurationMs2(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
       if (error instanceof RunCancelledError2) {
-        await this.settleCancelled(runId, input, deps);
+        await this.settleCancelled(runId, input, deps, durationMs);
         return;
       }
       const message = error instanceof Error ? error.message : String(error);
@@ -6956,12 +7038,16 @@ var InlineAgentRunner = class _InlineAgentRunner {
       this.logger.error(`agent run ${runId} failed (${code}): ${message}`);
       publishAgentRunFailed2({
         runId,
+        threadId: input.threadId,
         code,
-        message
+        message,
+        ...timed,
+        context: agentRunContext2(input)
       });
       await this.store.recordRunEnd?.({
         runId,
         status: "failed",
+        ...timed,
         errorCode: code,
         errorMessage: message
       });
@@ -7033,8 +7119,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
+    } : {};
+    publishAgentRunCancelled2({
+      runId,
+      threadId: input.threadId,
+      ...timed,
+      context: agentRunContext2(input)
+    });
     const writer = await deps.sink.open(runId);
     await this.settleQueue(writer, input, runId, "cancelled");
     await releaseThreadRun3(this.store, input.threadId, runId);
@@ -7044,7 +7139,8 @@ var InlineAgentRunner = class _InlineAgentRunner {
     await writer.end();
     await this.store.recordRunEnd?.({
       runId,
-      status: "cancelled"
+      status: "cancelled",
+      ...timed
     });
   }
   /**
@@ -7230,10 +7326,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
         }
       }), "startAgent")
     };
-    void runAgentLoop2({
-      ...deps,
-      day
-    }, {
+    const childInput = {
       threadId: subThread.id,
       actor,
       userText: task,
@@ -7243,13 +7336,29 @@ var InlineAgentRunner = class _InlineAgentRunner {
       delegationPath: path,
       parentRunId,
       deliverTo
-    }, hooks).catch(async (error) => {
+    };
+    void runAgentLoop2({
+      ...deps,
+      day
+    }, childInput, hooks).catch(async (error) => {
       const cancelled = error instanceof RunCancelledError2;
       const message = error instanceof Error ? error.message : String(error);
-      if (!cancelled) {
+      const durationMs = runDurationMs2(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
+      const settled = {
+        runId,
+        threadId: subThread.id,
+        ...timed,
+        context: agentRunContext2(childInput)
+      };
+      if (cancelled) {
+        publishAgentRunCancelled2(settled);
+      } else {
         this.logger.error(`detached agent run ${runId} failed: ${message}`);
         publishAgentRunFailed2({
-          runId,
+          ...settled,
           code: agentFailureCode2(error),
           message
         });
@@ -7257,6 +7366,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
       await this.store.recordRunEnd?.({
         runId,
         status: cancelled ? "cancelled" : "failed",
+        ...timed,
         ...cancelled ? {} : {
           errorCode: agentFailureCode2(error),
           errorMessage: message