@dudousxd/nestjs-agent 1.4.0 → 1.5.0

package/dist/durable/index.cjs

@@ -143,8 +143,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 = (0, import_nestjs_agent_core.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((0, import_nestjs_agent_core.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 : {
@@ -154,11 +161,12 @@ var AgentRunSteps = class {
         classify: localClassify
       } : {}
     };
-    return (0, import_nestjs_agent_core.traceToolExecution)(input.ctx.runId, {
+    const output = await (0, import_nestjs_agent_core.traceToolExecution)(input.ctx.runId, {
       toolCallId: input.toolCallId,
       toolName: input.toolName,
       toolType: input.toolType
     }, () => (0, import_nestjs_agent_core.invokeWithTransientRetry)(() => {
+      ui.restart();
       const invocation = deps.registry.invoke(input.toolName, input.input, ctx, deps.rolesPolicy);
       return input.timeoutMs !== void 0 ? (0, import_nestjs_agent_core.withToolTimeout)(invocation, input.timeoutMs, input.toolName) : invocation;
     }, transientRetry, {
@@ -171,6 +179,7 @@ var AgentRunSteps = class {
         });
       }, "onRetry")
     }));
+    return input.collectUi === true ? (0, import_nestjs_agent_core.wrapToolStepOutput)(output, ui.components()) : output;
   }
 };
 _ts_decorate([
@@ -414,7 +423,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")
       } : {}
     };
@@ -3146,6 +3158,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"),