From 57ba38ec722e775bf9c4204472cee4ce1b1b4990 Mon Sep 17 00:00:00 2001
From: Xuan Son Nguyen <son@huggingface.co>
Date: Thu, 26 Jun 2025 17:22:39 +0200
Subject: [PATCH] ollama: fix gemma2 template

---
 packages/ollama-utils/src/chat-template.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/ollama-utils/src/chat-template.ts b/packages/ollama-utils/src/chat-template.ts
index fa65c651d5..3520b0774b 100644
--- a/packages/ollama-utils/src/chat-template.ts
+++ b/packages/ollama-utils/src/chat-template.ts
@@ -31,7 +31,7 @@ const CUSTOM_TEMPLATE_MAPPING: ((ggufTmpl: string) => OllamaCustomMappedTemplate
 			? {
 					// for some reason, gemma2 has weird variants
 					ollamaTmpl:
-						"<start_of_turn>user\n{{ if .System }}{{ .System }} {{ end }}{{ .Prompt }}<end_of_turn>\n<start_of_turn>model\n{{ .Response }}<end_of_turn>\n",
+						'{{- range $i, $_ := .Messages }}\n{{- $last := eq (len (slice $.Messages $i)) 1 }}\n{{- if or (eq .Role "user") (eq .Role "system") }}<start_of_turn>user\n{{ .Content }}<end_of_turn>\n{{ if $last }}<start_of_turn>model\n{{ end }}\n{{- else if eq .Role "assistant" }}<start_of_turn>model\n{{ .Content }}{{ if not $last }}<end_of_turn>\n{{ end }}\n{{- end }}\n{{- end }}',
 					stop: "<end_of_turn>",
 			  }
 			: undefined,