@posthog/ai 7.13.2 → 7.14.0

package/dist/langchain/index.d.ts

@@ -63,6 +63,7 @@ declare class LangChainCallbackHandler extends BaseCallbackHandler {
     private _convertLcToolCallsToOai;
     private _extractRawResponse;
     private _convertMessageToDict;
+    private _extractStopReason;
     private _parseUsageModel;
     private parseUsage;
 }

package/dist/langchain/index.mjs

@@ -1,7 +1,7 @@
 import * as uuid from 'uuid';
 import '@posthog/core';
 
-var version = "7.13.2";
+var version = "7.14.0";
 
 // Type guards for safer type checking
 
@@ -965,6 +965,12 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
         eventProperties['$ai_web_search_count'] = additionalTokenData.webSearchCount;
       }
 
+      // Extract stop reason from generation info
+      const stopReason = this._extractStopReason(output);
+      if (stopReason) {
+        eventProperties['$ai_stop_reason'] = stopReason;
+      }
+
       // Handle generations/completions
       let completions;
       if (output.generations && Array.isArray(output.generations)) {
@@ -1100,6 +1106,37 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     // Sanitize the message content to redact base64 images
     return sanitizeLangChain(messageDict);
   }
+  _extractStopReason(output) {
+    if (!output.generations || !Array.isArray(output.generations)) {
+      return undefined;
+    }
+    const lastGeneration = output.generations[output.generations.length - 1];
+    if (!Array.isArray(lastGeneration) || lastGeneration.length === 0) {
+      return undefined;
+    }
+    const gen = lastGeneration[0];
+
+    // Check generationInfo for finish_reason (OpenAI format)
+    if (gen.generationInfo?.finish_reason) {
+      return String(gen.generationInfo.finish_reason);
+    }
+
+    // Check generationInfo for response_metadata.stop_reason (Anthropic format)
+    if (gen.generationInfo?.response_metadata?.stop_reason) {
+      return String(gen.generationInfo.response_metadata.stop_reason);
+    }
+
+    // Check message response_metadata for finish_reason (common LangChain format)
+    if (gen.generationInfo?.response_metadata?.finish_reason) {
+      return String(gen.generationInfo.response_metadata.finish_reason);
+    }
+
+    // Check for stop_reason directly in generationInfo
+    if (gen.generationInfo?.stop_reason) {
+      return String(gen.generationInfo.stop_reason);
+    }
+    return undefined;
+  }
   _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]) => {