modelmix 3.5.6 → 3.6.0

package/index.js

@@ -1,6 +1,6 @@
 const axios = require('axios');
 const fs = require('fs');
-const mime = require('mime-types');
+const { fileTypeFromBuffer } = require('file-type');
 const log = require('lemonlog')('ModelMix');
 const Bottleneck = require('bottleneck');
 const path = require('path');
@@ -203,6 +203,12 @@ class ModelMix {
     }
 
     addImageFromBuffer(buffer, { role = "user" } = {}) {
+
+        const fileType = fileTypeFromBuffer(buffer);
+        if (!fileType || !fileType.mime.startsWith('image/')) {
+            throw new Error('Invalid image buffer - unable to detect valid image format');
+        }
+
         const data = buffer.toString('base64');
 
         const imageMessage = {
@@ -212,7 +218,7 @@ class ModelMix {
                     type: "image",
                     "source": {
                         type: "base64",
-                        media_type: mimeType,
+                        media_type: fileType.mime,
                         data
                     }
                 }
@@ -225,14 +231,7 @@ class ModelMix {
 
     addImage(filePath, { role = "user" } = {}) {
         const imageBuffer = this.readFile(filePath, { encoding: null });
-        const mimeType = mime.lookup(filePath);
-
-        if (!mimeType || !mimeType.startsWith('image/')) {
-            throw new Error('Invalid image file type');
-        }
-
-        this.addImageFromBuffer(imageBuffer, { role });
-        return this;
+        return this.addImageFromBuffer(imageBuffer, { role });
     }
 
     addImageFromUrl(url, config = { role: "user" }) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "modelmix",
-  "version": "3.5.6",
+  "version": "3.6.0",
   "description": "🧬 ModelMix - Unified API for Diverse AI LLM.",
   "main": "index.js",
   "repository": {
@@ -23,7 +23,7 @@
     "mixtral",
     "chat",
     "opus",
-    "sonnet",    
+    "sonnet",
     "multimodal",
     "groq",
     "gemini",
@@ -53,6 +53,7 @@
     "@modelcontextprotocol/sdk": "^1.11.2",
     "axios": "^1.8.4",
     "bottleneck": "^2.19.5",
+    "file-type": "^16.5.4",
     "lemonlog": "^1.1.2"
   }
 }