@posthog/ai 7.2.1 → 7.3.0

package/dist/openai/index.cjs

@@ -6,7 +6,7 @@ var openai = require('openai');
 var buffer = require('buffer');
 var uuid = require('uuid');
 
-var version = "7.2.1";
+var version = "7.3.0";
 
 // Type guards for safer type checking
 
@@ -85,6 +85,14 @@ const formatResponseOpenAI = response => {
             });
           }
         }
+
+        // Handle audio output (gpt-4o-audio-preview)
+        if (choice.message.audio) {
+          content.push({
+            type: 'audio',
+            ...choice.message.audio
+          });
+        }
       }
       if (content.length > 0) {
         output.push({
@@ -466,6 +474,15 @@ function formatOpenAIResponsesInput(input, instructions) {
 
 const REDACTED_IMAGE_PLACEHOLDER = '[base64 image redacted]';
 
+// ============================================
+// Multimodal Feature Toggle
+// ============================================
+
+const isMultimodalEnabled = () => {
+  const val = process.env._INTERNAL_LLMA_MULTIMODAL || '';
+  return val.toLowerCase() === 'true' || val === '1' || val.toLowerCase() === 'yes';
+};
+
 // ============================================
 // Base64 Detection Helpers
 // ============================================
@@ -493,6 +510,7 @@ const isRawBase64 = str => {
   return str.length > 20 && /^[A-Za-z0-9+/]+=*$/.test(str);
 };
 function redactBase64DataUrl(str) {
+  if (isMultimodalEnabled()) return str;
   if (!isString(str)) return str;
 
   // Check for data URL format
@@ -555,6 +573,15 @@ const sanitizeOpenAIImage = item => {
       }
     };
   }
+
+  // Handle audio format
+  if (item.type === 'audio' && 'data' in item) {
+    if (isMultimodalEnabled()) return item;
+    return {
+      ...item,
+      data: REDACTED_IMAGE_PLACEHOLDER
+    };
+  }
   return item;
 };
 
@@ -618,6 +645,7 @@ class WrappedCompletions extends Completions {
             try {
               const contentBlocks = [];
               let accumulatedContent = '';
+              let modelFromResponse;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0,
@@ -627,6 +655,10 @@ class WrappedCompletions extends Completions {
               // Map to track in-progress tool calls
               const toolCallsInProgress = new Map();
               for await (const chunk of stream1) {
+                // Extract model from chunk (Chat Completions chunks have model field)
+                if (!modelFromResponse && chunk.model) {
+                  modelFromResponse = chunk.model;
+                }
                 const choice = chunk?.choices?.[0];
                 const chunkWebSearchCount = calculateWebSearchCount(chunk);
                 if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
@@ -720,7 +752,7 @@ class WrappedCompletions extends Completions {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                model: openAIParams.model,
+                model: openAIParams.model ?? modelFromResponse,
                 provider: 'openai',
                 input: sanitizeOpenAI(openAIParams.messages),
                 output: formattedOutput,
@@ -774,7 +806,7 @@ class WrappedCompletions extends Completions {
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
-            model: openAIParams.model,
+            model: openAIParams.model ?? result.model,
             provider: 'openai',
             input: sanitizeOpenAI(openAIParams.messages),
             output: formattedOutput,
@@ -798,7 +830,7 @@ class WrappedCompletions extends Completions {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model,
           provider: 'openai',
           input: sanitizeOpenAI(openAIParams.messages),
           output: [],
@@ -847,6 +879,7 @@ class WrappedResponses extends Responses {
           (async () => {
             try {
               let finalContent = [];
+              let modelFromResponse;
               let usage = {
                 inputTokens: 0,
                 outputTokens: 0,
@@ -854,6 +887,10 @@ class WrappedResponses extends Responses {
               };
               for await (const chunk of stream1) {
                 if ('response' in chunk && chunk.response) {
+                  // Extract model from response object in chunk (for stored prompts)
+                  if (!modelFromResponse && chunk.response.model) {
+                    modelFromResponse = chunk.response.model;
+                  }
                   const chunkWebSearchCount = calculateWebSearchCount(chunk.response);
                   if (chunkWebSearchCount > 0 && chunkWebSearchCount > (usage.webSearchCount ?? 0)) {
                     usage.webSearchCount = chunkWebSearchCount;
@@ -877,8 +914,7 @@ class WrappedResponses extends Responses {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                //@ts-expect-error
-                model: openAIParams.model,
+                model: openAIParams.model ?? modelFromResponse,
                 provider: 'openai',
                 input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
                 output: finalContent,
@@ -900,7 +936,6 @@ class WrappedResponses extends Responses {
               await sendEventToPosthog({
                 client: this.phClient,
                 ...posthogParams,
-                //@ts-expect-error
                 model: openAIParams.model,
                 provider: 'openai',
                 input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
@@ -933,8 +968,7 @@ class WrappedResponses extends Responses {
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
-            //@ts-expect-error
-            model: openAIParams.model,
+            model: openAIParams.model ?? result.model,
             provider: 'openai',
             input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
             output: formattedOutput,
@@ -958,7 +992,7 @@ class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model,
           provider: 'openai',
           input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
@@ -995,7 +1029,7 @@ class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model ?? result.model,
           provider: 'openai',
           input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: result.output,
@@ -1016,7 +1050,7 @@ class WrappedResponses extends Responses {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model,
           provider: 'openai',
           input: formatOpenAIResponsesInput(openAIParams.input, openAIParams.instructions),
           output: [],
@@ -1206,7 +1240,7 @@ class WrappedTranscriptions extends Transcriptions {
           await sendEventToPosthog({
             client: this.phClient,
             ...posthogParams,
-            model: String(openAIParams.model ?? ''),
+            model: openAIParams.model,
             provider: 'openai',
             input: openAIParams.prompt,
             output: result.text,
@@ -1226,7 +1260,7 @@ class WrappedTranscriptions extends Transcriptions {
         await sendEventToPosthog({
           client: this.phClient,
           ...posthogParams,
-          model: String(openAIParams.model ?? ''),
+          model: openAIParams.model,
           provider: 'openai',
           input: openAIParams.prompt,
           output: [],