1import { NovitaSDK, TaskStatus } from "novita-sdk";
2
3const novitaClient = new NovitaSDK("your_api_key");
4const params = {
5  request: {
6    model_name: "",
7    prompt: "",
8    negative_prompt: "",
9    width: 512,
10    height: 512,
11    sampler_name: "DPM++ 2S a Karras",
12    guidance_scale: 7.5,
13    steps: 20,
14    image_num: 4,
15    clip_skip: 1,
16    seed: -1,
17    loras: [],
18  },
19};
20novitaClient.txt2ImgV3(params)
21  .then((res) => {
22    if (res && res.task_id) {
23      const timer = setInterval(() => {
24        novitaClient.progress({
25          task_id: res.task_id,
26        })
27          .then((progressRes) => {
28            if (progressRes.task.status === TaskStatus.SUCCEED) {
29              console.log("finished!", progressRes.images);
30              clearInterval(timer);
31            }
32            if (progressRes.task.status === TaskStatus.FAILED) {
33              console.warn("failed!", progressRes.task.reason);
34              clearInterval(timer);
35            }
36            if (progressRes.task.status === TaskStatus.QUEUED) {
37              console.log("queueing");
38            }
39          })
40          .catch((err) => {
41            console.error("progress error:", err);
42          })
43      }, 1000);
44    }
45  })
46  .catch((err) => {
47    console.error("txt2Img error:", err);
48  })