diff --git a/parser/parser.go b/parser/parser.go index 6832351fb..a429e4672 100644 --- a/parser/parser.go +++ b/parser/parser.go @@ -557,7 +557,7 @@ func isNewline(r rune) bool { } func isValidMessageRole(role string) bool { - return role == "system" || role == "user" || role == "assistant" + return role == "system" || role == "user" || role == "assistant" || role == "control" } func isValidCommand(cmd string) bool {