Only include user and assistant messages when generating titles

(again)
2024-06-23 18:54:45 +00:00 · 2024-06-23 18:54:45 +00:00 · cea5118cac
commit cea5118cac
parent a43a91c6ff
1 changed files with 4 additions and 1 deletions
--- a/pkg/cmd/util/util.go
+++ b/pkg/cmd/util/util.go
@ -172,7 +172,10 @@ Example response:

 	var msgs []msg
 	for _, m := range messages {
-		msgs = append(msgs, msg{string(m.Role), m.Content})
+		switch m.Role {
+			case api.MessageRoleAssistant, api.MessageRoleUser:
+				msgs = append(msgs, msg{string(m.Role), m.Content})
+		}
 	}

 	// Serialize the conversation to JSON