@posthog/ai 7.20.12 → 7.20.13

package/dist/langchain/index.cjs

@@ -699,7 +699,7 @@ var BaseCallbackHandler = class extends BaseCallbackHandlerMethodsClass {
 	}
 };
 
-var version = "7.20.12";
+var version = "7.20.13";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;

package/dist/langchain/index.mjs

@@ -677,7 +677,7 @@ var BaseCallbackHandler = class extends BaseCallbackHandlerMethodsClass {
 	}
 };
 
-var version = "7.20.12";
+var version = "7.20.13";
 
 const DEFAULT_MAX_DEPTH = 3;
 const MAX_STACK_LINES = 20;

package/dist/openai/index.cjs

@@ -534,7 +534,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;
@@ -765,6 +765,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 {
@@ -807,7 +834,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 () => {
@@ -1000,13 +1027,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',
@@ -1054,7 +1082,7 @@ class WrappedCompletions extends Completions {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
 }
@@ -1080,7 +1108,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 () => {
@@ -1185,6 +1213,7 @@ class WrappedResponses extends Responses {
         }
         return value;
       });
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     } else {
       const wrappedPromise = parentPromise.then(async result => {
         if ('output' in result) {
@@ -1193,7 +1222,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',
@@ -1240,7 +1269,7 @@ class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
   parse(body, options) {
@@ -1299,7 +1328,7 @@ class WrappedResponses extends Responses {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     } finally {
       // Restore our wrapped create method
       originalSelfRecord['create'] = tempCreate;
@@ -1360,7 +1389,7 @@ class WrappedEmbeddings extends Embeddings {
       });
       throw error;
     });
-    return wrappedPromise;
+    return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
   }
 }
 class WrappedAudio extends Audio {
@@ -1399,7 +1428,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 () => {
@@ -1467,11 +1496,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',
@@ -1487,8 +1517,8 @@ class WrappedTranscriptions extends Transcriptions {
               rawUsage: result.usage
             }
           });
-          return result;
         }
+        return result;
       }, async error => {
         await captureAiGeneration(this.phClient, {
           ...posthogParams,
@@ -1507,7 +1537,7 @@ class WrappedTranscriptions extends Transcriptions {
         });
         throw error;
       });
-      return wrappedPromise;
+      return preserveAPIPromiseHelpers(parentPromise, wrappedPromise);
     }
   }
 }