zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-12 06:35:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(go-models): increase JieKouAI SSE scanner buffer (#15737)

Browse Source 
## Summary
- Raise the JieKouAI streaming SSE scanner buffer to handle larger data
chunks without truncation.
This commit is contained in:
oktofeesh
2026-06-07 22:10:10 -07:00
committed by GitHub
parent 5e25e2600b
commit b1a2210d06
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
internal/entity/models/jiekouai.go
View File

@@ -292,6 +292,7 @@ func (j *JieKouAIModel) ChatStreamlyWithSender(modelName string, messages []Mess
// SSE parsing: read line by line
scanner := bufio.NewScanner(resp.Body)
scanner.Buffer(make([]byte, 64*1024), 1024*1024)
for scanner.Scan() {
line := scanner.Text()

1
internal/entity/models/sse_scanner_buffer_test.go
View File

@@ -59,6 +59,7 @@ func TestChatStreamLargeChunkNotTruncated(t *testing.T) {
{"lmstudio", build(func(b map[string]string, s URLSuffix) chatStreamer { return NewLmStudioModel(b, s) })},
{"gitee", build(func(b map[string]string, s URLSuffix) chatStreamer { return NewGiteeModel(b, s) })},
{"tokenhub", build(func(b map[string]string, s URLSuffix) chatStreamer { return NewTokenHubModel(b, s) })},
{"jiekouai", build(func(b map[string]string, s URLSuffix) chatStreamer { return NewJieKouAIModel(b, s) })},
}
for _, tc := range cases {