fix(helper): improve error handling in FlushWriter and related functions

This commit is contained in:
CaIon committed 2025-12-13 13:29:21 +08:00
1 parent e0a79e853d
commit 7d586ef507
2 files changed
+42 -15

No files matched your search

+2 -2
View File
@@ -94,10 +94,10 @@ func GeminiTextGenerationStreamHandler(c *gin.Context, info *relaycommon.RelayIn
helper.SetEventStreamHeaders(c)
return geminiStreamHandler(c, info, resp, func(data string, geminiResponse *dto.GeminiChatResponse) bool {
// 直接发送 GeminiChatResponse 响应
err := helper.StringData(c, data)
if err != nil {
logger.LogError(c, err.Error())
logger.LogError(c, "failed to write stream data: "+err.Error())
return false
}
info.SendResponseCount++
return true