@dudousxd/nestjs-agent 0.4.0 → 0.4.1

package/dist/durable/index.cjs

@@ -82,7 +82,7 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
     const day = input.day ?? utcDay();
     const deps = this.factory.forAgent(input.agentName);
     if (input.sinkRunId !== void 0) {
-      await ctx.step("activate", () => this.store.setActiveStream(input.threadId, ctx.runId));
+      await ctx.localStep("activate", () => this.store.setActiveStream(input.threadId, ctx.runId));
     }
     const sinkRunId = input.sinkRunId ?? ctx.runId;
     const hooks = {
@@ -90,9 +90,9 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
       // A child forwards into the top-level sink but must not end/fail it (the top-level run owns it).
       openSink: /* @__PURE__ */ __name(async () => input.sinkRunId !== void 0 ? childSinkWriter(await deps.sink.open(sinkRunId)) : deps.sink.open(ctx.runId), "openSink"),
       awaitApproval: /* @__PURE__ */ __name((call) => ctx.waitForSignal(`tool:${ctx.runId}:${call.id}`), "awaitApproval"),
-      step: /* @__PURE__ */ __name((name, fn) => ctx.step(name, fn), "step"),
+      step: /* @__PURE__ */ __name((name, fn) => ctx.localStep(name, fn), "step"),
       runAgent: /* @__PURE__ */ __name(async (agentName, task) => {
-        const subThreadId = await ctx.step(`subthread:${agentName}`, async () => {
+        const subThreadId = await ctx.localStep(`subthread:${agentName}`, async () => {
           const thread = await this.store.createThread({
             actor: input.actor,
             transient: true
@@ -115,7 +115,7 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
         ...deps,
         day
       }, input, hooks);
-      await ctx.step("deactivate", () => this.store.setActiveStream(input.threadId, null));
+      await ctx.localStep("deactivate", () => this.store.setActiveStream(input.threadId, null));
       return result;
     } catch (error) {
       if (error instanceof import_nestjs_durable_core.WorkflowSuspended || error instanceof import_nestjs_durable_core.ContinueAsNew) {
@@ -128,7 +128,7 @@ var AgentRunWorkflow = class _AgentRunWorkflow {
         code,
         message
       });
-      await ctx.step("deactivate", () => this.store.setActiveStream(input.threadId, null));
+      await ctx.localStep("deactivate", () => this.store.setActiveStream(input.threadId, null));
       const writer = await hooks.openSink();
       await writer.fail({
         code,