🐛 Fix listFiles pagination

coyotte508 · coyotte508 · commit fa3ad620d020 · 2023-03-02T00:30:56.000+01:00
The root cause of the problem is in the reponse by HF's api, it will be fixed soon
diff --git a/packages/hub/package.json b/packages/hub/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "@huggingface/hub",
-	"version": "0.1.0",
+	"version": "0.1.1",
 	"description": "Utilities to interact with the Hugging Face hub",
 	"repository": "https://github.com/huggingface/huggingface.js.git",
 	"publishConfig": {
diff --git a/packages/hub/src/lib/list-files.ts b/packages/hub/src/lib/list-files.ts
@@ -46,6 +46,10 @@ export async function* listFiles(params: {
 			yield item;
 		}
 
-		url = json.nextUrl;
+		url = json.nextUrl
+			? json.nextUrl?.startsWith(params.hubUrl ?? HUB_URL)
+				? json.nextUrl
+				: `${params.hubUrl ?? HUB_URL}${json.nextUrl}`
+			: undefined;
 	}
 }

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "@huggingface/hub",`
`3`		`- "version": "0.1.0",`
	`3`	`+ "version": "0.1.1",`
`4`	`4`	`"description": "Utilities to interact with the Hugging Face hub",`
`5`	`5`	`"repository": "https://github.com/huggingface/huggingface.js.git",`
`6`	`6`	`"publishConfig": {`
Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,10 @@ export async function* listFiles(params: {`
`46`	`46`	`yield item;`
`47`	`47`	`}`
`48`	`48`
`49`		`- url = json.nextUrl;`
	`49`	`+ url = json.nextUrl`
	`50`	`+ ? json.nextUrl?.startsWith(params.hubUrl ?? HUB_URL)`
	`51`	`+ ? json.nextUrl`
	`52`	+ : `${params.hubUrl ?? HUB_URL}${json.nextUrl}`
	`53`	`+ : undefined;`
`50`	`54`	`}`
`51`	`55`	`}`