@dudousxd/nestjs-agent 1.3.0 → 1.5.0

package/dist/durable/index.d.cts

@@ -83,6 +83,13 @@ interface DispatchedLlmInput extends LlmStepEnvelope {
 interface DispatchedToolInput extends ToolStepEnvelope {
     toolCallId: string;
     toolType: 'read' | 'action';
+    /**
+     * Where a component the tool pushes (`ctx.emitUi`) streams live — the same routing as
+     * {@link DispatchedLlmInput}. Absent on a task enqueued by a workflow that predates it: the
+     * components are then still collected and persisted, just not streamed.
+     */
+    sinkRunId?: string;
+    childSink?: boolean;
 }
 /**
  * The two long steps every turn dispatches as routed remote steps (`AgentRunSteps.llm` /

package/dist/durable/index.d.ts

@@ -83,6 +83,13 @@ interface DispatchedLlmInput extends LlmStepEnvelope {
 interface DispatchedToolInput extends ToolStepEnvelope {
     toolCallId: string;
     toolType: 'read' | 'action';
+    /**
+     * Where a component the tool pushes (`ctx.emitUi`) streams live — the same routing as
+     * {@link DispatchedLlmInput}. Absent on a task enqueued by a workflow that predates it: the
+     * components are then still collected and persisted, just not streamed.
+     */
+    sinkRunId?: string;
+    childSink?: boolean;
 }
 /**
  * The two long steps every turn dispatches as routed remote steps (`AgentRunSteps.llm` /

package/dist/durable/index.js

@@ -24,7 +24,7 @@ function childSinkWriter(writer) {
 __name(childSinkWriter, "childSinkWriter");
 
 // src/durable/agent-run.steps.ts
-import { AGENT_DEPS_FACTORY, createFrameBuffer, invokeWithTransientRetry, observeTurnFrames, publishAgentToolRetry, stampToolKinds, traceLlmTurn, traceToolExecution, withAskTool, withMemoryTool, withSkillTool, withToolTimeout, withTurnFrames } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY, createFrameBuffer, createUiCollector, encodeStreamEvent, invokeWithTransientRetry, observeTurnFrames, publishAgentToolRetry, stampToolKinds, traceLlmTurn, traceToolExecution, withAskTool, withMemoryTool, withSkillTool, withToolTimeout, withTurnFrames, wrapToolStepOutput } from "@dudousxd/nestjs-agent-core";
 import { Step } from "@dudousxd/nestjs-durable";
 import { Inject, Injectable } from "@nestjs/common";
 function _ts_decorate(decorators, target, key, desc) {
@@ -115,8 +115,15 @@ var AgentRunSteps = class {
     if (deps.registry.spec(input.toolName)?.kind === "action" && input.toolType !== "action") {
       throw new Error(`tool "${input.toolName}" is an action tool but was dispatched without approval: the dispatching process resolved its kind against a registry that does not have it`);
     }
+    const sinkRunId = input.sinkRunId;
+    let opened;
+    const ui = createUiCollector(input.toolCallId, sinkRunId === void 0 ? void 0 : async (event) => {
+      opened ??= Promise.resolve(deps.sink.open(sinkRunId)).then((writer) => input.childSink === true ? childSinkWriter(writer) : writer);
+      await (await opened).write(encodeStreamEvent(event));
+    });
     const ctx = {
-      ...input.ctx
+      ...input.ctx,
+      emitUi: ui.emit
     };
     const localClassify = deps.toolTransientRetry !== false ? deps.toolTransientRetry?.classify : void 0;
     const transientRetry = input.transientRetry === false ? false : {
@@ -126,11 +133,12 @@ var AgentRunSteps = class {
         classify: localClassify
       } : {}
     };
-    return traceToolExecution(input.ctx.runId, {
+    const output = await traceToolExecution(input.ctx.runId, {
       toolCallId: input.toolCallId,
       toolName: input.toolName,
       toolType: input.toolType
     }, () => invokeWithTransientRetry(() => {
+      ui.restart();
       const invocation = deps.registry.invoke(input.toolName, input.input, ctx, deps.rolesPolicy);
       return input.timeoutMs !== void 0 ? withToolTimeout(invocation, input.timeoutMs, input.toolName) : invocation;
     }, transientRetry, {
@@ -143,6 +151,7 @@ var AgentRunSteps = class {
         });
       }, "onRetry")
     }));
+    return input.collectUi === true ? wrapToolStepOutput(output, ui.components()) : output;
   }
 };
 _ts_decorate([
@@ -386,7 +395,10 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
         dispatchTool: /* @__PURE__ */ __name((call, envelope) => ctx.step(AgentRunSteps.prototype.tool, {
           ...envelope,
           toolCallId: call.id,
-          toolType: call.kind === "action" ? "action" : "read"
+          toolType: call.kind === "action" ? "action" : "read",
+          // Where `ctx.emitUi` streams from the worker — the llm step's routing.
+          sinkRunId,
+          childSink: input.sinkRunId !== void 0
         }), "dispatchTool")
       } : {}
     };
@@ -471,7 +483,7 @@ __name(isSignalTimeout, "isSignalTimeout");
 
 // src/durable/durable-agent-runner.ts
 import { randomUUID } from "crypto";
-import { AGENT_SINK, AGENT_STORE as AGENT_STORE2, encodeStreamEvent } from "@dudousxd/nestjs-agent-core";
+import { AGENT_SINK, AGENT_STORE as AGENT_STORE2, encodeStreamEvent as encodeStreamEvent2 } from "@dudousxd/nestjs-agent-core";
 import { RUN_GATEWAY as RUN_GATEWAY2, WorkflowService } from "@dudousxd/nestjs-durable";
 import { isWorkflowControlFlowSignal as isWorkflowControlFlowSignal2 } from "@dudousxd/nestjs-durable-core";
 import { Inject as Inject3, Injectable as Injectable3, Logger } from "@nestjs/common";
@@ -650,7 +662,7 @@ var DurableAgentRunner = class _DurableAgentRunner {
       await this.store.setActiveStream(threadId, null);
     }
     const writer = await this.sink.open(runId);
-    await writer.write(encodeStreamEvent({
+    await writer.write(encodeStreamEvent2({
       kind: "cancelled"
     }));
     await writer.end();
@@ -1048,7 +1060,7 @@ AgentDepsFactory2 = _ts_decorate5([
 ], AgentDepsFactory2);
 
 // src/agent.service.ts
-import { AGENT_ATTACHMENT_STAGING, AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY3, AGENT_QUOTA_STORE as AGENT_QUOTA_STORE2, AGENT_RUNNER, AGENT_STORE as AGENT_STORE4, REQUESTER_APPROVER, mayDecideApproval } from "@dudousxd/nestjs-agent-core";
+import { AGENT_ATTACHMENT_STAGING, AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY3, AGENT_QUOTA_STORE as AGENT_QUOTA_STORE2, AGENT_RUNNER, AGENT_STORE as AGENT_STORE4, REQUESTER_APPROVER, mayDecideApproval, readElicitationQuestions, validateElicitationAnswer } from "@dudousxd/nestjs-agent-core";
 import { BadRequestException, ForbiddenException, GoneException, Inject as Inject5, Injectable as Injectable5, NotFoundException, NotImplementedException, Optional as Optional3 } from "@nestjs/common";
 function _ts_decorate6(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
@@ -1224,6 +1236,7 @@ var AgentService = class {
   */
   async answer(actor, toolCallId2, answers2 = {}) {
     await this.assertOwnsToolCall(actor, toolCallId2);
+    await this.assertAnswersFit(toolCallId2, answers2);
     const reply = {
       answers: answers2,
       answeredByRef: actor.id
@@ -1492,6 +1505,27 @@ var AgentService = class {
     }
   }
   /**
+  * Check a reply against the questions it answers before it is signalled: every submitted value a
+  * question's own rules accept (a typed input's type, bounds and pattern; a pick from the offered
+  * options), and every `required` question answered — by the reply, or by the defaults an omitted
+  * question falls back to. A reply the loop would have to drop parts of is refused here instead, so
+  * the person who typed it hears why. Needs the store to hand back the call's recorded questions;
+  * without that seam the loop's own filtering is the only check.
+  */
+  async assertAnswersFit(toolCallId2, answers2) {
+    if (this.store.toolCallInput === void 0) {
+      return;
+    }
+    const questions = readElicitationQuestions(await this.store.toolCallInput(toolCallId2));
+    for (const question of questions) {
+      const submitted = Object.hasOwn(answers2, question.id) ? answers2[question.id] : void 0;
+      const problem = validateElicitationAnswer(question, submitted ?? question.defaults ?? []);
+      if (problem !== null) {
+        throw new BadRequestException(`answers["${question.id}"] ${problem}`);
+      }
+    }
+  }
+  /**
   * Authorization seam for approve/reject. The call's recorded approver decides who may settle it:
   * the requester (the default, and every call recorded before approvers existed) is the thread's
   * own actor — the ownership check this always was; any other approver goes through the policy's
@@ -3096,6 +3130,9 @@ var AiToolDiscoveryService = class _AiToolDiscoveryService {
         } : {},
         ...meta.presentation !== void 0 ? {
           presentation: meta.presentation
+        } : {},
+        ...meta.terminal === true ? {
+          terminal: true
         } : {}
       }, {
         execute: /* @__PURE__ */ __name((input, ctx) => instance.execute(input, ctx), "execute"),
@@ -3305,7 +3342,7 @@ var LedgerQuotaStore = class {
 };
 
 // src/runner/inline-agent-runner.ts
-import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY6, AGENT_STORE as AGENT_STORE5, RunCancelledError as RunCancelledError2, agentFailureCode as agentFailureCode2, encodeStreamEvent as encodeStreamEvent2, publishAgentRunFailed as publishAgentRunFailed2, runAgentLoop as runAgentLoop2, settleAll as settleAll2, settleUnsettledDelegation as settleUnsettledDelegation2 } from "@dudousxd/nestjs-agent-core";
+import { AGENT_DEPS_FACTORY as AGENT_DEPS_FACTORY6, AGENT_STORE as AGENT_STORE5, RunCancelledError as RunCancelledError2, agentFailureCode as agentFailureCode2, encodeStreamEvent as encodeStreamEvent3, publishAgentRunFailed as publishAgentRunFailed2, runAgentLoop as runAgentLoop2, settleAll as settleAll2, settleUnsettledDelegation as settleUnsettledDelegation2 } from "@dudousxd/nestjs-agent-core";
 import { Inject as Inject18, Injectable as Injectable10, Logger as Logger5 } from "@nestjs/common";
 function _ts_decorate20(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
@@ -3443,7 +3480,7 @@ var InlineAgentRunner = class _InlineAgentRunner {
     this.logger.log(`agent run ${runId} cancelled`);
     await this.store.setActiveStream(threadId, null);
     const writer = await deps.sink.open(runId);
-    await writer.write(encodeStreamEvent2({
+    await writer.write(encodeStreamEvent3({
       kind: "cancelled"
     }));
     await writer.end();