| import { FILE_PATH_SEPARATOR_REGEX } from '$lib/constants'; | |
| /** | |
| * Normalizes a model name by extracting the filename from a path, but preserves Hugging Face repository format. | |
| * | |
| * Handles both forward slashes (/) and backslashes (\) as path separators. | |
| * - If the model name has exactly one slash (org/model format), preserves the full "org/model" name | |
| * - If the model name has no slash or multiple slashes, extracts just the filename | |
| * - If the model name is just a filename (no path), returns it as-is. | |
| * | |
| * @param modelName - The model name or path to normalize | |
| * @returns The normalized model name | |
| * | |
| * @example | |
| * normalizeModelName('models/llama-3.1-8b') // Returns: 'llama-3.1-8b' (multiple slashes -> filename) | |
| * normalizeModelName('C:\\Models\\gpt-4') // Returns: 'gpt-4' (multiple slashes -> filename) | |
| * normalizeModelName('meta-llama/Llama-3.1-8B') // Returns: 'meta-llama/Llama-3.1-8B' (Hugging Face format) | |
| * normalizeModelName('simple-model') // Returns: 'simple-model' (no slash) | |
| * normalizeModelName(' spaced ') // Returns: 'spaced' | |
| * normalizeModelName('') // Returns: '' | |
| */ | |
| export function normalizeModelName(modelName: string): string { | |
| const trimmed = modelName.trim(); | |
| if (!trimmed) { | |
| return ''; | |
| } | |
| const segments = trimmed.split(FILE_PATH_SEPARATOR_REGEX); | |
| // If we have exactly 2 segments (one slash), treat it as Hugging Face repo format | |
| // and preserve the full "org/model" format | |
| if (segments.length === 2) { | |
| const [org, model] = segments; | |
| const trimmedOrg = org?.trim(); | |
| const trimmedModel = model?.trim(); | |
| if (trimmedOrg && trimmedModel) { | |
| return `${trimmedOrg}/${trimmedModel}`; | |
| } | |
| } | |
| // For other cases (no slash, or multiple slashes), extract just the filename | |
| const candidate = segments.pop(); | |
| const normalized = candidate?.trim(); | |
| return normalized && normalized.length > 0 ? normalized : trimmed; | |
| } | |
| /** | |
| * Validates if a model name is valid (non-empty after normalization). | |
| * | |
| * @param modelName - The model name to validate | |
| * @returns true if valid, false otherwise | |
| */ | |
| export function isValidModelName(modelName: string): boolean { | |
| return normalizeModelName(modelName).length > 0; | |
| } | |