oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1524 from feitianbubu/pr/fix-last-resp

Browse Source 
fix: ensure last message is sent successfully
This commit is contained in:
Calcium-Ion
2025-08-08 14:55:47 +08:00
committed by GitHub
parent 7d869c9af1 cf91cf1b14
commit f8e1b084cd
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
relay/channel/openai/helper.go
View File

@@ -3,6 +3,7 @@ package openai
import ( import (
"encoding/json" "encoding/json"
"errors" "errors"
"github.com/samber/lo"
"net/http" "net/http"
"one-api/common" "one-api/common"
"one-api/dto" "one-api/dto"
@@ -186,7 +187,9 @@ func handleLastResponse(lastStreamData string, responseId *string, createAt *int
*containStreamUsage = true *containStreamUsage = true
*usage = lastStreamResponse.Usage *usage = lastStreamResponse.Usage
if !info.ShouldIncludeUsage { if !info.ShouldIncludeUsage {
*shouldSendLastResp = false *shouldSendLastResp = lo.SomeBy(lastStreamResponse.Choices, func(choice dto.ChatCompletionsStreamResponseChoice) bool {
return choice.Delta.GetContentString() != "" || choice.Delta.GetReasoningContent() != ""
})
} }
} }