@posthog/ai 8.7.0 → 8.8.0

package/dist/langchain/index.mjs

@@ -7,6 +7,24 @@ const isObject = value => {
   return value !== null && typeof value === 'object' && !Array.isArray(value);
 };
 
+/** @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);
+}
+
 const DATA_URL_PREFIX_RE = /^data:([^;,\s]+)(?:;[^;,\s]+)*;base64,/i;
 const BASE64_ALPHABET_RE = /^[A-Za-z0-9+/_=-]+$/;
 class Base64Recognizer {
@@ -112,7 +130,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);
   }
@@ -156,14 +173,11 @@ 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();
-const sanitizeLangChain = data => redactor.redact(data);
+const sanitize = (data, client) => isFullAiCaptureEnabled(client) ? data : redactor.redact(data);
+const sanitizeLangChain = (data, client) => sanitize(data, client);
 
 const STRING_FORMAT = 'utf8';
 
@@ -231,7 +245,7 @@ function sanitizeValues(obj) {
   return jsonSafe;
 }
 
-var version = "8.7.0";
+var version = "8.8.0";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -502,7 +516,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     }) || 'generation';
     const generation = {
       name: runNameFound,
-      input: sanitizeLangChain(messages),
+      input: sanitizeLangChain(messages, this.client),
       startTime: Date.now()
     };
     if (extraParams) {
@@ -547,7 +561,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
   }
   _safeCapture(message) {
     try {
-      this.client.capture(message);
+      captureAiEvent(this.client, message);
     } catch {
       // Telemetry delivery must never affect the LangChain callback lifecycle.
     }
@@ -573,7 +587,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       $ai_lib: 'posthog-ai',
       $ai_lib_version: version,
       $ai_trace_id: traceId,
-      $ai_input_state: withPrivacyMode(this.client, this.privacyMode, sanitizeLangChain(run.input)),
+      $ai_input_state: withPrivacyMode(this.client, this.privacyMode, sanitizeLangChain(run.input, this.client)),
       $ai_latency: latency,
       $ai_span_name: run.name,
       $ai_span_id: runId,
@@ -595,14 +609,14 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
         if (interrupts !== undefined) {
           eventProperties['$ai_output_state'] = withPrivacyMode(this.client, this.privacyMode, sanitizeLangChain({
             __interrupt__: interrupts
-          }));
+          }, this.client));
         }
       } else {
         eventProperties['$ai_error'] = stringifyError(outputs);
         eventProperties['$ai_is_error'] = true;
       }
     } else if (outputs !== undefined) {
-      eventProperties['$ai_output_state'] = withPrivacyMode(this.client, this.privacyMode, sanitizeLangChain(outputs));
+      eventProperties['$ai_output_state'] = withPrivacyMode(this.client, this.privacyMode, sanitizeLangChain(outputs, this.client));
     }
     this._safeCapture({
       distinctId: this.distinctId ? this.distinctId.toString() : runId,
@@ -818,7 +832,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     }
 
     // Sanitize the message content to redact base64 images
-    return sanitizeLangChain(messageDict);
+    return sanitizeLangChain(messageDict, this.client);
   }
   _extractStopReason(output) {
     if (!output.generations || !Array.isArray(output.generations)) {