From 5690cd5ab20c66c9c708fbd0f3b8de773d87c311 Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Mon, 20 Nov 2023 14:21:37 -0500 Subject: [PATCH] Update routes.go --- server/routes.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server/routes.go b/server/routes.go index 8d9f1957..eabd9ad2 100644 --- a/server/routes.go +++ b/server/routes.go @@ -349,13 +349,13 @@ func promptFromMessages(model *Model, messages []api.Message) (string, error) { var prompt strings.Builder vars := &PromptVars{} for _, m := range messages { - if (m.Role == "system" || m.Role == "user") && vars.Prompt != "" { + if (m.Role == "system" || m.Role == "user") && vars.User != "" { if err := flush(vars, model, &prompt); err != nil { return "", err } } - if m.Role == "assistant" && (vars.Prompt != "" || vars.System != "") { + if m.Role == "assistant" && (vars.User != "" || vars.System != "") { if err := flush(vars, model, &prompt); err != nil { return "", err }