add repoExists function

Author: coyotte508

packages/hub/src/lib/index.ts

@@ -23,6 +23,7 @@ export * from "./oauth-handle-redirect";
 export * from "./oauth-login-url";
 export * from "./parse-safetensors-metadata";
 export * from "./paths-info";
+export * from "./repo-exists";
 export * from "./snapshot-download";
 export * from "./space-info";
 export * from "./upload-file";

packages/hub/src/lib/repo-exists.spec.ts

@@ -0,0 +1,13 @@
+import { describe, expect, it } from "vitest";
+import { repoExists } from "./repo-exists";
+
+describe("repoExists", () => {
+	it("should check if a repo exists", async () => {
+		const exists1 = await repoExists({ repo: { type: "model", name: "openai-community/gpt2" } });
+
+		expect(exists1).toBe(true);
+
+		const exists2 = await repoExists({ repo: { type: "model", name: "openai-community/gpt9000" } });
+		expect(exists2).toBe(false);
+	});
+});

packages/hub/src/lib/repo-exists.ts

@@ -0,0 +1,43 @@
+import { HUB_URL } from "../consts";
+import { createApiError } from "../error";
+import type { RepoDesignation } from "../types/public";
+import { toRepoId } from "../utils/toRepoId";
+
+export async function repoExists(params: {
+	repo: RepoDesignation;
+
+	hubUrl?: string;
+	/**
+	 * An optional Git revision id which can be a branch name, a tag, or a commit hash.
+	 */
+	revision?: string;
+	/**
+	 * Custom fetch function to use instead of the default one, for example to use a proxy or edit headers.
+	 */
+	fetch?: typeof fetch;
+	accessToken?: string;
+}): Promise<boolean> {
+	const repoId = toRepoId(params.repo);
+
+	const res = await (params.fetch ?? fetch)(
+		`${params.hubUrl ?? HUB_URL}/api/${repoId.type}s/${repoId.name}?expand[]=likes`,
+		{
+			method: "GET",
+			headers: {
+				...(params.accessToken && {
+					Authorization: `Bearer ${params.accessToken}`,
+				}),
+			},
+		}
+	);
+
+	if (res.status === 404 || res.status === 401) {
+		return false;
+	}
+
+	if (!res.ok) {
+		throw await createApiError(res);
+	}
+
+	return true;
+}