@probelabs/probe 0.6.0-rc227 → 0.6.0-rc228

package/build/agent/index.js

@@ -68681,8 +68681,6 @@ async function truncateIfNeeded(content, tokenCounter, sessionId, maxTokens) {
   if (tokenCount <= limit) {
     return { truncated: false, content };
   }
-  const maxChars = limit * CHARS_PER_TOKEN2;
-  const truncatedContent = content.substring(0, maxChars);
   let tempFilePath = null;
   let fileError = null;
   try {
@@ -68694,22 +68692,38 @@ async function truncateIfNeeded(content, tokenCounter, sessionId, maxTokens) {
     fileError = err.message || "Unknown file system error";
     tempFilePath = null;
   }
+  let truncatedBody;
+  const useTail = limit >= MIN_LIMIT_FOR_TAIL;
+  if (useTail) {
+    const headTokens = limit - TAIL_TOKENS;
+    const headChars = headTokens * CHARS_PER_TOKEN2;
+    const tailChars = TAIL_TOKENS * CHARS_PER_TOKEN2;
+    const headContent = content.substring(0, headChars);
+    const tailContent = content.substring(content.length - tailChars);
+    const omittedTokens = tokenCount - headTokens - TAIL_TOKENS;
+    truncatedBody = `${headContent}
+
+... ${omittedTokens} tokens omitted ...
+
+${tailContent}`;
+  } else {
+    const maxChars = limit * CHARS_PER_TOKEN2;
+    truncatedBody = content.substring(0, maxChars);
+  }
   let message;
   if (tempFilePath) {
     message = `Output exceeded maximum size (${tokenCount} tokens, limit: ${limit}).
 Full output saved to: ${tempFilePath}
 
---- Truncated Output (first ${limit} tokens approx) ---
-${truncatedContent}
-...
+--- Truncated Output ---
+${truncatedBody}
 --- End of Truncated Output ---`;
   } else {
     message = `Output exceeded maximum size (${tokenCount} tokens, limit: ${limit}).
 Warning: Could not save full output to file (${fileError}).
 
---- Truncated Output (first ${limit} tokens approx) ---
-${truncatedContent}
-...
+--- Truncated Output ---
+${truncatedBody}
 --- End of Truncated Output ---`;
   }
   return {
@@ -68720,12 +68734,14 @@ ${truncatedContent}
     error: fileError || void 0
   };
 }
-var DEFAULT_MAX_OUTPUT_TOKENS, CHARS_PER_TOKEN2;
+var DEFAULT_MAX_OUTPUT_TOKENS, CHARS_PER_TOKEN2, TAIL_TOKENS, MIN_LIMIT_FOR_TAIL;
 var init_outputTruncator = __esm({
   "src/agent/outputTruncator.js"() {
     "use strict";
     DEFAULT_MAX_OUTPUT_TOKENS = 2e4;
     CHARS_PER_TOKEN2 = 4;
+    TAIL_TOKENS = 1e3;
+    MIN_LIMIT_FOR_TAIL = 2e3;
   }
 });
 

package/build/agent/outputTruncator.js

@@ -5,6 +5,8 @@ import { randomUUID } from 'crypto';
 
 const DEFAULT_MAX_OUTPUT_TOKENS = 20000;
 const CHARS_PER_TOKEN = 4; // Conservative approximation
+const TAIL_TOKENS = 1000; // Number of tokens to show from the end of truncated output
+const MIN_LIMIT_FOR_TAIL = 2000; // Minimum token limit to use head+tail split
 
 /**
  * Validate and normalize a token limit value.
@@ -61,10 +63,6 @@ export async function truncateIfNeeded(content, tokenCounter, sessionId, maxToke
     return { truncated: false, content };
   }
 
-  // Truncate to approximately maxTokens worth of characters
-  const maxChars = limit * CHARS_PER_TOKEN;
-  const truncatedContent = content.substring(0, maxChars);
-
   // Try to write full output to temp file
   let tempFilePath = null;
   let fileError = null;
@@ -79,22 +77,37 @@ export async function truncateIfNeeded(content, tokenCounter, sessionId, maxToke
     tempFilePath = null;
   }
 
+  // Build truncated content with head + tail for better context
+  let truncatedBody;
+  const useTail = limit >= MIN_LIMIT_FOR_TAIL;
+
+  if (useTail) {
+    const headTokens = limit - TAIL_TOKENS;
+    const headChars = headTokens * CHARS_PER_TOKEN;
+    const tailChars = TAIL_TOKENS * CHARS_PER_TOKEN;
+    const headContent = content.substring(0, headChars);
+    const tailContent = content.substring(content.length - tailChars);
+    const omittedTokens = tokenCount - headTokens - TAIL_TOKENS;
+    truncatedBody = `${headContent}\n\n... ${omittedTokens} tokens omitted ...\n\n${tailContent}`;
+  } else {
+    const maxChars = limit * CHARS_PER_TOKEN;
+    truncatedBody = content.substring(0, maxChars);
+  }
+
   let message;
   if (tempFilePath) {
     message = `Output exceeded maximum size (${tokenCount} tokens, limit: ${limit}).
 Full output saved to: ${tempFilePath}
 
---- Truncated Output (first ${limit} tokens approx) ---
-${truncatedContent}
-...
+--- Truncated Output ---
+${truncatedBody}
 --- End of Truncated Output ---`;
   } else {
     message = `Output exceeded maximum size (${tokenCount} tokens, limit: ${limit}).
 Warning: Could not save full output to file (${fileError}).
 
---- Truncated Output (first ${limit} tokens approx) ---
-${truncatedContent}
-...
+--- Truncated Output ---
+${truncatedBody}
 --- End of Truncated Output ---`;
   }