@hasna/models 0.0.4 → 0.0.6

package/dist/cli/index.js

@@ -2806,6 +2806,15 @@ import { pipeline } from "stream/promises";
 import { Readable } from "stream";
 import { randomUUID } from "crypto";
 var HF_ENDPOINT = process.env["HF_ENDPOINT"] || "https://huggingface.co";
+
+class HuggingFaceApiError extends Error {
+  status;
+  constructor(response, body) {
+    super(`Hugging Face request failed ${response.status} ${response.statusText}: ${body.slice(0, 300)}`);
+    this.name = "HuggingFaceApiError";
+    this.status = response.status;
+  }
+}
 function apiBase() {
   return HF_ENDPOINT.replace(/\/+$/, "");
 }
@@ -2823,7 +2832,7 @@ async function hfJson(path, init) {
   });
   if (!response.ok) {
     const text = await response.text().catch(() => "");
-    throw new Error(`Hugging Face request failed ${response.status} ${response.statusText}: ${text.slice(0, 300)}`);
+    throw new HuggingFaceApiError(response, text);
   }
   return response.json();
 }
@@ -2901,7 +2910,7 @@ function safeDestinationPath(root, filePath) {
   }
   return destination;
 }
-function normalizeEntry(raw, kind) {
+function normalizeEntry(raw, kind, fallbackRevision = "main") {
   const repoId = String(raw.id ?? raw.modelId ?? "");
   const tags = Array.isArray(raw.tags) ? raw.tags.map(String) : [];
   const cardData = raw.cardData;
@@ -2909,7 +2918,7 @@ function normalizeEntry(raw, kind) {
     provider: "huggingface",
     entityKind: kind,
     repoId,
-    revision: String(raw.sha ?? "main"),
+    revision: String(raw.sha ?? fallbackRevision),
     canonicalUrl: canonicalUrl(kind, repoId),
     title: repoId,
     author: typeof raw.author === "string" ? raw.author : repoId.split("/")[0],
@@ -2992,17 +3001,21 @@ async function searchHuggingFace(input = {}) {
 }
 async function getHuggingFaceInfo(refOrInput, defaultKind = "model") {
   const ref = typeof refOrInput === "string" ? parseProviderRef(refOrInput, defaultKind) : refOrInput;
-  const raw = await hfJson(`/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}`);
-  return normalizeEntry(raw, ref.entityKind);
+  const revision = ref.revision || "main";
+  const raw = await hfJson(`/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}/revision/${encodeURIComponent(revision)}`);
+  return normalizeEntry(raw, ref.entityKind, revision);
 }
 async function listHuggingFaceFiles(refOrInput, defaultKind = "model") {
   const ref = typeof refOrInput === "string" ? parseProviderRef(refOrInput, defaultKind) : refOrInput;
-  const treePath = `/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}/tree/${encodeURIComponent(ref.revision)}?recursive=1&expand=1`;
+  const revision = ref.revision || "main";
+  const treePath = `/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}/tree/${encodeURIComponent(revision)}?recursive=1&expand=1`;
   try {
     const raw = await hfJson(treePath);
     return raw.map((entry) => normalizeTreeFile(entry, ref)).filter((entry) => Boolean(entry));
-  } catch {
-    const info = await hfJson(`/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}`);
+  } catch (error) {
+    if (!(error instanceof HuggingFaceApiError) || error.status !== 404)
+      throw error;
+    const info = await hfJson(`/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}/revision/${encodeURIComponent(revision)}`);
     const siblings = Array.isArray(info.siblings) ? info.siblings : [];
     return siblings.map((entry) => normalizeSibling(entry, ref)).filter((entry) => Boolean(entry));
   }

package/dist/index.js

@@ -430,6 +430,15 @@ import { pipeline } from "stream/promises";
 import { Readable } from "stream";
 import { randomUUID } from "crypto";
 var HF_ENDPOINT = process.env["HF_ENDPOINT"] || "https://huggingface.co";
+
+class HuggingFaceApiError extends Error {
+  status;
+  constructor(response, body) {
+    super(`Hugging Face request failed ${response.status} ${response.statusText}: ${body.slice(0, 300)}`);
+    this.name = "HuggingFaceApiError";
+    this.status = response.status;
+  }
+}
 function apiBase() {
   return HF_ENDPOINT.replace(/\/+$/, "");
 }
@@ -447,7 +456,7 @@ async function hfJson(path, init) {
   });
   if (!response.ok) {
     const text = await response.text().catch(() => "");
-    throw new Error(`Hugging Face request failed ${response.status} ${response.statusText}: ${text.slice(0, 300)}`);
+    throw new HuggingFaceApiError(response, text);
   }
   return response.json();
 }
@@ -525,7 +534,7 @@ function safeDestinationPath(root, filePath) {
   }
   return destination;
 }
-function normalizeEntry(raw, kind) {
+function normalizeEntry(raw, kind, fallbackRevision = "main") {
   const repoId = String(raw.id ?? raw.modelId ?? "");
   const tags = Array.isArray(raw.tags) ? raw.tags.map(String) : [];
   const cardData = raw.cardData;
@@ -533,7 +542,7 @@ function normalizeEntry(raw, kind) {
     provider: "huggingface",
     entityKind: kind,
     repoId,
-    revision: String(raw.sha ?? "main"),
+    revision: String(raw.sha ?? fallbackRevision),
     canonicalUrl: canonicalUrl(kind, repoId),
     title: repoId,
     author: typeof raw.author === "string" ? raw.author : repoId.split("/")[0],
@@ -616,17 +625,21 @@ async function searchHuggingFace(input = {}) {
 }
 async function getHuggingFaceInfo(refOrInput, defaultKind = "model") {
   const ref = typeof refOrInput === "string" ? parseProviderRef(refOrInput, defaultKind) : refOrInput;
-  const raw = await hfJson(`/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}`);
-  return normalizeEntry(raw, ref.entityKind);
+  const revision = ref.revision || "main";
+  const raw = await hfJson(`/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}/revision/${encodeURIComponent(revision)}`);
+  return normalizeEntry(raw, ref.entityKind, revision);
 }
 async function listHuggingFaceFiles(refOrInput, defaultKind = "model") {
   const ref = typeof refOrInput === "string" ? parseProviderRef(refOrInput, defaultKind) : refOrInput;
-  const treePath = `/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}/tree/${encodeURIComponent(ref.revision)}?recursive=1&expand=1`;
+  const revision = ref.revision || "main";
+  const treePath = `/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}/tree/${encodeURIComponent(revision)}?recursive=1&expand=1`;
   try {
     const raw = await hfJson(treePath);
     return raw.map((entry) => normalizeTreeFile(entry, ref)).filter((entry) => Boolean(entry));
-  } catch {
-    const info = await hfJson(`/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}`);
+  } catch (error) {
+    if (!(error instanceof HuggingFaceApiError) || error.status !== 404)
+      throw error;
+    const info = await hfJson(`/api/${apiKind(ref.entityKind)}/${encodeRepoId(ref.repoId)}/revision/${encodeURIComponent(revision)}`);
     const siblings = Array.isArray(info.siblings) ? info.siblings : [];
     return siblings.map((entry) => normalizeSibling(entry, ref)).filter((entry) => Boolean(entry));
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hasna/models",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "description": "CLI-first local model and dataset lifecycle tool for open-source/open-weight catalogs",
   "type": "module",
   "license": "Apache-2.0",