import ora from "ora";
import { Gradient } from "@gradientai/nodejs-sdk";

const spinner = ora("").start();

async function run() {
  spinner.start("Creating a new Gradient client");
  const gradient = new Gradient({});
  spinner.stopAndPersist({
    text: "Gradient client created successfully",
    symbol: "✅",
  });

  spinner.start("Obtaining a list of models");
  const models = await gradient.listModels({ onlyBase: false });
  spinner.stopAndPersist({
    symbol: "✅",
    text: "Models obtained successfully",
  });

  for (const model of models) {
    const name = model.baseModelId === undefined ? model.slug : model.name;
    console.log(name);
  }
}

run()
  .catch(console.error)
  .finally(() => {
    spinner.stop();
  });