Function main

packages/core/script/generate-helicone.ts:173–233

()

}

async function main() {
  const endpoint = DEFAULT_ENDPOINT;

  const outDir = path.join(
    import.meta.dirname,
    "..",
    "..",
    "..",
    "providers",
    "helicone",
    "models",
  );

  const res = await fetch(endpoint);
  if (!res.ok) {
    console.error(`Failed to fetch registry: ${res.status} ${res.statusText}`);
    process.exit(1);
  }
  const json = await res.json();

  const parsed = HeliconeResponse.safeParse(json);
  if (!parsed.success) {
    parsed.error.cause = json;
    console.error("Invalid Helicone response:", parsed.error.errors);
    console.error("When parsing:", parsed.error.cause);
    process.exit(1);
  }

  const models = parsed.data.data.models;
  const existing = new Map<string, ExistingModel>();
  await mkdir(outDir, { recursive: true });
  for await (const file of new Bun.Glob("**/*.toml").scan({ cwd: outDir })) {
    const filePath = path.join(outDir, file);
    const model = await loadExistingModel(filePath);
    if (model !== undefined) existing.set(file, model);
  }

  // Clean output directory: remove subfolders and existing TOML files
  for (const entry of await readdir(outDir)) {
    const p = path.join(outDir, entry);
    const st = await stat(p);
    if (st.isDirectory()) {
      await rm(p, { recursive: true, force: true });
    } else if (st.isFile() && entry.endsWith(".toml")) {
      await rm(p, { force: true });
    }
  }
  let created = 0;

  for (const m of models) {
    const fileSafeId = m.id.replaceAll("/", "-");
    const filePath = path.join(outDir, `${fileSafeId}.toml`);
    const toml = formatToml(m, existing.get(`${fileSafeId}.toml`));
    await Bun.write(filePath, toml);
    created++;
  }

  console.log(