From 8acb233668dd14a42bd8c6dc9ee3e85544d29bca Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Wed, 1 May 2024 10:01:09 -0700 Subject: [PATCH] use strings.Builder --- types/model/file.go | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/types/model/file.go b/types/model/file.go index b4b7578f..c614fd32 100644 --- a/types/model/file.go +++ b/types/model/file.go @@ -29,24 +29,20 @@ type Command struct { } func (c Command) String() string { - name := c.Name - args := c.Args - + var sb strings.Builder switch c.Name { case "model": - name = "from" - args = c.Args + fmt.Fprintf(&sb, "FROM %s", c.Args) case "license", "template", "system", "adapter": - args = quote(args) + fmt.Fprintf(&sb, "%s %s", strings.ToUpper(c.Name), quote(c.Args)) case "message": role, message, _ := strings.Cut(c.Args, ": ") - args = role + " " + quote(message) + fmt.Fprintf(&sb, "MESSAGE %s %s", role, quote(message)) default: - name = "parameter" - args = c.Name + " " + quote(c.Args) + fmt.Fprintf(&sb, "PARAMETER %s %s", c.Name, quote(c.Args)) } - return fmt.Sprintf("%s %s", strings.ToUpper(name), args) + return sb.String() } type state int