@posthog/ai 7.17.2 → 7.17.4

package/dist/index.mjs

@@ -411,6 +411,31 @@ const formatResponseOpenAI = response => {
   }
   return output;
 };
+const buildInlineDataBlock = (mimeType, data) => {
+  if (mimeType.startsWith('audio/')) {
+    return {
+      type: 'audio',
+      mime_type: mimeType,
+      data
+    };
+  }
+  if (mimeType.startsWith('image/')) {
+    return {
+      type: 'image',
+      inline_data: {
+        mime_type: mimeType,
+        data
+      }
+    };
+  }
+  return {
+    type: 'document',
+    inline_data: {
+      mime_type: mimeType,
+      data
+    }
+  };
+};
 const formatResponseGemini = response => {
   const output = [];
   if (response.candidates && Array.isArray(response.candidates)) {
@@ -432,8 +457,8 @@ const formatResponseGemini = response => {
               }
             });
           } else if (part.inlineData) {
-            // Handle audio/media inline data
-            const mimeType = part.inlineData.mimeType || 'audio/pcm';
+            // Handle inline data (images, audio, documents)
+            const mimeType = part.inlineData.mimeType || part.inlineData.mime_type || 'application/octet-stream';
             let data = part.inlineData.data;
             // Handle binary data (Uint8Array/Buffer -> base64)
             if (data instanceof Uint8Array) {
@@ -449,11 +474,7 @@ const formatResponseGemini = response => {
             }
             // Sanitize base64 data for images and other large inline data
             data = redactBase64DataUrl(data);
-            content.push({
-              type: 'audio',
-              mime_type: mimeType,
-              data: data
-            });
+            content.push(buildInlineDataBlock(mimeType, data));
           }
         }
         if (content.length > 0) {
@@ -756,7 +777,7 @@ function formatOpenAIResponsesInput(input, instructions) {
   return messages;
 }
 
-var version = "7.17.2";
+var version = "7.17.4";
 
 /**
  * Capture an `$ai_generation` (or `$ai_embedding`) event to PostHog.
@@ -3160,15 +3181,8 @@ class WrappedModels {
       // Handle inlineData (images, audio, PDFs)
       else if (part && typeof part === 'object' && 'inlineData' in part) {
         const inlineData = part.inlineData;
-        const mimeType = inlineData.mimeType || inlineData.mime_type || '';
-        const contentType = mimeType.startsWith('image/') ? 'image' : 'document';
-        blocks.push({
-          type: contentType,
-          inline_data: {
-            data: inlineData.data,
-            mime_type: mimeType
-          }
-        });
+        const mimeType = inlineData.mimeType || inlineData.mime_type || 'application/octet-stream';
+        blocks.push(buildInlineDataBlock(mimeType, inlineData.data));
       }
     }
     return blocks;