import { rawGet, save, KEY } from "./lib.ts";

const models = await rawGet("/models");
save("00-models", models);
console.log("GET /models", models.status, models.headers);
for (const m of (models.body as any).data ?? []) console.log(JSON.stringify(m));

for (const id of ["kimi-k3", "kimi-k2.6"]) {
  const r = await rawGet(`/models/${id}`);
  save(`00-model-${id}`, r);
  console.log(`GET /models/${id}`, r.status, JSON.stringify(r.body).slice(0, 400));
}

const bal = await rawGet("/users/me/balance");
save("00-balance", bal);
console.log("GET /users/me/balance", bal.status, JSON.stringify(bal.body));

const est = await fetch("https://api.moonshot.ai/v1/tokenizers/estimate-token-count", {
  method: "POST", headers: { "content-type": "application/json", authorization: `Bearer ${KEY}` },
  body: JSON.stringify({ model: "kimi-k3", messages: [{ role: "user", content: "Hello, how many tokens is this?" }] }),
});
const estBody = await est.text();
save("00-estimate", { status: est.status, body: estBody });
console.log("POST /tokenizers/estimate-token-count", est.status, estBody.slice(0, 300));