@posthog/ai 7.13.2 → 7.14.0

package/dist/openai/index.mjs

@@ -2,7 +2,7 @@ import { OpenAI } from 'openai';
 import { v4 } from 'uuid';
 import { uuidv7 } from '@posthog/core';
 
-var version = "7.13.2";
+var version = "7.14.0";
 
 // Type guards for safer type checking
 
@@ -520,6 +520,7 @@ const sendEventToPosthog = async ({
   usage = {},
   error,
   exceptionId,
+  stopReason,
   tools,
   captureImmediate = false
 }) => {
@@ -590,6 +591,9 @@ const sendEventToPosthog = async ({
     ...(distinctId ? {} : {
       $process_person_profile: false
     }),
+    ...(stopReason ? {
+      $ai_stop_reason: stopReason
+    } : {}),
     ...(tools ? {
       $ai_tools: tools
     } : {}),
@@ -721,6 +725,7 @@ class WrappedCompletions extends Completions {
               let accumulatedContent = '';
               let modelFromResponse;
               let firstTokenTime;
+              let stopReason;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0,
@@ -736,6 +741,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;
@@ -853,6 +861,7 @@ class WrappedCompletions extends Completions {
                   webSearchCount: usage.webSearchCount,
                   rawUsage: rawUsageData
                 },
+                stopReason,
                 tools: availableTools
               });
             } catch (error) {
@@ -906,6 +915,7 @@ class WrappedCompletions extends Completions {
               webSearchCount: calculateWebSearchCount(result),
               rawUsage: result.usage
             },
+            stopReason: result.choices[0]?.finish_reason ?? undefined,
             tools: availableTools
           });
         }
@@ -965,6 +975,7 @@ class WrappedResponses extends Responses {
               let finalContent = [];
               let modelFromResponse;
               let firstTokenTime;
+              let stopReason;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0,
@@ -988,6 +999,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;
@@ -1023,6 +1037,7 @@ class WrappedResponses extends Responses {
                   webSearchCount: usage.webSearchCount,
                   rawUsage: rawUsageData
                 },
+                stopReason,
                 tools: availableTools
               });
             } catch (error) {
@@ -1076,6 +1091,7 @@ class WrappedResponses extends Responses {
               webSearchCount: calculateWebSearchCount(result),
               rawUsage: result.usage
             },
+            stopReason: result.status ?? undefined,
             tools: availableTools
           });
         }
@@ -1135,7 +1151,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 => {