@posthog/ai 7.20.12 → 7.20.13

package/dist/anthropic/index.cjs

@@ -299,7 +299,7 @@ function addDefaults(params) {
   };
 }
 
-var version = "7.20.12";
+var version = "7.20.13";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;

package/dist/anthropic/index.mjs

@@ -291,7 +291,7 @@ function addDefaults(params) {
   };
 }
 
-var version = "7.20.12";
+var version = "7.20.13";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;

package/dist/gemini/index.cjs

@@ -384,7 +384,7 @@ function addDefaults(params) {
   };
 }
 
-var version = "7.20.12";
+var version = "7.20.13";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;

package/dist/gemini/index.mjs

@@ -380,7 +380,7 @@ function addDefaults(params) {
   };
 }
 
-var version = "7.20.12";
+var version = "7.20.13";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;

package/dist/index.cjs

@@ -733,7 +733,7 @@ function formatOpenAIResponsesInput(input, instructions) {
   return messages;
 }
 
-var version = "7.20.12";
+var version = "7.20.13";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;
@@ -956,6 +956,33 @@ const Responses = openai.OpenAI.Responses;
 const Embeddings = openai.OpenAI.Embeddings;
 const Audio = openai.OpenAI.Audio;
 const Transcriptions = openai.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.OpenAI {
   constructor(config) {
     const {
@@ -991,7 +1018,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 () => {
@@ -1176,13 +1203,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',
@@ -1230,7 +1258,7 @@ let WrappedCompletions$1 = class WrappedCompletions extends Completions {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
 };
@@ -1249,7 +1277,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 () => {
@@ -1354,6 +1382,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) {
@@ -1362,7 +1391,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',
@@ -1409,7 +1438,7 @@ let WrappedResponses$1 = class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
   parse(body, options) {
@@ -1468,7 +1497,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;
@@ -1529,7 +1558,7 @@ let WrappedEmbeddings$1 = class WrappedEmbeddings extends Embeddings {
       });
       throw error;
     });
-    return wrappedPromise;
+    return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
   }
 };
 class WrappedAudio extends Audio {
@@ -1553,7 +1582,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 () => {
@@ -1621,11 +1650,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',
@@ -1641,8 +1671,8 @@ class WrappedTranscriptions extends Transcriptions {
               rawUsage: result.usage
             }
           });
-          return result;
         }
+        return result;
       }, async error => {
         await captureAiGeneration(this.phClient, {
           ...posthogParams,
@@ -1661,7 +1691,7 @@ class WrappedTranscriptions extends Transcriptions {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
 }