Team Ai
Modelpublic

Felipe97/llama-cpp-compiled

sourceHugging Faceupdated 21d agoView on Hugging Face
0likes1.2kdownloads
model-names.test.ts52 linesDownload Raw Back to unit
1import { isValidModelName, normalizeModelName } from '$lib/utils/model-names';2import { describe, expect, it } from 'vitest';3 4describe('normalizeModelName', () => {5	it('preserves Hugging Face org/model format (single slash)', () => {6		// Single slash is treated as Hugging Face format and preserved7		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 filename13		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});52