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

fix error on accumulating final chat response

Browse source
This commit is contained in:
Jeffrey Morgan 2023-12-10 11:24:39 -05:00
parent 32064a0646
commit 045b855db9
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
server/routes.go
View file

@ -1025,7 +1025,10 @@ func ChatHandler(c *gin.Context) {
for resp := range ch { for resp := range ch {
switch r := resp.(type) { switch r := resp.(type) {
case api.ChatResponse: case api.ChatResponse:
if r.Message != nil {
sb.WriteString(r.Message.Content) sb.WriteString(r.Message.Content)
}
final = r final = r
case gin.H: case gin.H:
if errorMsg, ok := r["error"].(string); ok { if errorMsg, ok := r["error"].(string); ok {