Michael Yang 1 năm trước cách đây
mục cha
commit
238715037d
1 tập tin đã thay đổi với 1 bổ sung4 xóa
  1. 1 4
      parser/parser.go

+ 1 - 4
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 {