@posthog/ai 8.6.1 → 8.6.3

package/dist/langchain/index.mjs

@@ -215,7 +215,7 @@ function sanitizeValues(obj) {
   return jsonSafe;
 }
 
-var version = "8.6.1";
+var version = "8.6.3";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -529,6 +529,13 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     }
     return parentRunId;
   }
+  _safeCapture(message) {
+    try {
+      this.client.capture(message);
+    } catch {
+      // Telemetry delivery must never affect the LangChain callback lifecycle.
+    }
+  }
   _popRunAndCaptureTraceOrSpan(runId, parentRunId, outputs) {
     const traceId = this._getTraceId(runId);
     this._popParentOfRun(runId);
@@ -581,7 +588,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     } else if (outputs !== undefined) {
       eventProperties['$ai_output_state'] = withPrivacyMode(this.client, this.privacyMode, sanitizeLangChain(outputs));
     }
-    this.client.capture({
+    this._safeCapture({
       distinctId: this.distinctId ? this.distinctId.toString() : runId,
       event: eventName,
       properties: eventProperties,
@@ -679,7 +686,7 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     if (!this.distinctId) {
       eventProperties['$process_person_profile'] = false;
     }
-    this.client.capture({
+    this._safeCapture({
       distinctId: this.distinctId ? this.distinctId.toString() : traceId,
       event: '$ai_generation',
       properties: eventProperties,
@@ -802,27 +809,10 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       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;
+    const messageResponseMetadata = gen.message?.response_metadata;
+    const generationResponseMetadata = gen.generationInfo?.response_metadata;
+    const stopReason = messageResponseMetadata?.finish_reason || messageResponseMetadata?.stop_reason || gen.generationInfo?.finish_reason || generationResponseMetadata?.stop_reason || generationResponseMetadata?.finish_reason || gen.generationInfo?.stop_reason;
+    return stopReason != null ? String(stopReason) : 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']];
@@ -940,14 +930,11 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
     if (llmUsage[0] === 0 && llmUsage[1] === 0 && response.generations) {
       for (const generation of response.generations) {
         for (const genChunk of generation) {
-          // Check other paths for usage information
-          if (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;
+          const message = genChunk.message ?? {};
+          const messageResponseMetadata = message.response_metadata ?? {};
+          const generationInfo = genChunk.generationInfo ?? {};
+          const generationResponseMetadata = generationInfo.response_metadata ?? {};
+          const chunkUsage = message.usage_metadata ?? messageResponseMetadata['usage'] ?? messageResponseMetadata['amazon-bedrock-invocationMetrics'] ?? generationInfo.usage_metadata ?? generationResponseMetadata['usage'] ?? generationResponseMetadata['amazon-bedrock-invocationMetrics'];
           if (chunkUsage) {
             llmUsage = this._parseUsageModel(chunkUsage, provider, model);
             return llmUsage;