fix: change writer type (#663)

controllers/message.go

@@ -145,7 +145,7 @@ func (c *ApiController) GetMessageAnswer() {
 	fmt.Printf("Refined Question: [%s]\n", realQuestion)
 	fmt.Printf("Answer: [")
 
-	writer := &RefinedWriter{c.Ctx.ResponseWriter, *NewCleaner(6), []byte{}}
+	writer := &RefinedWriter{*c.Ctx.ResponseWriter, *NewCleaner(6), []byte{}}
 	stringBuilder := &strings.Builder{}
 	err = modelProviderObj.QueryText(realQuestion, writer, stringBuilder)
 	if err != nil {

controllers/message_writer.go

@@ -16,17 +16,18 @@ package controllers
 
 import (
 	"fmt"
-	"net/http"
 	"strings"
+
+	"github.com/astaxie/beego/context"
 )
 
 type RefinedWriter struct {
-	http.ResponseWriter
+	context.Response
 	writerCleaner Cleaner
 	buf           []byte
 }
 
-func newRefinedWriter(w http.ResponseWriter) *RefinedWriter {
+func newRefinedWriter(w context.Response) *RefinedWriter {
 	return &RefinedWriter{w, *NewCleaner(6), []byte{}}
 }