@dudousxd/nestjs-agent 1.23.0 → 1.23.1

package/dist/index.cjs

@@ -6452,7 +6452,10 @@ var AgUiRunHandler = class {
     if (owner !== null && owner !== actor.id) {
       throw new import_common28.ForbiddenException("thread belongs to another actor");
     }
-    if (owner !== null && turn.media.length === 0) {
+    if (forwarded.regenerate === true && owner === null) {
+      throw invalid("invalid_input", "regenerate needs a thread that already has an exchange");
+    }
+    if (owner !== null && forwarded.regenerate !== true && turn.media.length === 0 && turn.staged.length === 0) {
       const decision = await this.agent.handleTextDecision(input.threadId, actor, turn.text);
       if ("proposalDecision" in decision) {
         this.writeEvents(res, [
@@ -6467,7 +6470,7 @@ var AgUiRunHandler = class {
         return;
       }
     }
-    const refs = await this.stage(actor, turn.media, warnings);
+    const refs = await this.stage(actor, turn.media, warnings, turn.staged);
     if (turn.text.trim().length === 0 && refs.length === 0) {
       throw invalid("no_user_message", "the user message to answer is empty");
     }
@@ -6503,6 +6506,9 @@ var AgUiRunHandler = class {
       } : {},
       ...refs.length > 0 ? {
         attachments: refs
+      } : {},
+      ...forwarded.regenerate === true ? {
+        regenerate: true
       } : {}
     });
     await this.pipe(res, started.runId, {
@@ -6514,10 +6520,25 @@ var AgUiRunHandler = class {
       preamble: warningEvents(warnings)
     });
   }
-  /** Stage the turn's inline media through the bound store, within its limits; say what was not used. */
-  async stage(actor, media, warnings) {
+  /**
+  * The turn's attachments: the ones already staged (by mediaId), then its inline media staged
+  * through the bound store — all within the store's limits; say what was not used.
+  */
+  async stage(actor, media, warnings, staged = []) {
     const refs = [];
     const limits = attachmentLimits(this.options, this.staging);
+    for (const ref of staged) {
+      const label = `The attachment ${ref.mediaId} was not used`;
+      if (this.staging === void 0) {
+        warnings.push(`${label}: attachments are not enabled on this agent.`);
+      } else if (refs.length >= limits.maxPerMessage) {
+        warnings.push(`${label}: a message carries at most ${limits.maxPerMessage} attachments.`);
+      } else {
+        refs.push({
+          mediaId: ref.mediaId
+        });
+      }
+    }
     for (const item of media) {
       const label = `The ${item.kind} "${item.filename}" was not used`;
       if (this.staging === void 0) {
@@ -6530,7 +6551,7 @@ var AgUiRunHandler = class {
         warnings.push(`${label}: it is larger than ${limits.maxBytes} bytes.`);
       } else {
         try {
-          const staged = await this.staging.stage({
+          const staged2 = await this.staging.stage({
             data: item.data,
             filename: item.filename,
             contentType: item.contentType,
@@ -6538,7 +6559,7 @@ var AgUiRunHandler = class {
             actor
           });
           refs.push({
-            mediaId: staged.mediaId
+            mediaId: staged2.mediaId
           });
         } catch {
           warnings.push(`${label}: it could not be stored.`);
@@ -6656,11 +6677,17 @@ var AgUiRunHandler = class {
     res.setHeader("X-Agent-Run-Id", runId);
     res.flushHeaders();
     let terminal = false;
+    const cursor = {
+      seq: 0
+    };
     try {
       const frames = (0, import_ag_ui.agUiFramesFromNdjson)(this.agent.subscribe(runId));
-      for await (const event of (0, import_ag_ui.agUiEvents)(frames, options)) {
+      for await (const event of (0, import_ag_ui.agUiEvents)(frames, {
+        ...options,
+        cursor
+      })) {
         if (event.type === "RUN_FINISHED" || event.type === "RUN_ERROR") terminal = true;
-        res.write((0, import_ag_ui.agUiSse)(event));
+        res.write((0, import_ag_ui.agUiSse)(event, cursor.seq));
       }
     } catch {
       if (!terminal) {