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

fix: Prevent resource leaks by adding body close in stream handlers

Browse Source
This commit is contained in:
1808837298@qq.com
2025-03-05 19:51:22 +08:00
parent 37a83ecc33
commit 558e625a01
4 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
relay/channel/claude/relay-claude.go
View File

@@ -506,7 +506,7 @@ func ClaudeStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.
} }
} }
helper.Done(c) helper.Done(c)
resp.Body.Close() //resp.Body.Close()
return nil, usage return nil, usage
} }

2
relay/channel/gemini/relay-gemini.go
View File

@@ -534,7 +534,7 @@ func GeminiChatStreamHandler(c *gin.Context, resp *http.Response, info *relaycom
} }
} }
helper.Done(c) helper.Done(c)
resp.Body.Close() //resp.Body.Close()
return nil, usage return nil, usage
} }

2
relay/channel/openai/relay-openai.go
View File

@@ -249,7 +249,7 @@ func OaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.Rel
helper.Done(c) helper.Done(c)
resp.Body.Close() //resp.Body.Close()
return nil, usage return nil, usage
} }

5
relay/helper/stream_scanner.go
View File

@@ -15,7 +15,10 @@ import (
) )
func StreamScannerHandler(c *gin.Context, resp *http.Response, info *relaycommon.RelayInfo, dataHandler func(data string) bool) { func StreamScannerHandler(c *gin.Context, resp *http.Response, info *relaycommon.RelayInfo, dataHandler func(data string) bool) {
if resp == nil {
return
}
defer resp.Body.Close()
streamingTimeout := time.Duration(constant.StreamingTimeout) * time.Second streamingTimeout := time.Duration(constant.StreamingTimeout) * time.Second
if strings.HasPrefix(info.UpstreamModelName, "o1") || strings.HasPrefix(info.UpstreamModelName, "o3") { if strings.HasPrefix(info.UpstreamModelName, "o1") || strings.HasPrefix(info.UpstreamModelName, "o3") {
// twice timeout for thinking model // twice timeout for thinking model