heimoshuiyu/openai-api-route
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

fix

Browse Source
This commit is contained in:
黑墨水鱼
2023-07-18 18:54:00 +08:00
parent e70180ea81
commit 07b86bb95b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
record.go
View File

@@ -65,8 +65,8 @@ func recordAssistantResponse(contentType string, db *gorm.DB, trackID uuid.UUID,
// stream mode // stream mode
if strings.HasPrefix(contentType, "text/event-stream") { if strings.HasPrefix(contentType, "text/event-stream") {
resp := string(body) resp := string(body)
var chunk StreamModeChunk
for _, line := range strings.Split(resp, "\n") { for _, line := range strings.Split(resp, "\n") {
chunk := StreamModeChunk{}
line = strings.TrimPrefix(line, "data:") line = strings.TrimPrefix(line, "data:")
line = strings.TrimSpace(line) line = strings.TrimSpace(line)
if line == "" { if line == "" {