feat(templates): add sprig to multimodal templates

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
mudler · Oct 17, 2024 · 0309bfb · 0309bfb
1 parent 54c0f15
commit 0309bfb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/templates/multimodal.go b/pkg/templates/multimodal.go
@@ -3,11 +3,13 @@ package templates
 import (
 	"bytes"
 	"text/template"
+
+	"github.com/Masterminds/sprig/v3"
 )
 
 func TemplateMultiModal(templateString string, templateID int, text string) (string, error) {
 	// compile the template
-	tmpl, err := template.New("template").Parse(templateString)
+	tmpl, err := template.New("template").Funcs(sprig.FuncMap()).Parse(templateString)
 	if err != nil {
 		return "", err
 	}