소스 검색

refactor: ensure graceful closure of response body in relay responses

CaIon 7 달 전
부모
커밋
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