Parcourir la source

refactor: ensure graceful closure of response body in relay responses

CaIon il y a 6 mois
Parent
commit
726f1632b0
1 fichiers modifiés avec 2 ajouts et 0 suppressions
  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