@ai-react-markdown/engine 2.9.1 → 2.10.1

package/dist/index.js

@@ -4335,24 +4335,32 @@ var SMOOTH_STREAM_PACING_PRESETS = Object.freeze({
     correctionTauMs: 280,
     emaTauMs: 2600,
     minCharsPerSecond: 4,
-    drainMs: 320
+    drainMs: 320,
+    maxLagMs: 3500
   }),
   balanced: Object.freeze({
     bufferFactor: 1,
     correctionTauMs: 180,
     emaTauMs: 1800,
     minCharsPerSecond: 6,
-    drainMs: 240
+    drainMs: 240,
+    maxLagMs: 2500
   }),
   responsive: Object.freeze({
     bufferFactor: 0.45,
     correctionTauMs: 110,
     emaTauMs: 1100,
     minCharsPerSecond: 10,
-    drainMs: 150
+    drainMs: 150,
+    maxLagMs: 800
   })
 });
-var MAX_TARGET_BUFFER_MS = 1200;
+var GAP_WINDOW = 16;
+var INTERVAL_QUANTILE = 0.9;
+var HORIZON_PAD_MS = 50;
+var DEADLINE_FLOOR_MS = 33;
+var DRAIN_CATCHUP_FACTOR = 2;
+var DRAIN_MAX_FACTOR = 3;
 var defaultNow = typeof performance !== "undefined" && typeof performance.now === "function" ? () => performance.now() : () => Date.now();
 var defaultSchedule = (cb) => {
   if (typeof requestAnimationFrame === "function") {
@@ -4403,18 +4411,18 @@ var createSmoothStreamController = (options = {}) => {
   let lastTickAt = 0;
   let cancelFrame;
   let disposed = false;
-  let rateEma;
-  let intervalEma;
+  let gapWindow = [];
   let lastArrivalAt;
   const listeners = /* @__PURE__ */ new Set();
   const resolveParams = () => {
-    const preset = SMOOTH_STREAM_PACING_PRESETS[options.pacing] ?? SMOOTH_STREAM_PACING_PRESETS.balanced;
+    const preset = Object.hasOwn(SMOOTH_STREAM_PACING_PRESETS, options.pacing) ? SMOOTH_STREAM_PACING_PRESETS[options.pacing] : SMOOTH_STREAM_PACING_PRESETS.balanced;
     return {
       bufferFactor: finiteOr(options.bufferFactor, preset.bufferFactor),
       correctionTauMs: finiteOr(options.correctionTauMs, preset.correctionTauMs),
       emaTauMs: finiteOr(options.emaTauMs, preset.emaTauMs),
       minCharsPerSecond: finiteOr(options.minCharsPerSecond, preset.minCharsPerSecond),
-      drainMs: finiteOr(options.drainMs, preset.drainMs)
+      drainMs: finiteOr(options.drainMs, preset.drainMs),
+      maxLagMs: finiteOr(options.maxLagMs, preset.maxLagMs ?? 2500)
     };
   };
   const recordArrival = (t, added) => {
@@ -4425,12 +4433,8 @@ var createSmoothStreamController = (options = {}) => {
     }
     const gap = Math.max(1, t - lastArrivalAt);
     lastArrivalAt = t;
-    const tau = Math.max(1, resolveParams().emaTauMs);
-    if (gap > tau) return;
-    const alpha = 1 - Math.exp(-gap / tau);
-    const instantRate = added * 1e3 / gap;
-    rateEma = (rateEma ?? 0) + alpha * (instantRate - (rateEma ?? 0));
-    intervalEma = (intervalEma ?? 0) + alpha * (gap - (intervalEma ?? 0));
+    gapWindow.push({ gap, added });
+    if (gapWindow.length > GAP_WINDOW) gapWindow.shift();
   };
   const notify = () => {
     visibleCache = source.slice(0, visibleEnd);
@@ -4456,15 +4460,12 @@ var createSmoothStreamController = (options = {}) => {
     let rate;
     if (finished && drainDeadlineAt !== void 0) {
       rate = Math.max(params.minCharsPerSecond, pending.length * 1e3 / Math.max(1, drainDeadlineAt - t));
-    } else if (rateEma !== void 0 && intervalEma !== void 0) {
-      const targetBuffer = Math.max(
-        1,
-        Math.min(params.bufferFactor * rateEma * (intervalEma / 1e3), rateEma * MAX_TARGET_BUFFER_MS / 1e3)
-      );
-      rate = Math.max(
-        params.minCharsPerSecond,
-        rateEma + (pending.length - targetBuffer) * 1e3 / Math.max(1, params.correctionTauMs)
-      );
+    } else if (gapWindow.length > 0 && lastArrivalAt !== void 0) {
+      const gaps = gapWindow.map((s) => s.gap).sort((a, b) => a - b);
+      const intervalQ = gaps[Math.min(gaps.length - 1, Math.floor(gaps.length * INTERVAL_QUANTILE))];
+      const horizon = Math.max(16, Math.min(params.bufferFactor * intervalQ + HORIZON_PAD_MS, params.maxLagMs));
+      const deadline = Math.max(lastArrivalAt + horizon, t + DEADLINE_FLOOR_MS);
+      rate = Math.max(params.minCharsPerSecond, pending.length * 1e3 / Math.max(1, deadline - t));
     } else {
       rate = Math.max(params.minCharsPerSecond, pending.length * 1e3 / Math.max(1, params.correctionTauMs));
     }
@@ -4510,8 +4511,7 @@ var createSmoothStreamController = (options = {}) => {
     initialized = true;
     finished = false;
     drainDeadlineAt = void 0;
-    rateEma = void 0;
-    intervalEma = void 0;
+    gapWindow = [];
     lastArrivalAt = void 0;
     source = next;
     visibleEnd = next.length;
@@ -4548,10 +4548,26 @@ var createSmoothStreamController = (options = {}) => {
       if (finished) return;
       finished = true;
       lastArrivalAt = void 0;
-      drainDeadlineAt = now() + resolveParams().drainMs;
       if (tentativeEnd > (pending.length > 0 ? pending[pending.length - 1] : visibleEnd)) {
         pending.push(tentativeEnd);
       }
+      const params = resolveParams();
+      let drainWin = params.drainMs;
+      let chars = 0;
+      let ms = 0;
+      for (let i = 0; i < gapWindow.length - 1; i += 1) {
+        chars += gapWindow[i].added;
+        ms += gapWindow[i].gap;
+      }
+      const backlog = source.length - visibleEnd;
+      if (chars > 0 && ms > 0 && backlog > 0) {
+        const throughput = chars * 1e3 / ms;
+        drainWin = Math.min(
+          Math.max(backlog * 1e3 / (DRAIN_CATCHUP_FACTOR * throughput), params.drainMs),
+          DRAIN_MAX_FACTOR * params.drainMs
+        );
+      }
+      drainDeadlineAt = now() + drainWin;
       seam = source.length;
       ensureScheduled();
     },
@@ -4883,8 +4899,20 @@ function escapeLatexPipesInUnclosed(text) {
   const tail = text.substring(unclosedStart + delimLen);
   return before + delim + replaceUnescapedPipes(tail);
 }
+function opensMathFlow(text, pos) {
+  let i = pos;
+  let spaces = 0;
+  while (i > 0 && text[i - 1] !== "\n") {
+    i -= 1;
+    if (text[i] !== " ") return false;
+    spaces += 1;
+    if (spaces > 3) return false;
+  }
+  return true;
+}
 function truncateUnclosedLatexBlock(text, unclosedStart = findUnclosedDelimiterStart(text, "double-only")) {
   if (unclosedStart === -1) return text;
+  if (!opensMathFlow(text, unclosedStart)) return text;
   return text.substring(0, unclosedStart).trimEnd();
 }
 function escapeTextUnderscores(text) {
@@ -4985,7 +5013,7 @@ function processSlice(slice, probe = true) {
       unclosedDouble = findUnclosedDelimiterStart(text, "double-only");
       if (unclosedDouble !== -1) {
         quiescent = false;
-        if (index === 0 && text.slice(0, unclosedDouble).trim() === "") {
+        if (index === 0 && opensMathFlow(text, unclosedDouble) && text.slice(0, unclosedDouble).trim() === "") {
           truncatedAtSeamStart = true;
         }
       }