baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

use missingkey in set empty interface when missing

Browse source
This commit is contained in:
Bruce MacDonald 2023-11-22 14:46:49 -05:00 committed by Michael Yang
parent 32f62fbb8e
commit 47d4e22673
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
server/images.go
View file

@ -58,18 +58,21 @@ type PromptVars struct {
func (m *Model) Prompt(p PromptVars) (string, error) { func (m *Model) Prompt(p PromptVars) (string, error) {
var prompt strings.Builder var prompt strings.Builder
tmpl, err := template.New("").Parse(m.Template) // Use the "missingkey=zero" option to handle missing variables without panicking
tmpl, err := template.New("").Option("missingkey=zero").Parse(m.Template)
if err != nil { if err != nil {
return "", err return "", err
} }
if p.System == "" { vars := map[string]any{
// use the default system prompt for this model if one is not specified "System": p.System,
p.System = m.System "Prompt": p.Prompt,
"Response": p.Response,
"First": p.First,
} }
var sb strings.Builder var sb strings.Builder
if err := tmpl.Execute(&sb, p); err != nil { if err := tmpl.Execute(&sb, vars); err != nil {
return "", err return "", err
} }
prompt.WriteString(sb.String()) prompt.WriteString(sb.String())