moving

packages/agent/tests/model-class-router.test.ts

@@ -0,0 +1,50 @@
+import { describe, it, expect } from 'vitest';
+import { resolveModelForClass, LIGHTWEIGHT_MODEL } from '../src/model-class-router.js';
+
+describe('resolveModelForClass', () => {
+  it('routes a lightweight call to the cheap model', () => {
+    expect(
+      resolveModelForClass('claude-sonnet-4-6', { class: 'lightweight', lightweightModel: LIGHTWEIGHT_MODEL }),
+    ).toBe(LIGHTWEIGHT_MODEL);
+  });
+
+  it('leaves a lightweight call unchanged when no cheap model is supplied', () => {
+    expect(resolveModelForClass('claude-sonnet-4-6', { class: 'lightweight' })).toBe('claude-sonnet-4-6');
+  });
+
+  it('leaves general / reasoning / unclassified calls unchanged', () => {
+    const opts = { lightweightModel: LIGHTWEIGHT_MODEL };
+    expect(resolveModelForClass('claude-opus-4-8', { ...opts, class: 'reasoning' })).toBe('claude-opus-4-8');
+    expect(resolveModelForClass('claude-opus-4-8', { ...opts, class: 'general' })).toBe('claude-opus-4-8');
+    expect(resolveModelForClass('claude-opus-4-8', opts)).toBe('claude-opus-4-8');
+  });
+
+  it('routes a privacy-required call to the on-device model when provided', () => {
+    expect(
+      resolveModelForClass('claude-sonnet-4-6', { privacyRequired: true, localModel: 'ollama/llama3.1' }),
+    ).toBe('ollama/llama3.1');
+  });
+
+  it('FAILS CLOSED: privacy-required with no on-device model throws (never cloud)', () => {
+    // privacyRequired + lightweight, but no local model → must throw rather than
+    // fall through to the cloud lightweightModel or the requested cloud model.
+    expect(() =>
+      resolveModelForClass('claude-sonnet-4-6', {
+        privacyRequired: true,
+        class: 'lightweight',
+        lightweightModel: LIGHTWEIGHT_MODEL,
+      }),
+    ).toThrow(/privacyRequired/);
+  });
+
+  it('privacy precedence: on-device model wins over the lightweight override', () => {
+    expect(
+      resolveModelForClass('claude-sonnet-4-6', {
+        privacyRequired: true,
+        localModel: 'ollama/qwen3',
+        class: 'lightweight',
+        lightweightModel: LIGHTWEIGHT_MODEL,
+      }),
+    ).toBe('ollama/qwen3');
+  });
+});