baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

linting

Browse source
This commit is contained in:
Michael Yang 2024-04-24 16:08:51 -07:00
parent c0a00f68ae
commit 238715037d
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
parser/parser.go
View file

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