Commit c4ca4af8 by 1808837298@qq.com

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

parent d6472145
......@@ -506,7 +506,7 @@ func ClaudeStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.
}
}
helper.Done(c)
resp.Body.Close()
//resp.Body.Close()
return nil, usage
}
......
......@@ -534,7 +534,7 @@ func GeminiChatStreamHandler(c *gin.Context, resp *http.Response, info *relaycom
}
}
helper.Done(c)
resp.Body.Close()
//resp.Body.Close()
return nil, usage
}
......
......@@ -249,7 +249,7 @@ func OaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.Rel
helper.Done(c)
resp.Body.Close()
//resp.Body.Close()
return nil, usage
}
......
......@@ -15,7 +15,10 @@ import (
)
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
if strings.HasPrefix(info.UpstreamModelName, "o1") || strings.HasPrefix(info.UpstreamModelName, "o3") {
// twice timeout for thinking model
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment