diff --git a/parser/parser.go b/parser/parser.go index edb81615b..1b80ebec2 100644 --- a/parser/parser.go +++ b/parser/parser.go @@ -6,7 +6,6 @@ import ( "errors" "fmt" "io" - "slices" "strconv" "strings" ) @@ -66,7 +65,7 @@ func Parse(r io.Reader) (cmds []Command, err error) { cmd.Name = s } case stateMessage: - if !slices.Contains([]string{"system", "user", "assistant"}, b.String()) { + if !isValidRole(b.String()) { return nil, errInvalidRole } @@ -74,8 +73,6 @@ func Parse(r io.Reader) (cmds []Command, err error) { case stateComment, stateNil: // pass case stateValue: - s := b.String() - s, ok := unquote(b.String()) if !ok || isSpace(r) { if _, err := b.WriteRune(r); err != nil {