1import { describe, expect, it } from 'vitest';
2import { isValidModelName, normalizeModelName } from '$lib/utils/model-names';
3
4describe('normalizeModelName', () => {
5 it('preserves Hugging Face org/model format (single slash)', () => {
6 // Single slash is treated as Hugging Face format and preserved
7 expect(normalizeModelName('meta-llama/Llama-3.1-8B')).toBe('meta-llama/Llama-3.1-8B');
8 expect(normalizeModelName('models/model-name-1')).toBe('models/model-name-1');
9 });
10
11 it('extracts filename from multi-segment paths', () => {
12 // Multiple slashes -> extract just the filename
13 expect(normalizeModelName('path/to/model/model-name-2')).toBe('model-name-2');
14 expect(normalizeModelName('/absolute/path/to/model')).toBe('model');
15 });
16
17 it('extracts filename from backslash paths', () => {
18 expect(normalizeModelName('C\\Models\\model-name-1')).toBe('model-name-1');
19 expect(normalizeModelName('path\\to\\model\\model-name-2')).toBe('model-name-2');
20 });
21
22 it('handles mixed path separators', () => {
23 expect(normalizeModelName('path/to\\model/model-name-2')).toBe('model-name-2');
24 });
25
26 it('returns simple names as-is', () => {
27 expect(normalizeModelName('simple-model')).toBe('simple-model');
28 expect(normalizeModelName('model-name-2')).toBe('model-name-2');
29 });
30
31 it('trims whitespace', () => {
32 expect(normalizeModelName(' model-name ')).toBe('model-name');
33 });
34
35 it('returns empty string for empty input', () => {
36 expect(normalizeModelName('')).toBe('');
37 expect(normalizeModelName(' ')).toBe('');
38 });
39});
40
41describe('isValidModelName', () => {
42 it('returns true for valid names', () => {
43 expect(isValidModelName('model')).toBe(true);
44 expect(isValidModelName('path/to/model.bin')).toBe(true);
45 });
46
47 it('returns false for empty values', () => {
48 expect(isValidModelName('')).toBe(false);
49 expect(isValidModelName(' ')).toBe(false);
50 });
51});