Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: shouldn't close c.Request.Body too soon (close #35)

Browse Source
This commit is contained in:
JustSong 2023-04-29 14:49:10 +08:00
parent d93cb8f645
commit 7c66fc6c21
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
controller/relay.go
View File

@ -94,10 +94,6 @@ func relayHelper(c *gin.Context) error {
if err != nil { if err != nil {
return err return err
} }
err = c.Request.Body.Close()
if err != nil {
return err
}
req.Header.Set("Authorization", c.Request.Header.Get("Authorization")) req.Header.Set("Authorization", c.Request.Header.Get("Authorization"))
req.Header.Set("Content-Type", c.Request.Header.Get("Content-Type")) req.Header.Set("Content-Type", c.Request.Header.Get("Content-Type"))
req.Header.Set("Accept", c.Request.Header.Get("Accept")) req.Header.Set("Accept", c.Request.Header.Get("Accept"))
@ -111,7 +107,10 @@ func relayHelper(c *gin.Context) error {
if err != nil { if err != nil {
return err return err
} }
err = c.Request.Body.Close()
if err != nil {
return err
}
var textResponse TextResponse var textResponse TextResponse
isStream := resp.Header.Get("Content-Type") == "text/event-stream" isStream := resp.Header.Get("Content-Type") == "text/event-stream"
var streamResponseText string var streamResponseText string