Update api/common.go

api/common.go

@@ -96,15 +96,16 @@ func HandleConversationResponse(c *gin.Context, resp *http.Response) (bool, stri
 			continue
 		}
 
-		if strings.HasPrefix(line, "[DONE]") {
-			if Status {
-				break
-			}
-			continue
+		if strings.HasPrefix(line, "[DONE]") ||
+			Status {
+			break
 		}
 
 		c.Writer.Write([]byte(line + "\n\n"))
 		c.Writer.Flush()
+		if strings.HasPrefix(line, "[DONE]") {
+			continue
+		}
 
 		data := line[6:]
 		var result map[string]interface{}