modelmix 4.5.18 → 4.5.20

package/index.js

@@ -530,7 +530,21 @@ class ModelMix {
         if (mix.minimax) return this.attach('MiniMax-M2.7', new MixMiniMax({ options, config }));
         if (mix.together) return this.attach('MiniMaxAI/MiniMax-M2.7', new MixTogether({ options, config }));
         return this;
-    }  
+    }
+
+    mimo25({ options = {}, config = {}, mix = { openrouter: true } } = {}) {
+        mix = { ...this.mix, ...mix };
+        if (mix.openrouter) this.attach('xiaomi/mimo-v2.5', new MixOpenRouter({ options, config }));
+        if (mix.mimo) this.attach('mimo-v2.5', new MixMiMo({ options, config }));
+        return this;
+    }
+
+    mimo25pro({ options = {}, config = {}, mix = { openrouter: true } } = {}) {
+        mix = { ...this.mix, ...mix };
+        if (mix.openrouter) this.attach('xiaomi/mimo-v2.5-pro', new MixOpenRouter({ options, config }));
+        if (mix.mimo) this.attach('mimo-v2.5-pro', new MixMiMo({ options, config }));
+        return this;
+    }
 
     deepseekV4Pro({ options = {}, config = {}, mix = { fireworks: true } } = {}) {
         mix = { ...this.mix, ...mix };
@@ -2316,6 +2330,29 @@ class MixMiniMax extends MixOpenAI {
     }
 }
 
+class MixMiMo extends MixOpenAI {
+    getDefaultConfig(customConfig) {
+        if (!process.env.MIMO_API_KEY) {
+            throw new Error('MiMo API key not found. Please provide it in config or set MIMO_API_KEY environment variable.');
+        }
+
+        return MixCustom.prototype.getDefaultConfig.call(this, {
+            url: 'https://api.xiaomimimo.com/v1/chat/completions',
+            apiKey: process.env.MIMO_API_KEY,
+            ...customConfig
+        });
+    }
+
+    getDefaultHeaders(customHeaders) {
+        return {
+            'accept': 'application/json',
+            'content-type': 'application/json',
+            'api-key': this.config.apiKey,
+            ...customHeaders
+        };
+    }
+}
+
 class MixPerplexity extends MixCustom {
     getDefaultConfig(customConfig) {
 
@@ -2854,4 +2891,4 @@ class MixGoogle extends MixCustom {
     }
 }
 
-module.exports = { MixCustom, ModelMix, MixAnthropic, MixMiniMax, MixOpenAI, MixOpenAIResponses, MixOpenAIWebSocket, MixOpenRouter, MixPerplexity, MixOllama, MixLMStudio, MixGroq, MixTogether, MixGrok, MixCerebras, MixGoogle, MixFireworks, MixNVIDIA };
+module.exports = { MixCustom, ModelMix, MixAnthropic, MixMiniMax, MixMiMo, MixOpenAI, MixOpenAIResponses, MixOpenAIWebSocket, MixOpenRouter, MixPerplexity, MixOllama, MixLMStudio, MixGroq, MixTogether, MixGrok, MixCerebras, MixGoogle, MixFireworks, MixNVIDIA };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "modelmix",
-  "version": "4.5.18",
+  "version": "4.5.20",
   "description": "🧬 Reliable interface with automatic fallback for AI LLMs.",
   "main": "index.js",
   "repository": {
@@ -32,7 +32,7 @@
     "lmstudio",
     "deepseek",
     "oss",
-    "k2",
+    "k26",
     "reasoning",
     "minimax",
     "thinking",

package/test/tokens.test.js

@@ -1,5 +1,5 @@
 import { expect } from 'chai';
-import { ModelMix, MixAnthropic, MixCustom, MixGoogle, MixOpenAIResponses } from '../index.js';
+import { ModelMix, MixAnthropic, MixCustom, MixGoogle, MixMiMo, MixOpenAIResponses, MixOpenRouter } from '../index.js';
 import { createRequire } from 'module';
 
 const require = createRequire(import.meta.url);
@@ -88,6 +88,62 @@ describe('Token Usage Tracking', () => {
         expect(model.models[1].provider).to.be.instanceOf(MixOpenAIResponses);
     });
 
+    it('should register MiMo shortcuts with native and OpenRouter providers', function () {
+        const originalMimoApiKey = process.env.MIMO_API_KEY;
+        const originalOpenRouterApiKey = process.env.OPENROUTER_API_KEY;
+
+        process.env.MIMO_API_KEY = 'test-mimo-key';
+        process.env.OPENROUTER_API_KEY = 'test-openrouter-key';
+
+        try {
+            const model = ModelMix.new()
+                .mimo25()
+                .mimo25pro({ mix: { mimo: true, openrouter: true } });
+
+            expect(model.models).to.have.length(3);
+            expect(model.models[0].key).to.equal('xiaomi/mimo-v2.5');
+            expect(model.models[1].key).to.equal('mimo-v2.5-pro');
+            expect(model.models[2].key).to.equal('xiaomi/mimo-v2.5-pro');
+
+            expect(model.models[0].provider).to.be.instanceOf(MixOpenRouter);
+            expect(model.models[1].provider).to.be.instanceOf(MixMiMo);
+            expect(model.models[2].provider).to.be.instanceOf(MixOpenRouter);
+        } finally {
+            if (originalMimoApiKey === undefined) delete process.env.MIMO_API_KEY;
+            else process.env.MIMO_API_KEY = originalMimoApiKey;
+
+            if (originalOpenRouterApiKey === undefined) delete process.env.OPENROUTER_API_KEY;
+            else process.env.OPENROUTER_API_KEY = originalOpenRouterApiKey;
+        }
+    });
+
+    it('should use api-key header for MiMo provider', function () {
+        const originalMimoApiKey = process.env.MIMO_API_KEY;
+        process.env.MIMO_API_KEY = 'test-mimo-key';
+
+        try {
+            const provider = new MixMiMo();
+            expect(provider.headers['api-key']).to.equal('test-mimo-key');
+            expect(provider.headers.authorization).to.equal(undefined);
+            expect(provider.config.url).to.equal('https://api.xiaomimimo.com/v1/chat/completions');
+        } finally {
+            if (originalMimoApiKey === undefined) delete process.env.MIMO_API_KEY;
+            else process.env.MIMO_API_KEY = originalMimoApiKey;
+        }
+    });
+
+    it('should throw a clear error when MIMO_API_KEY is missing', function () {
+        const originalMimoApiKey = process.env.MIMO_API_KEY;
+        delete process.env.MIMO_API_KEY;
+
+        try {
+            expect(() => new MixMiMo()).to.throw('MIMO_API_KEY');
+        } finally {
+            if (originalMimoApiKey === undefined) delete process.env.MIMO_API_KEY;
+            else process.env.MIMO_API_KEY = originalMimoApiKey;
+        }
+    });
+
     it('should track tokens in OpenAI response', async function () {
         this.timeout(30000);