@posthog/ai 7.20.11 → 7.20.13

package/dist/openai/index.mjs

@@ -530,7 +530,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;
@@ -761,6 +761,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 {
@@ -803,7 +830,7 @@ 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 () => {
@@ -996,13 +1023,14 @@ 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',
@@ -1050,7 +1078,7 @@ class WrappedCompletions extends Completions {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
 }
@@ -1076,7 +1104,7 @@ 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 () => {
@@ -1181,6 +1209,7 @@ class WrappedResponses extends Responses {
         }
         return value;
       });
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
@@ -1189,7 +1218,7 @@ 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',
@@ -1236,7 +1265,7 @@ class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
   parse(body, options) {
@@ -1295,7 +1324,7 @@ class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     } finally {
       // Restore our wrapped create method
       originalSelfRecord['create'] = tempCreate;
@@ -1356,7 +1385,7 @@ class WrappedEmbeddings extends Embeddings {
       });
       throw error;
     });
-    return wrappedPromise;
+    return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
   }
 }
 class WrappedAudio extends Audio {
@@ -1395,7 +1424,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 () => {
@@ -1463,11 +1492,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',
@@ -1483,8 +1513,8 @@ class WrappedTranscriptions extends Transcriptions {
               rawUsage: result.usage
             }
           });
-          return result;
         }
+        return result;
       }, async error => {
         await captureAiGeneration(this.phClient, {
           ...posthogParams,
@@ -1503,7 +1533,7 @@ class WrappedTranscriptions extends Transcriptions {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
 }