rename aggregate to contents

This commit is contained in:
Michael Yang
2024-07-11 16:06:57 -07:00
parent 57ec6901eb
commit 5056bb9c01
2 changed files with 41 additions and 7 deletions

View File

@@ -103,15 +103,16 @@ var response = parse.ActionNode{
}
var funcs = template.FuncMap{
"aggregate": func(v []*api.Message, role string) string {
var aggregated []string
// contents returns the contents of messages with an optional role filter
"contents": func(v []*api.Message, role ...string) string {
var parts []string
for _, m := range v {
if m.Role == role {
aggregated = append(aggregated, m.Content)
if len(role) == 0 || role[0] == "" || m.Role == role[0] {
parts = append(parts, m.Content)
}
}
return strings.Join(aggregated, "\n\n")
return strings.Join(parts, "\n\n")
},
}