Merge pull request #1393 from jmorganca/mxyng/fix-whitespace

fix: trim space in modelfile fields

Merge pull request #1393 from jmorganca/mxyng/fix-whitespace
fix: trim space in modelfile fields
b12f1b98 · Michael Yang · GitHub · 195e3d9d · 38fe1a36 · b12f1b98
Unverified Commit b12f1b98 authored Dec 05, 2023 by Michael Yang Committed by GitHub Dec 05, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

parser/parser.go parser/parser.go +6 -3

No files found.
--- 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: