@dudousxd/nestjs-agent 1.24.2 → 1.25.0

package/dist/durable/index.js

@@ -35,7 +35,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, createFrameBuffer, createNoopEmitUi, encodeStreamEvent, intersectAllowLists, invokeWithTransientRetry, isControlFlowSignal, isReplayIntegrityError, observeTurnFrames, publishAgentToolRetry, stampToolKinds, toolCallContext, traceLlmTurn, traceToolExecution, withAskTool, withMemoryTool, withSkillTool, withToolTimeout, withTurnFrames, wrapToolPreflightDenied, wrapToolStepOutput } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY, createFrameBuffer, createNoopEmitUi, encodeStreamEvent, intersectAllowLists, invokeWithTransientRetry, isControlFlowSignal, isReplayIntegrityError, observeTurnFrames, previewToolInputs, publishAgentToolRetry, registryInputPreviews, stampToolKinds, toolCallContext, traceLlmTurn, traceToolExecution, withAskTool, withMemoryTool, withSkillTool, withToolTimeout, withTurnFrames, wrapToolPreflightDenied, wrapToolStepOutput } from "@dudousxd/nestjs-agent-core";
 import { Step } from "@dudousxd/nestjs-durable";
 import { Inject, Injectable } from "@nestjs/common";
 
@@ -131,7 +131,19 @@ var AgentRunSteps = class {
       const opened = await deps.sink.open(input.sinkRunId);
       writer = input.childSink ? childSinkWriter(opened) : opened;
     }
-    const frames = observeTurnFrames(writer);
+    const previews = previewToolInputs(writer, registryInputPreviews(deps.registry, tools, {
+      actor: input.actor,
+      ...input.threadId !== void 0 ? {
+        threadId: input.threadId
+      } : {},
+      ...input.agentName !== void 0 ? {
+        agentName: input.agentName
+      } : {},
+      ...input.preflightContext?.uiCapabilities !== void 0 ? {
+        uiCapabilities: input.preflightContext.uiCapabilities
+      } : {}
+    }));
+    const frames = observeTurnFrames(previews.writer);
     const turn = withTurnFrames(await traceLlmTurn(input.runId, input.step, () => deps.model.runTurn({
       system: input.system,
       messages: input.messages,
@@ -203,8 +215,13 @@ var AgentRunSteps = class {
         };
       }
     }
-    return buffer === void 0 ? stamped : {
+    const shown = previews.shown();
+    const previewed = shown.length > 0 ? {
       ...stamped,
+      previews: shown
+    } : stamped;
+    return buffer === void 0 ? previewed : {
+      ...previewed,
       bufferedFrames: buffer.frames()
     };
   }
@@ -1797,9 +1814,13 @@ var ActionProposalService = class {
   /** The words a text decision is made of, and what the agent answers — English unless configured. */
   vocabulary;
   replies;
+  hasCapability() {
+    const store = this.store;
+    return typeof store.getThreadActionProposalScope === "function" && typeof store.listActionProposals === "function";
+  }
   capability() {
     const store = this.store;
-    if (typeof store.getThreadActionProposalScope !== "function" || typeof store.listActionProposals !== "function") throw new NotImplementedException("Action proposals are unavailable");
+    if (!this.hasCapability()) throw new NotImplementedException("Action proposals are unavailable");
     return store;
   }
   async scope(threadId, actor) {
@@ -1815,7 +1836,16 @@ var ActionProposalService = class {
       requesterRef: proposal.actorRef
     });
   }
+  /**
+  * One page of a thread's proposals this actor may see. A store without the proposal capability
+  * can only run blocking approvals, so a thread there has no proposals: an empty, final page rather
+  * than `501` — a client that lists by default (React's `useAgentChat`) needs no flag. Deciding one
+  * still answers `501`.
+  */
   async listPage(threadId, actor, query = {}) {
+    if (!this.hasCapability()) return {
+      items: []
+    };
     const scope = await this.scope(threadId, actor);
     const rows = await this.capability().listActionProposals(scope, {
       limit: 1e3,
@@ -2549,12 +2579,13 @@ var AgentService = class {
   }
   /**
   * The proposals of a thread this actor may see — the thread's own actor, or one the policy lets
-  * decide them; what `GET <base>/threads/:threadId/action-proposals` serves (its first page). `404`
-  * when independent approvals are not on.
+  * decide them; what `GET <base>/threads/:threadId/action-proposals` serves (its first page).
+  * Without independent approvals (the default blocking mode) there are none, so the read answers an
+  * empty list rather than refusing; deciding one still refuses.
   */
   async listActionProposals(actor, threadId) {
     if (this.options?.actionApprovalMode !== "independent" || !this.proposals) {
-      throw new NotFoundException2("Proposal not found");
+      return [];
     }
     return this.proposals.list(threadId, actor);
   }
@@ -6433,6 +6464,11 @@ var InlineAgentRunner = class _InlineAgentRunner {
   * running it. A cancelled id is dropped once the run settles, so the set tracks live runs only.
   */
   cancelled = /* @__PURE__ */ new Set();
+  /**
+  * Aborted by {@link cancel}: what cuts short the model call or tool a stopped run is in (the
+  * loop's `abortSignal` hook). Keyed like {@link cancelled}, and dropped with it.
+  */
+  aborts = /* @__PURE__ */ new Map();
   /** Top-level runs this process is running now — what {@link isRunActive} answers from. */
   live = /* @__PURE__ */ new Set();
   constructor(factory, store, queue) {
@@ -6502,7 +6538,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
       await releaseThreadRun3(this.store, input.threadId, runId);
       await writer.fail(streamFailure3(error));
     }).finally(() => {
-      this.cancelled.delete(runId);
+      this.forgetCancel(runId);
       this.live.delete(runId);
     });
     return {
@@ -6521,8 +6557,10 @@ var InlineAgentRunner = class _InlineAgentRunner {
   * Ask a run to stop, and settle whatever it was parked on.
   *
   * The loop observes the request at its next safe point (between steps, before the turn's tools) and
-  * unwinds from there — so a tool already executing is left to finish, and its result is recorded
-  * exactly as it would have been. What this cannot wait for is a turn parked on a human: that wait
+  * unwinds from there. What the run is in right now is aborted (its `abortSignal`): the model call
+  * streaming the answer ends at once and the step it was writing is not persisted, and a tool that
+  * passes `ctx.abortSignal` on is cut short and recorded failed; a tool that ignores the signal is
+  * left to finish, and its result is recorded exactly as it would have been. What this cannot wait for is a turn parked on a human: that wait
   * is a promise nobody is going to resolve, so it is REJECTED here with the same
   * {@link RunCancelledError} the loop would have thrown, and the run unwinds through the identical
   * path.
@@ -6533,6 +6571,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
   */
   async cancel(runId) {
     this.cancelled.add(runId);
+    this.aborts.get(runId)?.abort(new RunCancelledError2());
     for (const [key, waiter] of this.pending) {
       if (key.startsWith(`${runId}:`)) {
         this.pending.delete(key);
@@ -6541,6 +6580,20 @@ var InlineAgentRunner = class _InlineAgentRunner {
     }
     await Promise.resolve();
   }
+  /** The signal {@link cancel} aborts for `runId` — one per run, shared by its awaited delegates. */
+  abortSignalFor(runId) {
+    let controller = this.aborts.get(runId);
+    if (controller === void 0) {
+      controller = new AbortController();
+      this.aborts.set(runId, controller);
+    }
+    return controller.signal;
+  }
+  /** Drop a settled run's Stop: its flag and its abort controller. */
+  forgetCancel(runId) {
+    this.cancelled.delete(runId);
+    this.aborts.delete(runId);
+  }
   /**
   * The single place a cancelled run's state is settled: recorded as its own terminal (never
   * `failed` — a user pressing Stop must not land in someone's error rate), the thread released, and
@@ -6660,6 +6713,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
       step: /* @__PURE__ */ __name((_name, fn) => fn(), "step"),
       parallel: settleAll2,
       cancelled: /* @__PURE__ */ __name(async () => this.cancelled.has(runId), "cancelled"),
+      abortSignal: this.abortSignalFor(runId),
       runAgent: /* @__PURE__ */ __name((agentName, task) => this.runNested({
         agentName,
         task,
@@ -6711,6 +6765,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
       step: /* @__PURE__ */ __name((_name, fn) => fn(), "step"),
       parallel: settleAll2,
       cancelled: /* @__PURE__ */ __name(async () => this.cancelled.has(runId), "cancelled"),
+      abortSignal: this.abortSignalFor(runId),
       runAgent: /* @__PURE__ */ __name((childName, childTask) => this.runNested({
         agentName: childName,
         task: childTask,
@@ -6786,7 +6841,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
         }
       });
     }).finally(async () => {
-      this.cancelled.delete(runId);
+      this.forgetCancel(runId);
       await this.store.setActiveStream(subThread.id, null);
     });
     return {
@@ -6818,6 +6873,8 @@ var InlineAgentRunner = class _InlineAgentRunner {
       // A child stops when it is cancelled OR when the run a human is actually watching is: the
       // canceller names the top-level run, which is the only id that ever left the server.
       cancelled: /* @__PURE__ */ __name(async () => this.cancelled.has(runId) || this.cancelled.has(sinkRunId), "cancelled"),
+      // Aborted by a Stop on the run the human is watching, which is the id a canceller names.
+      abortSignal: this.abortSignalFor(sinkRunId),
       runAgent: /* @__PURE__ */ __name((childName, childTask) => this.runNested({
         agentName: childName,
         task: childTask,