@dudousxd/nestjs-agent-core 0.42.0 → 0.43.0

package/dist/ag-ui/index.cjs

@@ -22,6 +22,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var ag_ui_exports = {};
 __export(ag_ui_exports, {
   AG_UI_CUSTOM: () => AG_UI_CUSTOM,
+  AG_UI_MEDIA_PROVIDER: () => AG_UI_MEDIA_PROVIDER,
   AG_UI_PROTOCOL_VERSION: () => AG_UI_PROTOCOL_VERSION,
   AgUiEncoder: () => AgUiEncoder,
   actionProposalDecisionEvents: () => actionProposalDecisionEvents,
@@ -30,6 +31,7 @@ __export(ag_ui_exports, {
   agUiSse: () => agUiSse,
   decodeInterruptId: () => decodeInterruptId,
   encodeInterruptId: () => encodeInterruptId,
+  frameSeq: () => frameSeq,
   parseRunInput: () => parseRunInput,
   planResume: () => planResume,
   readAnswersPayload: () => readAnswersPayload,
@@ -87,6 +89,7 @@ __name(decodeInterruptId, "decodeInterruptId");
 
 // src/ag-ui/types.ts
 var AG_UI_PROTOCOL_VERSION = "1.0";
+var AG_UI_MEDIA_PROVIDER = "nestjs-agent";
 var AG_UI_CUSTOM = {
   /** `{ runId, threadId }` — the library's own ids for the run behind this stream (cancel, re-attach). */
   run: "agora.run",
@@ -97,7 +100,7 @@ var AG_UI_CUSTOM = {
   /** The approval request as the native protocol carries it, for a consumer that renders it live. */
   approvalRequested: "agora.approval-requested",
   approvalSettled: "agora.approval-settled",
-  /** The whole question set of an elicitation (`ElicitationRequest`). */
+  /** `{ id, runId, request }` — the whole question set of an elicitation, under its tool-call id. */
   elicitation: "agora.elicitation",
   /** `{ usage, costUsd, reasoningMs }` of one model step. */
   stepUsage: "agora.step-usage",
@@ -376,6 +379,7 @@ var AgUiEncoder = class {
         });
         return [
           this.custom(AG_UI_CUSTOM.elicitation, {
+            id: event.id,
             runId: parked,
             request: event.request
           })
@@ -431,7 +435,7 @@ var AgUiEncoder = class {
           sawDelta: false,
           answered: false
         });
-        out.push(this.callStart(event.id, event.name));
+        out.push(this.callStart(event.id, event.name, event.toolKind, event.parentId));
         return out;
       }
       case "tool-input-delta": {
@@ -459,7 +463,7 @@ var AgUiEncoder = class {
         };
         if (!call.started) {
           call.started = true;
-          out.push(this.callStart(event.id, event.name));
+          out.push(this.callStart(event.id, event.name, event.toolKind, event.parentId));
         }
         if (!call.sawDelta) {
           out.push({
@@ -665,13 +669,29 @@ var AgUiEncoder = class {
       ...this.closeText()
     ];
   }
-  callStart(id, name) {
+  /**
+  * `metadata['agora.toolKind']` says whether the call is an `action` (it may park for approval) or
+  * a `read`, and `metadata['agora.parentId']` the call it runs under — what the native
+  * `tool-input-start` frame carries and AG-UI's event has no field for.
+  */
+  callStart(id, name, toolKind, parentId) {
+    const metadata = {
+      ...toolKind !== void 0 ? {
+        "agora.toolKind": toolKind
+      } : {},
+      ...parentId !== void 0 ? {
+        "agora.parentId": parentId
+      } : {}
+    };
     return {
       type: "TOOL_CALL_START",
       toolCallId: id,
       toolCallName: name,
       ...this.stepMessage !== void 0 ? {
         parentMessageId: this.stepMessage
+      } : {},
+      ...Object.keys(metadata).length > 0 ? {
+        metadata
       } : {}
     };
   }
@@ -829,18 +849,23 @@ function readUserTurn(messages) {
     }
   }
   if (last === void 0) return null;
-  if (typeof last.content === "string") return {
-    text: last.content,
-    media: [],
-    dropped: []
-  };
+  if (typeof last.content === "string") {
+    return {
+      text: last.content,
+      media: [],
+      staged: [],
+      dropped: []
+    };
+  }
   if (!Array.isArray(last.content)) return {
     text: "",
     media: [],
+    staged: [],
     dropped: []
   };
   const texts = [];
   const media = [];
+  const staged = [];
   const dropped = [];
   last.content.forEach((part, index) => {
     if (!isRecord(part)) return;
@@ -852,6 +877,12 @@ function readUserTurn(messages) {
       return;
     }
     const source = part.source;
+    if (isRecord(source) && source.type === "file" && source.provider === AG_UI_MEDIA_PROVIDER && typeof source.value === "string" && source.value.length > 0) {
+      staged.push({
+        mediaId: source.value
+      });
+      return;
+    }
     if (!isRecord(source) || source.type !== "data") {
       const how = isRecord(source) && typeof source.type === "string" ? source.type : "unknown";
       dropped.push(`The ${part.type} part ${index + 1} was not used: only inline data is accepted, not a ${how} source.`);
@@ -877,6 +908,7 @@ function readUserTurn(messages) {
   return {
     text: texts.join("\n"),
     media,
+    staged,
     dropped
   };
 }
@@ -911,6 +943,9 @@ function readForwardedProps(forwarded) {
     } : {},
     ...isRecord(forwarded.pageContext) ? {
       pageContext: forwarded.pageContext
+    } : {},
+    ...forwarded.regenerate === true ? {
+      regenerate: true
     } : {}
   };
 }
@@ -1010,6 +1045,11 @@ function decodeStreamEvent(line) {
 __name(decodeStreamEvent, "decodeStreamEvent");
 
 // src/ag-ui/stream.ts
+var frameSeqs = /* @__PURE__ */ new WeakMap();
+function frameSeq(frame) {
+  return frameSeqs.get(frame);
+}
+__name(frameSeq, "frameSeq");
 var DEFAULT_QUIET_MS = 750;
 var GRACE_MS = 50;
 async function* agUiEvents(frames, options) {
@@ -1036,6 +1076,9 @@ async function* agUiEvents(frames, options) {
         ended = true;
         break;
       }
+      if (options.cursor !== void 0) {
+        options.cursor.seq = frameSeq(next.value) ?? encoder2.consumed + 1;
+      }
       yield* encoder2.encode(next.value);
     }
   } finally {
@@ -1057,8 +1100,10 @@ function raceQuiet(upcoming, ms) {
   });
 }
 __name(raceQuiet, "raceQuiet");
-function agUiSse(event) {
-  return `data: ${JSON.stringify(event)}
+function agUiSse(event, id) {
+  const head = id !== void 0 && id > 0 ? `id: ${id}
+` : "";
+  return `${head}data: ${JSON.stringify(event)}
 
 `;
 }
@@ -1066,7 +1111,14 @@ __name(agUiSse, "agUiSse");
 async function* agUiFramesFromNdjson(chunks) {
   const decoder = new TextDecoder();
   let buffer = "";
+  let seq = 0;
   const read = /* @__PURE__ */ __name((line) => {
+    seq += 1;
+    const frames = parse(line);
+    for (const frame of frames) frameSeqs.set(frame, seq);
+    return frames;
+  }, "read");
+  const parse = /* @__PURE__ */ __name((line) => {
     const event = decodeStreamEvent(line);
     if (event !== null) return [
       event
@@ -1086,7 +1138,7 @@ async function* agUiFramesFromNdjson(chunks) {
         text: line
       }
     ];
-  }, "read");
+  }, "parse");
   try {
     for await (const chunk of chunks) {
       buffer += decoder.decode(chunk, {
@@ -1159,6 +1211,7 @@ __name(actionProposalDecisionEvents, "actionProposalDecisionEvents");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AG_UI_CUSTOM,
+  AG_UI_MEDIA_PROVIDER,
   AG_UI_PROTOCOL_VERSION,
   AgUiEncoder,
   actionProposalDecisionEvents,
@@ -1167,6 +1220,7 @@ __name(actionProposalDecisionEvents, "actionProposalDecisionEvents");
   agUiSse,
   decodeInterruptId,
   encodeInterruptId,
+  frameSeq,
   parseRunInput,
   planResume,
   readAnswersPayload,