Explorar el Código

refactor: ensure graceful closure of response body in relay responses

CaIon hace 6 meses
padre
commit
726f1632b0
Se han modificado 1 ficheros con 2 adiciones y 0 borrados
  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