Преглед изворни кода

refactor: ensure graceful closure of response body in relay responses

CaIon пре 6 месеци
родитељ
комит
726f1632b0
1 измењених фајлова са 2 додато и 0 уклоњено
  1. 2 0
      relay/channel/openai/relay_responses.go

+ 2 - 0
relay/channel/openai/relay_responses.go

@@ -59,6 +59,8 @@ func OaiResponsesStreamHandler(c *gin.Context, info *relaycommon.RelayInfo, resp
 		return nil, types.NewError(fmt.Errorf("invalid response"), types.ErrorCodeBadResponse)
 	}
 
+	defer service.CloseResponseBodyGracefully(resp)
+
 	var usage = &dto.Usage{}
 	var responseTextBuilder strings.Builder