import { raw, save, short } from "./lib.ts";

const list = await raw("/models");
save("00-models.json", list);
console.log("status", list.status, "headers", list.headers);
const data = (list.body as any)?.data ?? [];
console.log("count", data.length);
// compact table
for (const m of data) {
  const c = m.capabilities ?? {};
  console.log(
    [
      m.id.padEnd(40),
      String(m.max_context_length ?? "").padStart(8),
      c.completion_chat ? "chat" : "    ",
      c.completion_fim ? "fim" : "   ",
      c.function_calling ? "fn" : "  ",
      c.vision ? "vis" : "   ",
      c.classification ? "cls" : "   ",
      c.fine_tuning ? "ft" : "  ",
      m.deprecation ? `DEPR ${m.deprecation}` : "",
      m.deprecation_replacement_model ? `-> ${m.deprecation_replacement_model}` : "",
      `aliases=${JSON.stringify(m.aliases ?? [])}`,
    ].join(" "),
  );
}
// full example object
console.log("example object keys:", Object.keys(data[0] ?? {}));
console.log("capabilities keys:", Object.keys(data[0]?.capabilities ?? {}));
// retrieve one
const one = await raw("/models/mistral-large-latest");
save("00-model-large.json", one);
console.log("GET /models/mistral-large-latest", one.status, short(one.body, 800));