modelmix 3.5.4 → 3.5.8

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');
@@ -202,15 +202,14 @@ class ModelMix {
         return this;
     }
 
-    addImage(filePath, { role = "user" } = {}) {
-        const imageBuffer = this.readFile(filePath, { encoding: null });
-        const mimeType = mime.lookup(filePath);
+    addImageFromBuffer(buffer, { role = "user" } = {}) {
 
-        if (!mimeType || !mimeType.startsWith('image/')) {
-            throw new Error('Invalid image file type');
+        const fileType = fileTypeFromBuffer(buffer);
+        if (!fileType || !fileType.mime.startsWith('image/')) {
+            throw new Error('Invalid image buffer - unable to detect valid image format');
         }
 
-        const data = imageBuffer.toString('base64');
+        const data = buffer.toString('base64');
 
         const imageMessage = {
             ...{ role },
@@ -219,7 +218,7 @@ class ModelMix {
                     type: "image",
                     "source": {
                         type: "base64",
-                        media_type: mimeType,
+                        media_type: fileType.mime,
                         data
                     }
                 }
@@ -230,6 +229,11 @@ class ModelMix {
         return this;
     }
 
+    addImage(filePath, { role = "user" } = {}) {
+        const imageBuffer = this.readFile(filePath, { encoding: null });
+        return this.addImageFromBuffer(imageBuffer, { role });
+    }
+
     addImageFromUrl(url, config = { role: "user" }) {
         if (!this.imagesToProcess) {
             this.imagesToProcess = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "modelmix",
-  "version": "3.5.4",
+  "version": "3.5.8",
   "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": "^21.0.0",
     "lemonlog": "^1.1.2"
   }
 }