import { createTrainer } from "arkor";
const trainer = createTrainer({
name: "support-bot-v1",
model: "unsloth/gemma-4-E4B-it",
dataset: { type: "huggingface", name: "arkorlab/triage-demo" },
callbacks: {
onStarted: ({ job }) => console.log("started", job.id),
onLog: ({ step, loss }) => console.log("step", step, "loss", loss),
onCheckpoint: async ({ step, infer }) => {
const r = await infer({ messages: [{ role: "user", content: "Hi" }] });
console.log("checkpoint", step, await r.text());
},
onCompleted: ({ job }) => console.log("done", job.id),
onFailed: ({ error }) => console.error("failed", error),
},
});
await trainer.start();
await trainer.wait();