fix: try to fix panic #369

service/relay.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"errors"
 	"fmt"
 	"fmt"
 	"github.com/gin-gonic/gin"
 	"github.com/gin-gonic/gin"
+	"net/http"
 	"one-api/common"
 	"one-api/common"
 )
 )
 
 
@@ -20,10 +21,10 @@ func StringData(c *gin.Context, str string) error {
 	//str = strings.TrimPrefix(str, "data: ")
 	//str = strings.TrimPrefix(str, "data: ")
 	//str = strings.TrimSuffix(str, "\r")
 	//str = strings.TrimSuffix(str, "\r")
 	c.Render(-1, common.CustomEvent{Data: "data: " + str})
 	c.Render(-1, common.CustomEvent{Data: "data: " + str})
-	if c.Writer != nil {
-		c.Writer.Flush()
+	if flusher, ok := c.Writer.(http.Flusher); ok {
+		flusher.Flush()
 	} else {
 	} else {
-		return errors.New("writer is nil")
+		return errors.New("streaming error: flusher not found")
 	}
 	}
 	return nil
 	return nil
 }
 }