From 38fe1a368bc96b127c02ca40634282a75ada58b2 Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Tue, 5 Dec 2023 11:57:21 -0800 Subject: [PATCH] fix: trim space in modelfile fields --- parser/parser.go | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/parser/parser.go b/parser/parser.go index 9355779e..19d32c29 100644 --- a/parser/parser.go +++ b/parser/parser.go @@ -37,10 +37,13 @@ func Parse(reader io.Reader) ([]Command, error) { switch string(bytes.ToUpper(fields[0])) { case "FROM": command.Name = "model" - command.Args = string(fields[1]) + command.Args = string(bytes.TrimSpace(fields[1])) // copy command for validation 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.Args = string(fields[1]) case "PARAMETER": @@ -50,7 +53,7 @@ func Parse(reader io.Reader) ([]Command, error) { } command.Name = string(fields[0]) - command.Args = string(fields[1]) + command.Args = string(bytes.TrimSpace(fields[1])) case "EMBED": return nil, fmt.Errorf("deprecated command: EMBED is no longer supported, use the /embed API endpoint instead") default: