test-provider-snippets.js

// test-provider-snippet.mjs
import { snippets as hfInferenceSnippets } from "@huggingface/inference";

async function fetchModelData(modelId) {
  const res = await fetch(`https://huggingface.co/api/models/${modelId}`);
  if (!res.ok) throw new Error(`Model ${modelId} not found`);
  const apiModel = await res.json();
  return {
    id:           apiModel.modelId ?? modelId,
    pipeline_tag: apiModel.pipeline_tag,
    tags:         apiModel.tags ?? [],
  };
}

async function main() {
  const modelId = process.argv[2] || "gpt2";
  const modelData = await fetchModelData(modelId);

  const providerSnippet = hfInferenceSnippets
    .getInferenceSnippets(modelData, "auto")
    .find(sn => sn.language === "python" && sn.client === "huggingface_hub");

  if (!providerSnippet) {
    console.error("No matching Python/huggingface_hub snippet returned.");
    process.exit(1);
  }

  const content = Array.isArray(providerSnippet.content)
    ? providerSnippet.content.join("\n")
    : providerSnippet.content;

  console.log(`✅  snippet for ${modelId}\n`);
  console.log("──────── snippet start ────────");
  console.log(content);
  console.log("──────── snippet end ──────────");
}

main().catch(err => {
  console.error(err);
  process.exit(1);
});