From 706aa7233bc8c042046c0841db355ac029db399f Mon Sep 17 00:00:00 2001
From: QistChan <mail@chenyuqi.cn>
Date: Mon, 24 Feb 2025 01:59:54 +0800
Subject: [PATCH] fix: resolve "bufio.Scanner: token too long" error by
 increasing buffer size

---
 relay/adaptor/openai/main.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/relay/adaptor/openai/main.go b/relay/adaptor/openai/main.go
index 97080738..8eb3da96 100644
--- a/relay/adaptor/openai/main.go
+++ b/relay/adaptor/openai/main.go
@@ -27,6 +27,8 @@ const (
 func StreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*model.ErrorWithStatusCode, string, *model.Usage) {
 	responseText := ""
 	scanner := bufio.NewScanner(resp.Body)
+	buffer := make([]byte, 256*1024)
+	scanner.Buffer(buffer, len(buffer))
 	scanner.Split(bufio.ScanLines)
 	var usage *model.Usage