diff --git a/pkg/cmd/util/util.go b/pkg/cmd/util/util.go
index ba93cd5..2a49f04 100644
--- 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