@posthog/ai 6.3.0 → 6.3.1

package/dist/index.mjs

@@ -6,7 +6,7 @@ import { wrapLanguageModel } from 'ai';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 import { GoogleGenAI } from '@google/genai';
 
-var version = "6.3.0";
+var version = "6.3.1";
 
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
@@ -206,18 +206,43 @@ const mergeSystemPrompt = (params, provider) => {
 const withPrivacyMode = (client, privacyMode, input) => {
   return client.privacy_mode || privacyMode ? null : input;
 };
-const truncate = str => {
+function toSafeString(input) {
+  if (input === undefined || input === null) {
+    return '';
+  }
+  if (typeof input === 'string') {
+    return input;
+  }
   try {
-    const buffer = Buffer.from(str, STRING_FORMAT);
-    if (buffer.length <= MAX_OUTPUT_SIZE) {
-      return str;
-    }
-    const truncatedBuffer = buffer.slice(0, MAX_OUTPUT_SIZE);
-    return `${truncatedBuffer.toString(STRING_FORMAT)}... [truncated]`;
+    return JSON.stringify(input);
   } catch {
-    console.error('Error truncating, likely not a string');
-    return str;
+    console.warn('Failed to stringify input', input);
+    return '';
+  }
+}
+const truncate = input => {
+  const str = toSafeString(input);
+  if (str === '') {
+    return '';
+  }
+  // Check if we need to truncate and ensure STRING_FORMAT is respected
+  const encoder = new TextEncoder();
+  const buffer = encoder.encode(str);
+  if (buffer.length <= MAX_OUTPUT_SIZE) {
+    // Ensure STRING_FORMAT is respected
+    return new TextDecoder(STRING_FORMAT).decode(buffer);
+  }
+  // Truncate the buffer and ensure a valid string is returned
+  const truncatedBuffer = buffer.slice(0, MAX_OUTPUT_SIZE);
+  // fatal: false means we get U+FFFD at the end if truncation broke the encoding
+  const decoder = new TextDecoder(STRING_FORMAT, {
+    fatal: false
+  });
+  let truncatedStr = decoder.decode(truncatedBuffer);
+  if (truncatedStr.endsWith('\uFFFD')) {
+    truncatedStr = truncatedStr.slice(0, -1);
   }
+  return `${truncatedStr}... [truncated]`;
 };
 /**
  * Extract available tool calls from the request parameters.