baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Better types and naming closer to style.

Browse source
This commit is contained in:
Vyacheslav Moskalev 2024-08-01 19:43:44 +07:00
parent 49a5483139
commit b0c216584c
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
server/routes.go
View file

@ -187,9 +187,9 @@ func (s *Server) GenerateHandler(c *gin.Context) {
values.Messages = append(msgs, api.Message{Role: "user", Content: req.Prompt})
}
var s string
var b bytes.Buffer
var t bytes.Buffer
if err := tmpl.Execute(&t, values); err != nil {
if err := tmpl.Execute(&b, values); err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
@ -200,11 +200,11 @@ func (s *Server) GenerateHandler(c *gin.Context) {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
b.WriteString(prev)
s += prev
}
b.WriteString(t.String());
prompt = b.String()
s += b.String();
prompt = s
}
slog.Debug("generate request", "prompt", prompt, "images", images)