diff options
Diffstat (limited to 'models/models.go')
| -rw-r--r-- | models/models.go | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/models/models.go b/models/models.go index 36ec88f..30ec5f5 100644 --- a/models/models.go +++ b/models/models.go @@ -189,9 +189,12 @@ func (m *RoleMsg) ToText(i int) string { contentStr = strings.Join(textParts, " ") + " " } // check if already has role annotation (/completion makes them) - if !strings.HasPrefix(contentStr, m.Role+":") { - icon = fmt.Sprintf("(%d) <%s>: ", i, m.Role) - } + // in that case remove it, and then add to icon + // since icon and content are separated by \n + contentStr, _ = strings.CutPrefix(contentStr, m.Role+":") + // if !strings.HasPrefix(contentStr, m.Role+":") { + icon = fmt.Sprintf("(%d) <%s>: ", i, m.Role) + // } textMsg := fmt.Sprintf("[-:-:b]%s[-:-:-]\n%s\n", icon, contentStr) return strings.ReplaceAll(textMsg, "\n\n", "\n") } |