SDK de Comfy: clientes de Python y TypeScript para ComfyUI

pip install comfy-sdk

npm i @comfyorg/sdk

from comfy_sdk import Comfy

# Comfy Cloud
client = Comfy(api_key="comfyui-...")

wf = client.workflows.from_file("workflow_api.json")

asset = client.assets.from_file("photo.png")
wf.set_input("10", "image", asset)

job = client.run(wf)
for output in job.get_outputs("9"):
    output.to_file(output.name)

import { Comfy } from "@comfyorg/sdk";

// Comfy Cloud
const client = new Comfy({ apiKey: "comfyui-..." });

const wf = await client.workflows.fromFile("workflow_api.json");

const asset = client.assets.fromFile("photo.png");
wf.setInput("10", "image", asset);

const job = await client.run(wf);
await job.getOutputs("9")[0].toFile("out.png");

client = Comfy(api_key="comfyui-...")

const client = new Comfy({ apiKey: "comfyui-..." });

import uuid
from comfy_sdk import IdempotencyKeyReuse

# Ambos valores pertenecen a tu propio almacén, indexados por lo que la solicitud
# signifique para ti. `key` se genera una sola vez, en el primer intento; `job_id`
# se rellena en cuanto un envío devuelve.
state = load_submission_state()  # {"key": ..., "job_id": ...} or {}
key = state.get("key") or str(uuid.uuid4())
save_submission_state(key=key)

try:
    job = client.submit(wf, idempotency_key=key)
    save_submission_state(key=key, job_id=job.id)
except IdempotencyKeyReuse:
    # Un intento anterior con esta clave ya llegó al envío, así que puede
    # existir una tarea. No reenvíes: ve a buscarla.
    job_id = state.get("job_id")
    if job_id is None:
        raise  # no hay nada registrado que buscar; esto necesita a una persona
    job = client.jobs.get(job_id)

import { IdempotencyKeyReuse } from "@comfyorg/sdk";

// Ambos valores pertenecen a tu propio almacén, indexados por lo que la solicitud
// signifique para ti. `key` se genera una sola vez, en el primer intento; `jobId`
// se rellena en cuanto un envío devuelve.
const state = await loadSubmissionState(); // { key?, jobId? }
const key = state.key ?? crypto.randomUUID();
await saveSubmissionState({ key });

let job;
try {
  job = await client.submit(wf, { idempotencyKey: key });
  await saveSubmissionState({ key, jobId: job.id });
} catch (err) {
  // Un intento anterior con esta clave ya llegó al envío, así que puede
  // existir una tarea. No reenvíes: ve a buscarla.
  if (!(err instanceof IdempotencyKeyReuse) || !state.jobId) throw err;
  job = await client.jobs.get(state.jobId);
}

from comfy_sdk import Progress, Preview, OutputReady, StatusChange

job = client.submit(wf)

for event in job.events():
    match event:
        case Progress() as p:
            print(f"{p.value:.0%} {p.message}")
        case Preview() as pv:
            image = pv.to_pil()
        case OutputReady() as o:
            o.output.to_file(f"partial/{o.output.name}")
        case StatusChange(status="succeeded"):
            break

result = job.result()

const job = await client.submit(wf);

// La etiqueta es requerida: un `break` sin etiqueta dentro de un switch sale del switch,
// no del bucle.
eventLoop: for await (const event of job.events()) {
  switch (event.kind) {
    case "progress":
      console.log(event.value);
      break;
    case "outputReady":
      await event.output.toFile(`${event.output.name}`);
      break;
    case "statusChange":
      if (event.status === "succeeded") break eventLoop;
  }
}

output = job.outputs[0]
output.job_id          # la tarea que produjo este archivo

const output = job.outputs[0];
output.jobId; // la tarea que produjo este archivo

wf = job.get_workflow()

if wf.format == "save":
    ...  # el flujo de trabajo tal como se creó, con el diseño del lienzo y los nodos Note intactos
else:
    ...  # el gráfico ejecutado en formato API

const wf = await job.getWorkflow();

if (wf.format === "save") {
  // el flujo de trabajo tal como se creó, con el diseño del lienzo y los nodos Note intactos
} else {
  // el gráfico ejecutado en formato API
}

job = client.submit(wf, metadata={"client": "acme", "project": "spring"})
job.metadata  # {"client": "acme", "project": "spring"}

const job = await client.submit(wf, {
  metadata: { client: "acme", project: "spring" },
});
job.metadata; // { client: "acme", project: "spring" }

for job in client.list_jobs(metadata={"client": "acme"}):
    print(job.id, job.status)

for await (const job of client.listJobs({ metadata: { client: "acme" } })) {
  console.log(job.id, job.status);
}