@posthog/ai 7.13.2 → 7.14.0

package/dist/openai/index.cjs

@@ -6,7 +6,7 @@ var openai = require('openai');
 var uuid = require('uuid');
 var core = require('@posthog/core');
 
-var version = "7.13.2";
+var version = "7.14.0";
 
 // Type guards for safer type checking
 
@@ -524,6 +524,7 @@ const sendEventToPosthog = async ({
   usage = {},
   error,
   exceptionId,
+  stopReason,
   tools,
   captureImmediate = false
 }) => {
@@ -594,6 +595,9 @@ const sendEventToPosthog = async ({
     ...(distinctId ? {} : {
       $process_person_profile: false
     }),
+    ...(stopReason ? {
+      $ai_stop_reason: stopReason
+    } : {}),
     ...(tools ? {
       $ai_tools: tools
     } : {}),
@@ -725,6 +729,7 @@ class WrappedCompletions extends Completions {
               let accumulatedContent = '';
               let modelFromResponse;
               let firstTokenTime;
+              let stopReason;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0,
@@ -740,6 +745,9 @@ class WrappedCompletions extends Completions {
                   modelFromResponse = chunk.model;
                 }
                 const choice = chunk?.choices?.[0];
+                if (choice?.finish_reason) {
+                  stopReason = choice.finish_reason;
+                }
                 const chunkWebSearchCount = calculateWebSearchCount(chunk);
                 if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
                   usage.webSearchCount = chunkWebSearchCount;
@@ -857,6 +865,7 @@ class WrappedCompletions extends Completions {
                   webSearchCount: usage.webSearchCount,
                   rawUsage: rawUsageData
                 },
+                stopReason,
                 tools: availableTools
               });
             } catch (error) {
@@ -910,6 +919,7 @@ class WrappedCompletions extends Completions {
               webSearchCount: calculateWebSearchCount(result),
               rawUsage: result.usage
             },
+            stopReason: result.choices[0]?.finish_reason ?? undefined,
             tools: availableTools
           });
         }
@@ -969,6 +979,7 @@ class WrappedResponses extends Responses {
               let finalContent = [];
               let modelFromResponse;
               let firstTokenTime;
+              let stopReason;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0,
@@ -992,6 +1003,9 @@ class WrappedResponses extends Responses {
                 }
                 if (chunk.type === 'response.completed' && 'response' in chunk && chunk.response?.output && chunk.response.output.length > 0) {
                   finalContent = chunk.response.output;
+                  if (chunk.response.status) {
+                    stopReason = chunk.response.status;
+                  }
                 }
                 if ('response' in chunk && chunk.response?.usage) {
                   rawUsageData = chunk.response.usage;
@@ -1027,6 +1041,7 @@ class WrappedResponses extends Responses {
                   webSearchCount: usage.webSearchCount,
                   rawUsage: rawUsageData
                 },
+                stopReason,
                 tools: availableTools
               });
             } catch (error) {
@@ -1080,6 +1095,7 @@ class WrappedResponses extends Responses {
               webSearchCount: calculateWebSearchCount(result),
               rawUsage: result.usage
             },
+            stopReason: result.status ?? undefined,
             tools: availableTools
           });
         }
@@ -1139,7 +1155,8 @@ class WrappedResponses extends Responses {
             reasoningTokens: result.usage?.output_tokens_details?.reasoning_tokens ?? 0,
             cacheReadInputTokens: result.usage?.input_tokens_details?.cached_tokens ?? 0,
             rawUsage: result.usage
-          }
+          },
+          stopReason: result.status ?? undefined
         });
         return result;
       }, async error => {