fix error on accumulating final chat response

045b855d · Jeffrey Morgan · 32064a06 · 045b855d
Commit 045b855d authored Dec 10, 2023 by Jeffrey Morgan
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

server/routes.go server/routes.go +4 -1

No files found.
--- a/server/routes.go
+++ b/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 {