@posthog/ai 7.0.0 → 7.1.1

package/dist/langchain/index.mjs

@@ -1,7 +1,7 @@
 import 'buffer';
 import * as uuid from 'uuid';
 
-var version = "7.0.0";
+var version = "7.1.1";
 
 // Type guards for safer type checking
 
@@ -42,7 +42,7 @@ const getModelParams = params => {
     return {};
   }
   const modelParams = {};
-  const paramKeys = ['temperature', 'max_tokens', 'max_completion_tokens', 'top_p', 'frequency_penalty', 'presence_penalty', 'n', 'stop', 'stream', 'streaming'];
+  const paramKeys = ['temperature', 'max_tokens', 'max_completion_tokens', 'top_p', 'frequency_penalty', 'presence_penalty', 'n', 'stop', 'stream', 'streaming', 'language', 'response_format', 'timestamp_granularities'];
   for (const key of paramKeys) {
     if (key in params && params[key] !== undefined) {
       modelParams[key] = params[key];
@@ -933,7 +933,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       eventProperties['$ai_is_error'] = true;
     } else {
       // Handle token usage
-      const [inputTokens, outputTokens, additionalTokenData] = this.parseUsage(output);
+      const [inputTokens, outputTokens, additionalTokenData] = this.parseUsage(output, run.provider, run.model);
       eventProperties['$ai_input_tokens'] = inputTokens;
       eventProperties['$ai_output_tokens'] = outputTokens;
 
@@ -1083,7 +1083,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     // Sanitize the message content to redact base64 images
     return sanitizeLangChain(messageDict);
   }
-  _parseUsageModel(usage) {
+  _parseUsageModel(usage, provider, model) {
     const conversionList = [['promptTokens', 'input'], ['completionTokens', 'output'], ['input_tokens', 'input'], ['output_tokens', 'output'], ['prompt_token_count', 'input'], ['candidates_token_count', 'output'], ['inputTokenCount', 'input'], ['outputTokenCount', 'output'], ['input_token_count', 'input'], ['generated_token_count', 'output']];
     const parsedUsage = conversionList.reduce((acc, [modelKey, typeKey]) => {
       const value = usage[modelKey];
@@ -1156,20 +1156,28 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       additionalTokenData.webSearchCount = webSearchCount;
     }
 
-    // In LangChain, input_tokens is the sum of input and cache read tokens.
-    // Our cost calculation expects them to be separate, for Anthropic.
-    if (parsedUsage.input && additionalTokenData.cacheReadInputTokens) {
+    // For Anthropic providers, LangChain reports input_tokens as the sum of input and cache read tokens.
+    // Our cost calculation expects them to be separate for Anthropic, so we subtract cache tokens.
+    // For other providers (OpenAI, etc.), input_tokens already excludes cache tokens as expected.
+    // Match logic consistent with plugin-server: exact match on provider OR substring match on model
+    let isAnthropic = false;
+    if (provider && provider.toLowerCase() === 'anthropic') {
+      isAnthropic = true;
+    } else if (model && model.toLowerCase().includes('anthropic')) {
+      isAnthropic = true;
+    }
+    if (isAnthropic && parsedUsage.input && additionalTokenData.cacheReadInputTokens) {
       parsedUsage.input = Math.max(parsedUsage.input - additionalTokenData.cacheReadInputTokens, 0);
     }
     return [parsedUsage.input, parsedUsage.output, additionalTokenData];
   }
-  parseUsage(response) {
+  parseUsage(response, provider, model) {
     let llmUsage = [0, 0, {}];
     const llmUsageKeys = ['token_usage', 'usage', 'tokenUsage'];
     if (response.llmOutput != null) {
       const key = llmUsageKeys.find(k => response.llmOutput?.[k] != null);
       if (key) {
-        llmUsage = this._parseUsageModel(response.llmOutput[key]);
+        llmUsage = this._parseUsageModel(response.llmOutput[key], provider, model);
       }
     }
 
@@ -1179,14 +1187,14 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
         for (const genChunk of generation) {
           // Check other paths for usage information
           if (genChunk.generationInfo?.usage_metadata) {
-            llmUsage = this._parseUsageModel(genChunk.generationInfo.usage_metadata);
+            llmUsage = this._parseUsageModel(genChunk.generationInfo.usage_metadata, provider, model);
             return llmUsage;
           }
           const messageChunk = genChunk.generationInfo ?? {};
           const responseMetadata = messageChunk.response_metadata ?? {};
           const chunkUsage = responseMetadata['usage'] ?? responseMetadata['amazon-bedrock-invocationMetrics'] ?? messageChunk.usage_metadata;
           if (chunkUsage) {
-            llmUsage = this._parseUsageModel(chunkUsage);
+            llmUsage = this._parseUsageModel(chunkUsage, provider, model);
             return llmUsage;
           }
         }