From d59b164fa2a336e16bd89024959f0a79350c13bd Mon Sep 17 00:00:00 2001 From: Jeffrey Morgan Date: Thu, 20 Jul 2023 01:13:30 -0700 Subject: [PATCH] add prompt back to parser --- parser/parser.go | 2 +- server/images.go | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/parser/parser.go b/parser/parser.go index fcb50bf5..c417512d 100644 --- a/parser/parser.go +++ b/parser/parser.go @@ -39,7 +39,7 @@ func Parse(reader io.Reader) ([]Command, error) { command.Args = string(fields[1]) // copy command for validation modelCommand = command - case "LICENSE", "TEMPLATE", "SYSTEM": + case "LICENSE", "TEMPLATE", "SYSTEM", "PROMPT": command.Name = string(bytes.ToLower(fields[0])) command.Args = string(fields[1]) case "PARAMETER": diff --git a/server/images.go b/server/images.go index 19280d57..4258b201 100644 --- a/server/images.go +++ b/server/images.go @@ -250,7 +250,7 @@ func CreateModel(name string, path string, fn func(status string)) error { layers = append(layers, newLayer) } } - case "license", "template", "system": + case "license", "template", "system", "prompt": fn(fmt.Sprintf("creating %s layer", c.Name)) // remove the prompt layer if one exists mediaType := fmt.Sprintf("application/vnd.ollama.image.%s", c.Name)