@dudousxd/nestjs-agent-core 0.17.0 → 0.18.0

package/dist/index.js

@@ -58,6 +58,105 @@ function decodeStreamEvent(line) {
 }
 __name(decodeStreamEvent, "decodeStreamEvent");
 
+// src/turn-frames.ts
+function observeTurnFrames(writer, now = Date.now) {
+  const decoder2 = new TextDecoder();
+  let pending = "";
+  let reasoning = "";
+  let reasoningMs = 0;
+  let burstStartedAt;
+  const ui = /* @__PURE__ */ new Map();
+  function closeBurst() {
+    if (burstStartedAt !== void 0) {
+      reasoningMs += Math.max(0, now() - burstStartedAt);
+      burstStartedAt = void 0;
+    }
+  }
+  __name(closeBurst, "closeBurst");
+  function observeLine(line) {
+    if (line.trim().length === 0) {
+      return;
+    }
+    const event = decodeStreamEvent(line);
+    if (event === null) {
+      return;
+    }
+    if (event.kind === "reasoning") {
+      burstStartedAt ??= now();
+      reasoning += event.text;
+      return;
+    }
+    closeBurst();
+    if (event.kind === "ui") {
+      ui.set(event.id, {
+        id: event.id,
+        component: event.component,
+        props: event.props,
+        ...event.version !== void 0 ? {
+          version: event.version
+        } : {}
+      });
+    }
+  }
+  __name(observeLine, "observeLine");
+  function observe(chunk) {
+    pending += decoder2.decode(chunk, {
+      stream: true
+    });
+    let newline = pending.indexOf("\n");
+    while (newline !== -1) {
+      observeLine(pending.slice(0, newline));
+      pending = pending.slice(newline + 1);
+      newline = pending.indexOf("\n");
+    }
+  }
+  __name(observe, "observe");
+  return {
+    writer: {
+      write(chunk) {
+        observe(chunk);
+        return writer.write(chunk);
+      },
+      end: /* @__PURE__ */ __name(() => writer.end(), "end"),
+      fail: /* @__PURE__ */ __name((error) => writer.fail(error), "fail")
+    },
+    summary() {
+      if (pending.length > 0) {
+        observeLine(pending);
+        pending = "";
+      }
+      closeBurst();
+      return {
+        ...reasoning.length > 0 ? {
+          reasoning,
+          reasoningMs: Math.round(reasoningMs)
+        } : {},
+        ...ui.size > 0 ? {
+          ui: [
+            ...ui.values()
+          ]
+        } : {}
+      };
+    }
+  };
+}
+__name(observeTurnFrames, "observeTurnFrames");
+function withTurnFrames(result, frames) {
+  return {
+    ...result,
+    ...result.reasoning === void 0 && frames.reasoning !== void 0 ? {
+      reasoning: frames.reasoning
+    } : {},
+    ...result.reasoningMs === void 0 && frames.reasoningMs !== void 0 ? {
+      reasoningMs: frames.reasoningMs
+    } : {},
+    ...result.ui === void 0 && frames.ui !== void 0 ? {
+      ui: frames.ui
+    } : {}
+  };
+}
+__name(withTurnFrames, "withTurnFrames");
+
 // src/spi/pricing-store.ts
 async function seedModelPrices(store, prices) {
   for (const price of prices) {
@@ -3363,12 +3462,13 @@ ${block}`;
           writer
         }) : void 0;
         const buffer = gateMode === "whole" ? createFrameBuffer() : void 0;
-        const result = stampToolKinds(await traceLlmTurn(hooks.runId, i, () => deps.model.runTurn({
+        const frames = observeTurnFrames(incremental?.writer ?? buffer?.writer ?? writer);
+        const result = stampToolKinds(withTurnFrames(await traceLlmTurn(hooks.runId, i, () => deps.model.runTurn({
           system: prompt.system,
           messages: prompt.messages,
           tools,
-          sink: incremental?.writer ?? buffer?.writer ?? writer
-        })), deps);
+          sink: frames.writer
+        })), frames.summary()), deps);
         if (incremental !== void 0) {
           await incremental.settled();
           const refusal = incremental.rejection();
@@ -3556,6 +3656,15 @@ ${block}`;
       ...input.agentName !== void 0 ? {
         agentName: input.agentName
       } : {},
+      ...turn.reasoning !== void 0 ? {
+        reasoning: turn.reasoning
+      } : {},
+      ...turn.reasoningMs !== void 0 ? {
+        reasoningMs: turn.reasoningMs
+      } : {},
+      ...turn.ui !== void 0 && turn.ui.length > 0 ? {
+        ui: turn.ui
+      } : {},
       ...messageCalls.length > 0 ? {
         toolCalls: messageCalls
       } : {},
@@ -3613,7 +3722,10 @@ ${block}`;
         await writer.write(encodeStreamEvent({
           kind: "step-finish",
           usage: turn.usage,
-          costUsd
+          costUsd,
+          ...turn.reasoningMs !== void 0 ? {
+            reasoningMs: turn.reasoningMs
+          } : {}
         }));
       });
       break;
@@ -3666,7 +3778,10 @@ ${block}`;
       await writer.write(encodeStreamEvent({
         kind: "step-finish",
         usage: turn.usage,
-        costUsd
+        costUsd,
+        ...turn.reasoningMs !== void 0 ? {
+          reasoningMs: turn.reasoningMs
+        } : {}
       }));
     });
   }
@@ -3820,6 +3935,7 @@ export {
   memoryWriteVerdict,
   normalizeDelegation,
   normalizeElicitationReply,
+  observeTurnFrames,
   offerMemories,
   offerSkills,
   publishAgentDelegated,
@@ -3869,6 +3985,7 @@ export {
   withMemoryTool,
   withSkillTool,
   withToolTimeout,
+  withTurnFrames,
   writeMemory
 };
 //# sourceMappingURL=index.js.map