model-key-normalization.test.ts 805 B

123456789101112131415161718192021
  1. /// <reference types="bun-types" />
  2. import { describe, expect, test } from 'bun:test';
  3. import { buildModelKeyAliases } from './model-key-normalization';
  4. describe('model key normalization', () => {
  5. test('normalizes multi-segment chutes model ids', () => {
  6. const aliases = buildModelKeyAliases(
  7. 'chutes/Qwen/Qwen3-Coder-480B-A35B-Instruct-FP8-TEE',
  8. );
  9. expect(aliases).toContain('qwen/qwen3-coder-480b-a35b-instruct');
  10. expect(aliases).toContain('qwen3-coder-480b-a35b-instruct');
  11. expect(aliases).not.toContain('qwen3-coder-480b-a35b-instruct-fp8-tee');
  12. });
  13. test('treats spaces and hyphens as equivalent aliases', () => {
  14. const aliases = buildModelKeyAliases('Qwen3 Coder 480B A35B Instruct');
  15. expect(aliases).toContain('qwen3-coder-480b-a35b-instruct');
  16. });
  17. });