@posthog/ai 7.20.11 → 7.20.13

package/dist/index.mjs

@@ -709,7 +709,7 @@ function formatOpenAIResponsesInput(input, instructions) {
   return messages;
 }
 
-var version = "7.20.11";
+var version = "7.20.13";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -932,6 +932,33 @@ const Responses = OpenAI.Responses;
 const Embeddings = OpenAI.Embeddings;
 const Audio = OpenAI.Audio;
 const Transcriptions = OpenAI.Audio.Transcriptions;
+function captureAiGenerationInBackground(...args) {
+  void captureAiGeneration(...args).catch(() => undefined);
+}
+async function captureAiGenerationAfterSuccess(...args) {
+  const [, options] = args;
+  if (options.captureImmediate) {
+    await captureAiGeneration(...args);
+  } else {
+    captureAiGenerationInBackground(...args);
+  }
+}
+function preserveAPIPromiseHelpers(parentPromise, wrappedPromise) {
+  const apiPromise = wrappedPromise;
+  if (typeof parentPromise.asResponse === 'function') {
+    apiPromise.asResponse = () => parentPromise.asResponse();
+  }
+  if (typeof parentPromise.withResponse === 'function') {
+    apiPromise.withResponse = async () => {
+      const [response, data] = await Promise.all([parentPromise.withResponse(), wrappedPromise]);
+      return {
+        ...response,
+        data
+      };
+    };
+  }
+  return apiPromise;
+}
 class PostHogOpenAI extends OpenAI {
   constructor(config) {
     const {
@@ -967,7 +994,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
-      return parentPromise.then(value => {
+      const wrappedPromise = parentPromise.then(value => {
         if ('tee' in value) {
           const [stream1, stream2] = value.tee();
           (async () => {
@@ -1152,13 +1179,14 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         }
         return value;
       });
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('choices' in result) {
           const latency = (Date.now() - startTime) / 1000;
           const availableTools = extractAvailableToolCalls('openai', openAIParams);
           const formattedOutput = formatResponseOpenAI(result);
-          await captureAiGeneration(this.phClient, {
+          await captureAiGenerationAfterSuccess(this.phClient, {
             ...posthogParams,
             model: openAIParams.model ?? result.model,
             provider: 'openai',
@@ -1206,7 +1234,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
 };
@@ -1225,7 +1253,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
     const startTime = Date.now();
     const parentPromise = super.create(openAIParams, options);
     if (openAIParams.stream) {
-      return parentPromise.then(value => {
+      const wrappedPromise = parentPromise.then(value => {
         if ('tee' in value && typeof value.tee === 'function') {
           const [stream1, stream2] = value.tee();
           (async () => {
@@ -1330,6 +1358,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         }
         return value;
       });
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
@@ -1338,7 +1367,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
           const formattedOutput = formatResponseOpenAI({
             output: result.output
           });
-          await captureAiGeneration(this.phClient, {
+          await captureAiGenerationAfterSuccess(this.phClient, {
             ...posthogParams,
             model: openAIParams.model ?? result.model,
             provider: 'openai',
@@ -1385,7 +1414,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
   parse(body, options) {
@@ -1444,7 +1473,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     } finally {
       // Restore our wrapped create method
       originalSelfRecord['create'] = tempCreate;
@@ -1505,7 +1534,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
       });
       throw error;
     });
-    return wrappedPromise;
+    return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
   }
 };
 class WrappedAudio extends Audio {
@@ -1529,7 +1558,7 @@ class WrappedTranscriptions extends Transcriptions {
     const startTime = Date.now();
     const parentPromise = openAIParams.stream ? super.create(openAIParams, options) : super.create(openAIParams, options);
     if (openAIParams.stream) {
-      return parentPromise.then(value => {
+      const wrappedPromise = parentPromise.then(value => {
         if ('tee' in value && typeof value.tee === 'function') {
           const [stream1, stream2] = value.tee();
           (async () => {
@@ -1597,11 +1626,12 @@ class WrappedTranscriptions extends Transcriptions {
         }
         return value;
       });
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
-        if ('text' in result) {
+        if (result && typeof result === 'object' && 'text' in result) {
           const latency = (Date.now() - startTime) / 1000;
-          await captureAiGeneration(this.phClient, {
+          await captureAiGenerationAfterSuccess(this.phClient, {
             ...posthogParams,
             model: openAIParams.model,
             provider: 'openai',
@@ -1617,8 +1647,8 @@ class WrappedTranscriptions extends Transcriptions {
               rawUsage: result.usage
             }
           });
-          return result;
         }
+        return result;
       }, async error => {
         await captureAiGeneration(this.phClient, {
           ...posthogParams,
@@ -1637,7 +1667,7 @@ class WrappedTranscriptions extends Transcriptions {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
 }