@dudousxd/nestjs-agent 1.4.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();
@@ -3118,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"),
@@ -3327,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;
@@ -3465,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();