Move library collections as a group and map v2 identity refs.

Collapsed collection actions now move, rename, delete, and rerun every part so chains stay together. Identity mode follows the v2u3 Ref2VA picture wiring.

Co-authored-by: Cursor <cursoragent@cursor.com>

server/api/edit.post.ts

@@ -1,5 +1,5 @@
-import { imageComfyConfigured, uploadImageEdit, queueImagePrompt, waitForImageEdit, downloadEditedImage } from '~/server/utils/imageComfy'
-import { ensureImageComfyReady } from '~/server/utils/imageComfyLifecycle'
+import { imageComfyConfigured, uploadImageEdit, queueImagePrompt, waitForImageEdit, downloadEditedImage, purgeImageComfyArtifacts, withImageComfyHost } from '~/server/utils/imageComfy'
+import { pickImageEditHost } from '~/server/utils/imageComfyLifecycle'
 import { buildEditWorkflow } from '~/server/utils/imageWorkflow'
 import { imageDimensions } from '~/server/utils/resolution'
 
@@ -43,7 +43,7 @@ export default defineEventHandler(async (event) => {
   if (!imageComfyConfigured()) {
     throw createError({
       statusCode: 503,
-      statusMessage: 'Image edit ComfyUI is not configured yet. Set IMAGE_COMFY_HOST on this instance.'
+      statusMessage: 'Image edit ComfyUI is not configured yet. Set COMFY_HOST for Beast and IMAGE_COMFY_HOST for Sidecar overflow.'
     })
   }
 
@@ -88,7 +88,7 @@ export default defineEventHandler(async (event) => {
     seed,
     cfg
   }
-  emitJob(job, { type: 'status', message: 'Checking sidecar ComfyUI...', progress: 2 })
+  emitJob(job, { type: 'status', message: 'Choosing Beast or Sidecar...', progress: 2 })
 
   void runEdit(job, {
     image,
@@ -127,7 +127,7 @@ async function runEdit(
     denoise: number
   }
 ) {
-  await ensureImageComfyReady((status) => {
+  const target = await pickImageEditHost((status) => {
     emitJob(job, {
       type: status.state === 'busy' ? 'busy' : 'status',
       message: status.message,
@@ -135,67 +135,84 @@ async function runEdit(
       busy: status.state === 'busy'
     })
   })
+  job.imageComfyHost = target.host
   if (job.status === 'cancelled') throw new Error('Job interrupted.')
 
-  emitJob(job, { type: 'status', message: 'Uploading still to Qwen Edit...', progress: 8 })
-  const uploaded = await uploadImageEdit(params.image, job.id)
-  if (job.status === 'cancelled') throw new Error('Job interrupted.')
+  await withImageComfyHost(target.host, async () => {
+    emitJob(job, {
+      type: 'status',
+      message: `Uploading still to ${target.label}...`,
+      progress: 8
+    })
+    const uploaded = await uploadImageEdit(params.image, job.id)
+    if (job.status === 'cancelled') throw new Error('Job interrupted.')
 
-  emitJob(job, { type: 'status', message: 'Queueing Qwen Image Edit Lightning...', progress: 12 })
-  const graph = buildEditWorkflow({
-    prompt: params.prompt,
-    negative: params.negative,
-    imageName: uploaded.name,
-    steps: params.steps,
-    seed: params.seed,
-    cfg: params.cfg,
-    denoise: params.denoise,
-    filenamePrefix: `aigen_edit_${job.id.slice(0, 8)}`
-  })
-  const queued = await queueImagePrompt(graph, job.clientId)
-  job.promptId = queued.prompt_id
-  job.status = 'running'
-  emitJob(job, { type: 'status', message: 'Editing still...', progress: 18, maxStep: params.steps })
+    emitJob(job, {
+      type: 'status',
+      message: `Queueing Qwen Image Edit on ${target.label}...`,
+      progress: 12
+    })
+    const graph = buildEditWorkflow({
+      prompt: params.prompt,
+      negative: params.negative,
+      imageName: uploaded.name,
+      steps: params.steps,
+      seed: params.seed,
+      cfg: params.cfg,
+      denoise: params.denoise,
+      filenamePrefix: `aigen_edit_${job.id.slice(0, 8)}`
+    })
+    const queued = await queueImagePrompt(graph, job.clientId)
+    job.promptId = queued.prompt_id
+    job.status = 'running'
+    emitJob(job, { type: 'status', message: `Editing still on ${target.label}...`, progress: 18, maxStep: params.steps })
 
-  const output = await waitForImageEdit({
-    promptId: queued.prompt_id,
-    clientId: job.clientId,
-    onProgress: (event) => {
-      emitJob(job, {
-        type: 'status',
-        message: event.message,
-        progress: event.progress,
-        step: event.step,
-        maxStep: event.maxStep || params.steps,
-        node: event.node
-      })
-    },
-    isCancelled: () => job.status === 'cancelled'
-  })
+    const output = await waitForImageEdit({
+      promptId: queued.prompt_id,
+      clientId: job.clientId,
+      onProgress: (event) => {
+        emitJob(job, {
+          type: 'status',
+          message: event.message,
+          progress: event.progress,
+          step: event.step,
+          maxStep: event.maxStep || params.steps,
+          node: event.node
+        })
+      },
+      isCancelled: () => job.status === 'cancelled'
+    })
 
-  emitJob(job, { type: 'status', message: 'Saving edited still...', progress: 94 })
-  const buffer = await downloadEditedImage(output)
-  const size = imageDimensions(buffer)
-  const still = await saveStill({
-    ownerKey: job.library!.ownerKey,
-    folderId: job.library!.folderId,
-    filename: job.library?.name ? `${job.library.name}.png` : output.filename,
-    data: buffer,
-    width: size?.width || 0,
-    height: size?.height || 0,
-    hideInput: job.library?.hideInput
-  })
-  job.stillId = still.id
-  job.status = 'complete'
-  emitJob(job, {
-    type: 'complete',
-    message: 'Edit finished',
-    progress: 100,
-    stillId: still.id,
-    filename: output.filename,
-    subfolder: output.subfolder,
-    mediaType: 'image',
-    hideThumbnail: job.hideThumbnail,
-    folderLocked: job.library?.folderLocked
+    emitJob(job, { type: 'status', message: 'Saving edited still...', progress: 94 })
+    const buffer = await downloadEditedImage(output)
+    const size = imageDimensions(buffer)
+    const still = await saveStill({
+      ownerKey: job.library!.ownerKey,
+      folderId: job.library!.folderId,
+      filename: job.library?.name ? `${job.library.name}.png` : output.filename,
+      data: buffer,
+      width: size?.width || 0,
+      height: size?.height || 0,
+      hideInput: job.library?.hideInput
+    })
+    job.stillId = still?.id
+    await purgeImageComfyArtifacts({
+      output,
+      inputName: uploaded.name,
+      inputSubfolder: uploaded.subfolder,
+      promptId: job.promptId
+    })
+    job.status = 'complete'
+    emitJob(job, {
+      type: 'complete',
+      message: `Edit finished on ${target.label}`,
+      progress: 100,
+      stillId: still?.id,
+      filename: output.filename,
+      subfolder: output.subfolder,
+      mediaType: 'image',
+      hideThumbnail: job.hideThumbnail,
+      folderLocked: job.library?.folderLocked
+    })
   })
 }