Function main

packages/core/script/generate-ambient.ts:111–167

()

}

async function main() {
  const dryRun = process.argv.includes("--dry-run");

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

  const res = await fetch(API_ENDPOINT);
  if (!res.ok) {
    console.error(`Fetch failed: ${res.status} ${res.statusText}`);
    process.exit(1);
  }

  const parsed = AmbientResponse.safeParse(await res.json());
  if (!parsed.success) {
    console.error("Invalid Ambient response:", parsed.error.issues);
    process.exit(1);
  }

  const selected = parsed.data.data.filter((m) => ALLOWLIST.has(m.id));
  const missing = [...ALLOWLIST].filter(
    (id) => !selected.some((m) => m.id === id),
  );
  if (missing.length > 0) {
    console.error(`Allowlisted models missing from API: ${missing.join(", ")}`);
    process.exit(1);
  }

  let count = 0;
  for (const model of selected) {
    const baseModel = BASE_MODEL_MAP[model.id];
    if (!baseModel) {
      console.error(`No BASE_MODEL_MAP entry for ${model.id}; skipping`);
      continue;
    }
    const filePath = path.join(outDir, `${model.id}.toml`);
    const toml = formatToml(model, baseModel);
    if (dryRun) {
      console.log(`--- ${path.relative(process.cwd(), filePath)} ---`);
      console.log(toml);
    } else {
      await mkdir(path.dirname(filePath), { recursive: true });
      await Bun.write(filePath, toml);
    }
    count++;
  }

  console.log(
    `${dryRun ? "Previewed" : "Wrote"} ${count} model file(s) under providers/ambient/models/`,
  );
}