min-agent 0.2.1 → 0.3.0

package/dist/assistant-stream.js

@@ -11,6 +11,12 @@ const THINKING_BLOCKS = [
 ];
 /** Max tail to keep when looking for a partial opening tag. */
 const PARTIAL_TAG_HOLD = 72;
+/**
+ * If an opening tag never gets closed (truncated / misbehaving model), the
+ * buffer would grow without bound. Beyond this size, treat the remainder as
+ * thinking text and move on.
+ */
+const UNCLOSED_THINK_MAX = 16 * 1024;
 function lower(s) {
     return s.toLowerCase();
 }
@@ -71,10 +77,6 @@ export class ThinkingBodySplitter {
                 this.buf = this.buf.slice(open.index);
             }
             const low = lower(this.buf);
-            if (!low.startsWith(open.tag.open)) {
-                this.buf = this.buf.slice(1);
-                continue;
-            }
             const afterOpen = open.tag.open.length;
             const closeRel = low.indexOf(open.tag.close, afterOpen);
             if (closeRel < 0) {
@@ -82,6 +84,11 @@ export class ThinkingBodySplitter {
                     thinking += this.buf.slice(afterOpen);
                     this.buf = "";
                 }
+                else if (this.buf.length - afterOpen > UNCLOSED_THINK_MAX) {
+                    thinking += this.buf.slice(afterOpen);
+                    this.buf = "";
+                    hadThinking = true;
+                }
                 break;
             }
             const inner = this.buf.slice(afterOpen, closeRel);
@@ -90,9 +97,6 @@ export class ThinkingBodySplitter {
             this.buf = this.buf.slice(closeRel + open.tag.close.length);
         }
         // Strip leading newlines from display that follow a thinking block
-        if (hadThinking && display.length === 0 && this.buf.startsWith("\n")) {
-            // Will be handled on next feed
-        }
         if (hadThinking) {
             display = display.replace(/^\n+/, "");
         }