@posthog/ai 8.7.1 → 8.8.0

package/dist/gemini/index.mjs

@@ -8,6 +8,33 @@ const isString = value => {
   return typeof value === 'string';
 };
 
+/** @internal */
+
+/** @internal */
+
+/** @internal */
+function isFullAiCaptureEnabled(client) {
+  return client?.enableFullAiCapture === true;
+}
+
+/** @internal */
+function captureAiEvent(client, event) {
+  if (isFullAiCaptureEnabled(client) && typeof client.captureAi === 'function') {
+    client.captureAi(event);
+    return;
+  }
+  client.capture(event);
+}
+
+/** @internal */
+async function captureAiEventImmediate(client, event) {
+  if (isFullAiCaptureEnabled(client) && typeof client.captureAiImmediate === 'function') {
+    await client.captureAiImmediate(event);
+    return;
+  }
+  await client.captureImmediate(event);
+}
+
 const DATA_URL_PREFIX_RE = /^data:([^;,\s]+)(?:;[^;,\s]+)*;base64,/i;
 const BASE64_ALPHABET_RE = /^[A-Za-z0-9+/_=-]+$/;
 class Base64Recognizer {
@@ -113,7 +140,6 @@ class BinaryContentRedactor {
     this.recognizer = recognizer;
   }
   redact(value, mediaType) {
-    if (this.isMultimodalEnabled()) return value;
     this.visited = new WeakSet();
     return this.walk(value, mediaType ? new MediaTypeContext(undefined, undefined, mediaType) : MediaTypeContext.EMPTY);
   }
@@ -157,17 +183,14 @@ class BinaryContentRedactor {
     if (mediaType === 'application/octet-stream') return '[base64 file redacted]';
     return `[base64 ${mediaType} redacted]`;
   }
-  isMultimodalEnabled() {
-    const val = process.env._INTERNAL_LLMA_MULTIMODAL || '';
-    return val.toLowerCase() === 'true' || val === '1' || val.toLowerCase() === 'yes';
-  }
 }
 
 const redactor = new BinaryContentRedactor();
 function redactBase64DataUrl(str, mediaType) {
   return redactor.redact(str, mediaType);
 }
-const sanitizeGemini = data => redactor.redact(data);
+const sanitize = (data, client) => isFullAiCaptureEnabled(client) ? data : redactor.redact(data);
+const sanitizeGemini = (data, client) => sanitize(data, client);
 
 const TOKEN_PROPERTY_KEYS = new Set(['$ai_input_tokens', '$ai_output_tokens', '$ai_cache_read_input_tokens', '$ai_cache_creation_input_tokens', '$ai_total_tokens', '$ai_reasoning_tokens']);
 
@@ -253,7 +276,7 @@ const buildInlineDataBlock = (mimeType, data) => {
     }
   };
 };
-const formatResponseGemini = response => {
+const formatResponseGemini = (response, client) => {
   const output = [];
   if (response.candidates && Array.isArray(response.candidates)) {
     for (const candidate of response.candidates) {
@@ -292,7 +315,7 @@ const formatResponseGemini = response => {
             }
 
             // Sanitize base64 data for images and other large inline data
-            data = redactBase64DataUrl(data, mimeType);
+            data = isFullAiCaptureEnabled(client) ? data : redactBase64DataUrl(data, mimeType);
             content.push(buildInlineDataBlock(mimeType, data));
           }
         }
@@ -395,7 +418,7 @@ function addDefaults(params) {
   };
 }
 
-var version = "8.7.1";
+var version = "8.8.0";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -644,9 +667,9 @@ const captureAiGeneration = async (client, options) => {
       groups: options.groups
     };
     if (options.captureImmediate) {
-      await client.captureImmediate(event);
+      await captureAiEventImmediate(client, event);
     } else {
-      client.capture(event);
+      captureAiEvent(client, event);
     }
   } catch (error) {
     // Telemetry failures must never affect the instrumented provider call.
@@ -687,7 +710,7 @@ class WrappedModels {
         model: geminiParams.model,
         provider: 'gemini',
         input: this.formatInputForPostHog(geminiParams),
-        output: formatResponseGemini(response),
+        output: formatResponseGemini(response, this.phClient),
         latency,
         baseURL: 'https://generativelanguage.googleapis.com',
         modelParameters: getModelParams(params),
@@ -1054,7 +1077,7 @@ class WrappedModels {
     return null;
   }
   formatInputForPostHog(params) {
-    const sanitized = sanitizeGemini(params.contents);
+    const sanitized = sanitizeGemini(params.contents, this.phClient);
     const messages = this.formatInput(sanitized);
     const systemInstruction = this.extractSystemInstruction(params);
     if (systemInstruction) {