modelmix 5.0.6 → 5.1.2

package/test/model-chain.test.js

@@ -0,0 +1,12 @@
+const { expect } = require('chai');
+
+const { ModelMix } = require('../index.js');
+const { listChainModelShortcuts } = require('../lib/model-chain');
+
+describe('model chain catalog', () => {
+    it('contains only implemented ModelMix shortcuts', () => {
+        for (const shortcut of listChainModelShortcuts()) {
+            expect(ModelMix.prototype[shortcut], shortcut).to.be.a('function');
+        }
+    });
+});

package/test/moderation.test.js

@@ -100,7 +100,7 @@ describe('OpenAI moderation', () => {
     it('rejects generative providers from the moderation chain', () => {
         const model = ModerationMix.new();
 
-        expect(() => model.gpt41nano()).to.throw(
+        expect(() => model.gpt5nano({ config: { apiKey: 'test-key' } })).to.throw(
             'ModerationMix only accepts moderation providers.'
         );
     });

package/test/public-api.test.js

@@ -0,0 +1,54 @@
+const { expect } = require('chai');
+
+const api = require('../index.js');
+
+describe('public module boundary', () => {
+    it('preserves the CommonJS export surface', () => {
+        expect(Object.keys(api).sort()).to.deep.equal([
+            'MixAnthropic',
+            'MixCerebras',
+            'MixCustom',
+            'MixFireworks',
+            'MixGoogle',
+            'MixGrok',
+            'MixGroq',
+            'MixKimi',
+            'MixLMStudio',
+            'MixMiMo',
+            'MixMiniMax',
+            'MixModeration',
+            'MixNVIDIA',
+            'MixOllama',
+            'MixOpenAI',
+            'MixOpenAIModeration',
+            'MixOpenAIResponses',
+            'MixOpenAIWebSocket',
+            'MixOpenRouter',
+            'MixPerplexity',
+            'MixTogether',
+            'ModelMix',
+            'ModerationMix',
+            'applyUnifiedEffort',
+            'normalizeEffort',
+            'resolveProviderFamily'
+        ]);
+    });
+
+    it('preserves provider inheritance and class identity', () => {
+        expect(Object.getPrototypeOf(api.MixOpenAIResponses.prototype)).to.equal(api.MixOpenAI.prototype);
+        expect(Object.getPrototypeOf(api.MixOpenAIModeration.prototype)).to.equal(api.MixModeration.prototype);
+        expect(Object.getPrototypeOf(api.ModerationMix.prototype)).to.equal(api.ModelMix.prototype);
+        expect(require('../index.js').MixCustom).to.equal(api.MixCustom);
+    });
+
+    it('keeps root exports usable by model shortcuts', () => {
+        const model = api.ModelMix.new().qwen35397b({ config: { apiKey: 'test-key' } });
+
+        expect(model.models).to.have.length(1);
+        expect(model.models[0].provider.constructor).to.equal(api.MixOpenRouter);
+    });
+
+    it('keeps the mutable pricing catalog private', () => {
+        expect(require('../lib/token-usage')).to.not.have.property('MODEL_PRICING');
+    });
+});

package/test/tokens.test.js

@@ -576,6 +576,26 @@ describe('Token Usage Tracking', () => {
         }
     });
 
+    it('should register GPT-OSS 120B through the current OpenRouter model ID', function () {
+        const originalOpenRouterApiKey = process.env.OPENROUTER_API_KEY;
+        process.env.OPENROUTER_API_KEY = 'test-openrouter-key';
+
+        try {
+            const model = ModelMix.new().gptOss({
+                mix: { cerebras: false, groq: true, openrouter: true, together: false }
+            });
+
+            expect(model.models.map(({ key }) => key)).to.deep.equal([
+                'openai/gpt-oss-120b',
+                'openai/gpt-oss-120b'
+            ]);
+            expect(model.models[1].provider).to.be.instanceOf(MixOpenRouter);
+        } finally {
+            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';