@posthog/ai 7.5.3 → 7.6.0

package/dist/vercel/index.mjs

@@ -2,7 +2,7 @@ import { v4 } from 'uuid';
 import { Buffer } from 'buffer';
 import { uuidv7 } from '@posthog/core';
 
-var version = "7.5.3";
+var version = "7.6.0";
 
 // Type guards for safer type checking
 
@@ -10,6 +10,59 @@ const isString = value => {
   return typeof value === 'string';
 };
 
+const REDACTED_IMAGE_PLACEHOLDER = '[base64 image redacted]';
+
+// ============================================
+// Multimodal Feature Toggle
+// ============================================
+
+const isMultimodalEnabled = () => {
+  const val = process.env._INTERNAL_LLMA_MULTIMODAL || '';
+  return val.toLowerCase() === 'true' || val === '1' || val.toLowerCase() === 'yes';
+};
+
+// ============================================
+// Base64 Detection Helpers
+// ============================================
+
+const isBase64DataUrl = str => {
+  return /^data:([^;]+);base64,/.test(str);
+};
+const isValidUrl = str => {
+  try {
+    new URL(str);
+    return true;
+  } catch {
+    // Not an absolute URL, check if it's a relative URL or path
+    return str.startsWith('/') || str.startsWith('./') || str.startsWith('../');
+  }
+};
+const isRawBase64 = str => {
+  // Skip if it's a valid URL or path
+  if (isValidUrl(str)) {
+    return false;
+  }
+
+  // Check if it's a valid base64 string
+  // Base64 images are typically at least a few hundred chars, but we'll be conservative
+  return str.length > 20 && /^[A-Za-z0-9+/]+=*$/.test(str);
+};
+function redactBase64DataUrl(str) {
+  if (isMultimodalEnabled()) return str;
+  if (!isString(str)) return str;
+
+  // Check for data URL format
+  if (isBase64DataUrl(str)) {
+    return REDACTED_IMAGE_PLACEHOLDER;
+  }
+
+  // Check for raw base64 (Vercel sends raw base64 for inline images)
+  if (isRawBase64(str)) {
+    return REDACTED_IMAGE_PLACEHOLDER;
+  }
+  return str;
+}
+
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
 const STRING_FORMAT = 'utf8';
@@ -314,6 +367,9 @@ const sendEventToPosthog = async ({
     } : {}),
     ...(usage.webSearchCount ? {
       $ai_web_search_count: usage.webSearchCount
+    } : {}),
+    ...(usage.rawUsage ? {
+      $ai_usage: usage.rawUsage
     } : {})
   };
   const properties = {
@@ -358,59 +414,6 @@ const sendEventToPosthog = async ({
   return Promise.resolve();
 };
 
-const REDACTED_IMAGE_PLACEHOLDER = '[base64 image redacted]';
-
-// ============================================
-// Multimodal Feature Toggle
-// ============================================
-
-const isMultimodalEnabled = () => {
-  const val = process.env._INTERNAL_LLMA_MULTIMODAL || '';
-  return val.toLowerCase() === 'true' || val === '1' || val.toLowerCase() === 'yes';
-};
-
-// ============================================
-// Base64 Detection Helpers
-// ============================================
-
-const isBase64DataUrl = str => {
-  return /^data:([^;]+);base64,/.test(str);
-};
-const isValidUrl = str => {
-  try {
-    new URL(str);
-    return true;
-  } catch {
-    // Not an absolute URL, check if it's a relative URL or path
-    return str.startsWith('/') || str.startsWith('./') || str.startsWith('../');
-  }
-};
-const isRawBase64 = str => {
-  // Skip if it's a valid URL or path
-  if (isValidUrl(str)) {
-    return false;
-  }
-
-  // Check if it's a valid base64 string
-  // Base64 images are typically at least a few hundred chars, but we'll be conservative
-  return str.length > 20 && /^[A-Za-z0-9+/]+=*$/.test(str);
-};
-function redactBase64DataUrl(str) {
-  if (isMultimodalEnabled()) return str;
-  if (!isString(str)) return str;
-
-  // Check for data URL format
-  if (isBase64DataUrl(str)) {
-    return REDACTED_IMAGE_PLACEHOLDER;
-  }
-
-  // Check for raw base64 (Vercel sends raw base64 for inline images)
-  if (isRawBase64(str)) {
-    return REDACTED_IMAGE_PLACEHOLDER;
-  }
-  return str;
-}
-
 // Union types for dual version support
 
 // Type guards
@@ -738,13 +741,31 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
 
           // V2 usage has simple numbers, V3 has objects with .total - normalize both
           const usageObj = result.usage;
+
+          // Extract raw response for providers that include detailed usage metadata
+          // For Gemini, candidatesTokensDetails is in result.response.body.usageMetadata
+          const rawUsageData = {
+            usage: result.usage,
+            providerMetadata
+          };
+
+          // Include response body usageMetadata if it contains detailed token breakdown (e.g., candidatesTokensDetails)
+          if (result.response && typeof result.response === 'object') {
+            const responseBody = result.response.body;
+            if (responseBody && typeof responseBody === 'object' && 'usageMetadata' in responseBody) {
+              rawUsageData.rawResponse = {
+                usageMetadata: responseBody.usageMetadata
+              };
+            }
+          }
           const usage = {
             inputTokens: extractTokenCount(result.usage.inputTokens),
             outputTokens: extractTokenCount(result.usage.outputTokens),
             reasoningTokens: extractReasoningTokens(usageObj),
             cacheReadInputTokens: extractCacheReadTokens(usageObj),
             webSearchCount,
-            ...additionalTokenValues
+            ...additionalTokenValues,
+            rawUsage: rawUsageData
           };
           adjustAnthropicV3CacheTokens(model, provider, usage);
           await sendEventToPosthog({
@@ -904,10 +925,14 @@ const wrapVercelLanguageModel = (model, phClient, options) => {
               }] : [];
               const webSearchCount = extractWebSearchCount(providerMetadata, usage);
 
-              // Update usage with web search count
+              // Update usage with web search count and raw metadata
               const finalUsage = {
                 ...usage,
-                webSearchCount
+                webSearchCount,
+                rawUsage: {
+                  usage,
+                  providerMetadata
+                }
               };
               adjustAnthropicV3CacheTokens(model, provider, finalUsage);
               await sendEventToPosthog({