Fix fileDownloadInfo fetching entirely the file when there is no cred… (

#634) …entials param --------- Co-authored-by: Eliott C. <[email protected]>
huggingface · Apr 18, 2024 · 19b2871 · 19b2871
1 parent 805488c
commit 19b2871
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/packages/hub/src/lib/file-download-info.ts b/packages/hub/src/lib/file-download-info.ts
@@ -48,12 +48,12 @@ export async function fileDownloadInfo(params: {
 
 	const resp = await (params.fetch ?? fetch)(url, {
 		method: "GET",
-		headers: params.credentials
-			? {
-					Authorization: `Bearer ${params.credentials.accessToken}`,
-					Range: "bytes=0-0",
-			  }
-			: {},
+		headers: {
+			...(params.credentials && {
+				Authorization: `Bearer ${params.credentials.accessToken}`,
+			}),
+			Range: "bytes=0-0",
+		},
 	});
 
 	if (resp.status === 404 && resp.headers.get("X-Error-Code") === "EntryNotFound") {
@@ -70,13 +70,14 @@ export async function fileDownloadInfo(params: {
 		throw new InvalidApiResponseFormatError("Expected ETag");
 	}
 
-	const sizeHeader = resp.headers.get("Content-Length");
+	const contentRangeHeader = resp.headers.get("content-range");
 
-	if (!sizeHeader) {
+	if (!contentRangeHeader) {
 		throw new InvalidApiResponseFormatError("Expected size information");
 	}
 
-	const size = parseInt(sizeHeader);
+	const [, parsedSize] = contentRangeHeader.split("/");
+	const size = parseInt(parsedSize);
 
 	if (isNaN(size)) {
 		throw new InvalidApiResponseFormatError("Invalid file size received");