Merge branch 'huggingface:main' into main

Author: antfin-oss

README.md

@@ -98,8 +98,8 @@ You can run our packages with vanilla JS, without any bundler, by using a CDN or
 
 ```html
 <script type="module">
-    import { InferenceClient } from 'https://cdn.jsdelivr.net/npm/@huggingface/[email protected].0/+esm';
-    import { createRepo, commit, deleteRepo, listFiles } from "https://cdn.jsdelivr.net/npm/@huggingface/[email protected].12/+esm";
+    import { InferenceClient } from 'https://cdn.jsdelivr.net/npm/@huggingface/[email protected].3/+esm';
+    import { createRepo, commit, deleteRepo, listFiles } from "https://cdn.jsdelivr.net/npm/@huggingface/[email protected].13/+esm";
 </script>
 ```
 

packages/hub/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "@huggingface/hub",
 	"packageManager": "[email protected]",
-	"version": "2.6.12",
+	"version": "2.6.13",
 	"description": "Utilities to interact with the Hugging Face hub",
 	"repository": "https://github.com/huggingface/huggingface.js.git",
 	"publishConfig": {

packages/hub/src/lib/add-collection-item.spec.ts

@@ -2,54 +2,83 @@ import { it, describe, expect } from "vitest";
 
 import { TEST_HUB_URL, TEST_ACCESS_TOKEN, TEST_USER } from "../test/consts";
 import { addCollectionItem } from "./add-collection-item";
-import { listCollections } from "./list-collections";
 import { collectionInfo } from "./collection-info";
 import { deleteCollectionItem } from "./delete-collection-item";
+import { createCollection } from "./create-collection";
+import { deleteCollection } from "./delete-collection";
 
 describe("addCollectionItem", () => {
 	it("should add a item to a collection", async () => {
 		let slug: string = "";
-		let itemId: string = "";
+
+		const randomString = crypto.randomUUID();
+		const title = `Test Collection ${randomString}`;
 
 		try {
-			for await (const entry of listCollections({
-				search: { owner: [TEST_USER] },
-				limit: 1,
+			const result = await createCollection({
+				collection: {
+					title,
+					namespace: TEST_USER,
+					description: "This is a test collection",
+					private: false,
+				},
+				accessToken: TEST_ACCESS_TOKEN,
 				hubUrl: TEST_HUB_URL,
-			})) {
-				slug = entry.slug;
-				break;
-			}
+			});
+
+			slug = result.slug;
+
+			expect(result.slug.startsWith(`${TEST_USER}/test-collection-${randomString}`)).toBe(true);
 
 			await addCollectionItem({
+				accessToken: TEST_ACCESS_TOKEN,
+				hubUrl: TEST_HUB_URL,
 				slug,
 				item: {
-					type: "model",
-					id: "quanghuynt14/TestAddCollectionItem",
+					type: "collection",
+					// temporary, later the slug should work on its own
+					id: result.slug.slice(-24),
 				},
 				note: "This is a test item",
-				accessToken: TEST_ACCESS_TOKEN,
-				hubUrl: TEST_HUB_URL,
 			});
 
-			const collection = await collectionInfo({
+			const items = await collectionInfo({
 				slug,
 				accessToken: TEST_ACCESS_TOKEN,
 				hubUrl: TEST_HUB_URL,
 			});
 
-			const item = collection.items.find((item) => item.id === "quanghuynt14/TestAddCollectionItem");
-
-			expect(item).toBeDefined();
+			expect(items.items.length).toBe(1);
+			expect(items.items[0].type).toBe("collection");
+			// temporary, later the slug should work on its own right?
+			expect(items.items[0].id).toBe(result.slug.slice(-24));
+			expect(items.items[0].note).toEqual({
+				html: "This is a test item",
+				text: "This is a test item",
+			});
 
-			itemId = item?._id || "";
-		} finally {
 			await deleteCollectionItem({
 				slug,
-				itemId,
+				itemId: items.items[0]._id,
 				accessToken: TEST_ACCESS_TOKEN,
 				hubUrl: TEST_HUB_URL,
 			});
+
+			const items2 = await collectionInfo({
+				slug,
+				accessToken: TEST_ACCESS_TOKEN,
+				hubUrl: TEST_HUB_URL,
+			});
+
+			expect(items2.items.length).toBe(0);
+		} finally {
+			if (slug) {
+				await deleteCollection({
+					slug,
+					accessToken: TEST_ACCESS_TOKEN,
+					hubUrl: TEST_HUB_URL,
+				});
+			}
 		}
 	});
 });

packages/hub/src/lib/add-collection-item.ts

@@ -27,6 +27,10 @@ export async function addCollectionItem(
 		fetch?: typeof fetch;
 	} & Partial<CredentialsParams>
 ): Promise<void> {
+	if (!params.slug) {
+		throw new TypeError("slug is required");
+	}
+
 	const accessToken = checkCredentials(params);
 
 	const res = await (params.fetch ?? fetch)(`${params.hubUrl ?? HUB_URL}/api/collections/${params.slug}/items`, {