Skip to content

GitLab

Commit d59b164f authored by Jeffrey Morgan's avatar Jeffrey Morgan
Browse files

add prompt back to parser

parent 55b5f5dc
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
parser/parser.go
View file @ d59b164f
...@@ -39,7 +39,7 @@ func Parse(reader io.Reader) ([]Command, error) { ...@@ -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":
......
server/images.go
View file @ d59b164f
...@@ -250,7 +250,7 @@ func CreateModel(name string, path string, fn func(status string)) error { ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment