Skip to content

GitLab

Commit 38fe1a36 authored by Michael Yang's avatar Michael Yang
Browse files

fix: trim space in modelfile fields

parent 00d06619
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
parser/parser.go
View file @ 38fe1a36
...@@ -37,10 +37,13 @@ func Parse(reader io.Reader) ([]Command, error) { ...@@ -37,10 +37,13 @@ func Parse(reader io.Reader) ([]Command, error) {
switch string(bytes.ToUpper(fields[0])) { switch string(bytes.ToUpper(fields[0])) {
case "FROM": case "FROM":
command.Name = "model" command.Name = "model"
command.Args = string(fields[1]) command.Args = string(bytes.TrimSpace(fields[1]))
// copy command for validation // copy command for validation
modelCommand = command modelCommand = command
case "LICENSE", "TEMPLATE", "SYSTEM", "PROMPT", "ADAPTER": case "ADAPTER":
command.Name = string(bytes.ToLower(fields[0]))
command.Args = string(bytes.TrimSpace(fields[1]))
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":
...@@ -50,7 +53,7 @@ func Parse(reader io.Reader) ([]Command, error) { ...@@ -50,7 +53,7 @@ func Parse(reader io.Reader) ([]Command, error) {
} }
command.Name = string(fields[0]) command.Name = string(fields[0])
command.Args = string(fields[1]) command.Args = string(bytes.TrimSpace(fields[1]))
case "EMBED": case "EMBED":
return nil, fmt.Errorf("deprecated command: EMBED is no longer supported, use the /embed API endpoint instead") return nil, fmt.Errorf("deprecated command: EMBED is no longer supported, use the /embed API endpoint instead")
default: default:
......
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