type TemplateType =
	| 'llama2'
	| 'alpaca'
	| 'zephyr'
	| 'phi2'
	| 'phind'
	| 'anthropic'
	| 'chatml'
	| 'none'
	| 'openchat'
	| 'deepseek'
	| 'xwin-coder'
	| 'neural-chat'
	| 'codellama-70b'
	| 'llava'
	| 'gemma';

export function autodetectTemplateType(model: string): TemplateType | undefined {
	const lower = model.toLowerCase();

	if (lower.includes('codellama') && lower.includes('70b')) {
		return 'codellama-70b';
	}

	if (
		lower.includes('gpt') ||
		lower.includes('command') ||
		lower.includes('chat-bison') ||
		lower.includes('pplx') ||
		lower.includes('gemini')
	) {
		return undefined;
	}

	if (lower.includes('llava')) {
		return 'llava';
	}

	if (lower.includes('tinyllama')) {
		return 'zephyr';
	}

	if (lower.includes('xwin')) {
		return 'xwin-coder';
	}

	if (lower.includes('dolphin')) {
		return 'chatml';
	}

	if (lower.includes('gemma')) {
		return 'gemma';
	}

	if (lower.includes('phi2')) {
		return 'phi2';
	}

	if (lower.includes('phind')) {
		return 'phind';
	}

	if (lower.includes('llama')) {
		return 'llama2';
	}

	if (lower.includes('zephyr')) {
		return 'zephyr';
	}

	// Claude requests always sent through Messages API, so formatting not necessary
	if (lower.includes('claude')) {
		return 'none';
	}

	if (lower.includes('alpaca') || lower.includes('wizard')) {
		return 'alpaca';
	}

	if (lower.includes('mistral') || lower.includes('mixtral')) {
		return 'llama2';
	}

	if (lower.includes('deepseek')) {
		return 'deepseek';
	}

	if (lower.includes('ninja') || lower.includes('openchat')) {
		return 'openchat';
	}

	if (lower.includes('neural-chat')) {
		return 'neural-chat';
	}

	return 'chatml';
}