mirror of
https://github.com/mudler/LocalAI.git
synced 2025-01-11 23:42:48 +00:00
648ffdf449
feat(multimodal): allow to template image placeholders Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
25 lines
466 B
Go
25 lines
466 B
Go
package templates
|
|
|
|
import (
|
|
"bytes"
|
|
"text/template"
|
|
)
|
|
|
|
func TemplateMultiModal(templateString string, templateID int, text string) (string, error) {
|
|
// compile the template
|
|
tmpl, err := template.New("template").Parse(templateString)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
result := bytes.NewBuffer(nil)
|
|
// execute the template
|
|
err = tmpl.Execute(result, struct {
|
|
ID int
|
|
Text string
|
|
}{
|
|
ID: templateID,
|
|
Text: text,
|
|
})
|
|
return result.String(), err
|
|
}
|