@posthog/ai 8.6.0 → 8.6.2

package/dist/langchain/index.mjs

@@ -215,7 +215,7 @@ function sanitizeValues(obj) {
   return jsonSafe;
 }
 
-var version = "8.6.0";
+var version = "8.6.2";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -306,6 +306,13 @@ const warnIfPostHogAiGateway = baseURL => {
   console.warn('[PostHog] The PostHog AI wrapper is pointed at the PostHog AI Gateway. ' + 'Both capture $ai_generation, so every call is double-counted and double-billed. ' + `Use one or the other — see ${GATEWAY_DOCS_URL}.`);
 };
 
+// Mirror LangGraph's isGraphBubbleUp guard without adding LangGraph as a dependency. Every
+// LangGraph control-flow exception (GraphInterrupt, NodeInterrupt, ParentCommand, GraphDrained,
+// and future subclasses) exposes a prototype getter `is_bubble_up` that returns true, which the
+// LangGraph runtime itself uses to distinguish control flow from real failures. The getter reads
+// as undefined on ordinary Errors and works across duplicated LangGraph package copies.
+const isLangGraphControlFlow = error => error.is_bubble_up === true;
+
 /** A run may either be a Span or a Generation */
 
 /** Storage for run metadata */
@@ -557,8 +564,20 @@ class LangChainCallbackHandler extends BaseCallbackHandler {
       eventProperties['$process_person_profile'] = false;
     }
     if (outputs instanceof Error) {
-      eventProperties['$ai_error'] = stringifyError(outputs);
-      eventProperties['$ai_is_error'] = true;
+      if (isLangGraphControlFlow(outputs)) {
+        // GraphInterrupt carries the pending interrupts (e.g. the question posed to a human).
+        // Surface them under the same `__interrupt__` key LangGraph hands back to the caller,
+        // so an interrupted span stays distinguishable from a node that returned nothing.
+        const interrupts = outputs.interrupts;
+        if (interrupts !== undefined) {
+          eventProperties['$ai_output_state'] = withPrivacyMode(this.client, this.privacyMode, sanitizeLangChain({
+            __interrupt__: interrupts
+          }));
+        }
+      } 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));
     }
@@ -783,27 +802,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']];
@@ -921,14 +923,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;