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

add prompt back to parser

Browse source
This commit is contained in:
Jeffrey Morgan 2023-07-20 01:13:30 -07:00
parent 55b5f5dc34
commit d59b164fa2
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
parser/parser.go
View file

@ -39,7 +39,7 @@ func Parse(reader io.Reader) ([]Command, error) {
command.Args = string(fields[1]) command.Args = string(fields[1])
// copy command for validation // copy command for validation
modelCommand = command modelCommand = command
case "LICENSE", "TEMPLATE", "SYSTEM": case "LICENSE", "TEMPLATE", "SYSTEM", "PROMPT":
command.Name = string(bytes.ToLower(fields[0])) command.Name = string(bytes.ToLower(fields[0]))
command.Args = string(fields[1]) command.Args = string(fields[1])
case "PARAMETER": case "PARAMETER":

2
server/images.go
View file

@ -250,7 +250,7 @@ func CreateModel(name string, path string, fn func(status string)) error {
layers = append(layers, newLayer) layers = append(layers, newLayer)
} }
} }
case "license", "template", "system": case "license", "template", "system", "prompt":
fn(fmt.Sprintf("creating %s layer", c.Name)) fn(fmt.Sprintf("creating %s layer", c.Name))
// remove the prompt layer if one exists // remove the prompt layer if one exists
mediaType := fmt.Sprintf("application/vnd.ollama.image.%s", c.Name) mediaType := fmt.Sprintf("application/vnd.ollama.image.%s", c.Name)