feat(templates): add sprig to multimodal templates (#3868)

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
mudler · Oct 17, 2024 · e1db6dc · e1db6dc
1 parent d5da8c3
commit e1db6dc
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
 	}