third-party-mirrors/ollama
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update routes.go

Browse Source
This commit is contained in:
Bruce MacDonald 2023-11-20 14:21:37 -05:00
parent 3befd3dad9
commit 5690cd5ab2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/routes.go
View File

@ -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
}