Browse Source

fix error on accumulating final chat response

Jeffrey Morgan 1 year ago
parent
commit
045b855db9
1 changed files with 4 additions and 1 deletions
  1. 4 1
      server/routes.go

+ 4 - 1
server/routes.go

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