From 05d6a889637599b2117d7a9a1d4d47441ea8451c Mon Sep 17 00:00:00 2001
From: synw <synwe@yahoo.com>
Date: Thu, 16 Jan 2025 16:49:44 +0100
Subject: [PATCH] Fix in go package

---
 codegen/golang/gen.go      |  4 +++-
 codegen/ts/gen.go          |  4 +++-
 go/templates/lmtemplate.go | 12 +++++-------
 3 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/codegen/golang/gen.go b/codegen/golang/gen.go
index fb9d386..db65539 100644
--- a/codegen/golang/gen.go
+++ b/codegen/golang/gen.go
@@ -1,7 +1,9 @@
 package golang
 
 func Gen(content string) string {
-	codeStart := `package templates
+	codeStart := `// Autogenerated code: do not edit
+
+package templates
 
 var templates = `
 	return codeStart + "`" + content + "`"
diff --git a/codegen/ts/gen.go b/codegen/ts/gen.go
index 02a77d0..7d02409 100644
--- a/codegen/ts/gen.go
+++ b/codegen/ts/gen.go
@@ -1,7 +1,9 @@
 package ts
 
 func Gen(content string) string {
-	codeStart := `import { LmTemplate } from "./interfaces.js";
+	codeStart := `// Autogenerated code: do not edit
+
+import { LmTemplate } from "./interfaces.js";
 
 const templates: Record<string, LmTemplate> = `
 	codeEnd := `;
diff --git a/go/templates/lmtemplate.go b/go/templates/lmtemplate.go
index a353d46..50f4216 100644
--- a/go/templates/lmtemplate.go
+++ b/go/templates/lmtemplate.go
@@ -138,17 +138,15 @@ func (tpl *PromptTemplate) _buildUserBlock(msg *string) string {
 func (tpl *PromptTemplate) _buildAssistantBlock(msg *string, isShot bool) string {
 	var buf []string
 	assistantBlock := tpl.Assistant
-	if (tpl._extraAssistant != "") && !isShot {
-		assistantBlock += tpl._extraAssistant
+	buf = append(buf, assistantBlock)
+	if tpl.Linebreaks != nil && tpl.Linebreaks.Assistant != nil {
+		buf = append(buf, strings.Repeat("\n", *tpl.Linebreaks.Assistant))
 	}
 	if msg != nil {
-		buf = append(buf, assistantBlock)
 		buf = append(buf, *msg)
-	} else {
-		buf = append(buf, assistantBlock)
 	}
-	if tpl.Linebreaks != nil && tpl.Linebreaks.Assistant != nil {
-		buf = append(buf, strings.Repeat("\n", *tpl.Linebreaks.Assistant))
+	if (tpl._extraAssistant != "") && !isShot {
+		buf = append(buf, tpl._extraAssistant)
 	}
 	return strings.Join(buf, "")
 }