@dudousxd/nestjs-agent 1.35.1 → 1.36.0

package/dist/index.js

@@ -5815,7 +5815,7 @@ ActionProposalController = _ts_decorate25([
 var AGENT_CHAT_QUEUE = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent:chat-queue");
 
 // src/runner/inline-agent-runner.ts
-import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY6, AGENT_STORE as AGENT_STORE6, RUN_ENDED_BEFORE_TOOL_CALL, RunCancelledError, agentFailureCode, encodeStreamEvent as encodeStreamEvent2, publishAgentRunFailed, releaseThreadRun, runAgentLoop, settleAll, settleUnsettledDelegation, streamFailure as streamFailure2 } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY6, AGENT_STORE as AGENT_STORE6, RUN_ENDED_BEFORE_TOOL_CALL, RunCancelledError, agentFailureCode, agentRunContext, encodeStreamEvent as encodeStreamEvent2, publishAgentRunCancelled, publishAgentRunFailed, releaseThreadRun, runAgentLoop, runDurationMs, settleAll, settleUnsettledDelegation, streamFailure as streamFailure2 } from "@dudousxd/nestjs-agent-core";
 import { Inject as Inject25, Injectable as Injectable11, Logger as Logger7, Optional as Optional10 } from "@nestjs/common";
 function _ts_decorate26(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
@@ -5908,8 +5908,12 @@ var InlineAgentRunner = class _InlineAgentRunner {
       ...deps,
       day
     }, input, hooks).then(() => releaseThreadRun(this.store, input.threadId, runId)).catch(async (error) => {
+      const durationMs = runDurationMs(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
       if (error instanceof RunCancelledError) {
-        await this.settleCancelled(runId, input, deps);
+        await this.settleCancelled(runId, input, deps, durationMs);
         return;
       }
       const message = error instanceof Error ? error.message : String(error);
@@ -5917,12 +5921,16 @@ var InlineAgentRunner = class _InlineAgentRunner {
       this.logger.error(`agent run ${runId} failed (${code}): ${message}`);
       publishAgentRunFailed({
         runId,
+        threadId: input.threadId,
         code,
-        message
+        message,
+        ...timed,
+        context: agentRunContext(input)
       });
       await this.store.recordRunEnd?.({
         runId,
         status: "failed",
+        ...timed,
         errorCode: code,
         errorMessage: message
       });
@@ -5994,8 +6002,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
+    } : {};
+    publishAgentRunCancelled({
+      runId,
+      threadId: input.threadId,
+      ...timed,
+      context: agentRunContext(input)
+    });
     const writer = await deps.sink.open(runId);
     await this.settleQueue(writer, input, runId, "cancelled");
     await releaseThreadRun(this.store, input.threadId, runId);
@@ -6005,7 +6022,8 @@ var InlineAgentRunner = class _InlineAgentRunner {
     await writer.end();
     await this.store.recordRunEnd?.({
       runId,
-      status: "cancelled"
+      status: "cancelled",
+      ...timed
     });
   }
   /**
@@ -6191,10 +6209,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
         }
       }), "startAgent")
     };
-    void runAgentLoop({
-      ...deps,
-      day
-    }, {
+    const childInput = {
       threadId: subThread.id,
       actor,
       userText: task,
@@ -6204,13 +6219,29 @@ var InlineAgentRunner = class _InlineAgentRunner {
       delegationPath: path,
       parentRunId,
       deliverTo
-    }, hooks).catch(async (error) => {
+    };
+    void runAgentLoop({
+      ...deps,
+      day
+    }, childInput, hooks).catch(async (error) => {
       const cancelled = error instanceof RunCancelledError;
       const message = error instanceof Error ? error.message : String(error);
-      if (!cancelled) {
+      const durationMs = runDurationMs(error);
+      const timed = durationMs !== void 0 ? {
+        durationMs
+      } : {};
+      const settled = {
+        runId,
+        threadId: subThread.id,
+        ...timed,
+        context: agentRunContext(childInput)
+      };
+      if (cancelled) {
+        publishAgentRunCancelled(settled);
+      } else {
         this.logger.error(`detached agent run ${runId} failed: ${message}`);
         publishAgentRunFailed({
-          runId,
+          ...settled,
           code: agentFailureCode(error),
           message
         });
@@ -6218,6 +6249,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
       await this.store.recordRunEnd?.({
         runId,
         status: cancelled ? "cancelled" : "failed",
+        ...timed,
         ...cancelled ? {} : {
           errorCode: agentFailureCode(error),
           errorMessage: message