@posthog/ai 7.9.2 → 7.9.3

package/dist/index.mjs

@@ -5,7 +5,7 @@ import { uuidv7 } from '@posthog/core';
 import AnthropicOriginal from '@anthropic-ai/sdk';
 import { GoogleGenAI } from '@google/genai';
 
-var version = "7.9.2";
+var version = "7.9.3";
 
 // Type guards for safer type checking
 const isString = value => {
@@ -224,6 +224,13 @@ const sanitizeLangChain = data => {
   return processMessages(data, sanitizeLangChainImage);
 };
 
+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']);
+function getTokensSource(posthogProperties) {
+  if (posthogProperties && Object.keys(posthogProperties).some(key => TOKEN_PROPERTY_KEYS.has(key))) {
+    return 'passthrough';
+  }
+  return 'sdk';
+}
 // limit large outputs by truncating to 200kb (approx 200k bytes)
 const MAX_OUTPUT_SIZE = 200000;
 const STRING_FORMAT = 'utf8';
@@ -803,6 +810,7 @@ const sendEventToPosthog = async ({
     $ai_trace_id: traceId,
     $ai_base_url: baseURL,
     ...params.posthogProperties,
+    $ai_tokens_source: getTokensSource(params.posthogProperties),
     ...(distinctId ? {} : {
       $process_person_profile: false
     }),