ima2-gen 1.1.5 → 1.1.7

package/routes/edit.js

@@ -4,11 +4,10 @@ import { randomBytes } from "crypto";
 import { editViaOAuth } from "../lib/oauthProxy.js";
 import { classifyUpstreamError } from "../lib/errorClassify.js";
 import { normalizeOAuthParams } from "../lib/oauthNormalize.js";
-import { normalizeImageModel } from "../lib/imageModels.js";
-import { getStyleSheet } from "../lib/sessionStore.js";
-import { renderStyleSheetPrefix } from "../lib/styleSheet.js";
+import { normalizeImageModel, normalizeReasoningEffort } from "../lib/imageModels.js";
 import { startJob, finishJob } from "../lib/inflight.js";
 import { logEvent, logError } from "../lib/logger.js";
+import { hasPngAlphaChannel, parsePngInfo } from "../lib/pngInfo.js";
 
 function validateModeration(ctx, moderation) {
   if (typeof moderation !== "string" || !ctx.config.oauth.validModeration.has(moderation)) {
@@ -17,6 +16,49 @@ function validateModeration(ctx, moderation) {
   return { moderation };
 }
 
+const MAX_EDIT_MASK_BYTES = 16 * 1024 * 1024;
+const BASE64_RE = /^[A-Za-z0-9+/]+={0,2}$/;
+
+function stripPngDataUrl(value) {
+  if (typeof value !== "string") return "";
+  return value.replace(/^data:image\/png;base64,/, "");
+}
+
+function decodePngDataUrl(value, invalidCode, pngCode) {
+  const b64 = stripPngDataUrl(value).replace(/\s+/g, "");
+  if (!b64 || b64.length % 4 !== 0 || !BASE64_RE.test(b64)) {
+    return { error: "image must be valid base64", code: invalidCode };
+  }
+  const buffer = Buffer.from(b64, "base64");
+  if (buffer.length === 0 || buffer.toString("base64").replace(/=+$/, "") !== b64.replace(/=+$/, "")) {
+    return { error: "image must be valid base64", code: invalidCode };
+  }
+  const info = parsePngInfo(buffer);
+  if (info.error) return { error: "image must be a PNG image", code: pngCode };
+  return { b64, buffer, info };
+}
+
+function validateEditMask(imageB64, mask) {
+  if (mask == null) return { mask: null, maskBytes: 0 };
+  if (typeof mask !== "string" || mask.length === 0) {
+    return { error: "mask must be a PNG data URL or base64 string", code: "INVALID_EDIT_MASK" };
+  }
+  const maskCheck = decodePngDataUrl(mask, "INVALID_EDIT_MASK_BASE64", "INVALID_EDIT_MASK_PNG");
+  if (maskCheck.error) return maskCheck;
+  if (maskCheck.buffer.length > MAX_EDIT_MASK_BYTES) {
+    return { error: "mask is too large", code: "EDIT_MASK_TOO_LARGE" };
+  }
+  if (!hasPngAlphaChannel(maskCheck.info)) {
+    return { error: "mask PNG must include an alpha channel", code: "EDIT_MASK_NO_ALPHA" };
+  }
+  const imageCheck = decodePngDataUrl(imageB64, "INVALID_EDIT_IMAGE_BASE64", "INVALID_EDIT_IMAGE_PNG");
+  if (imageCheck.error) return imageCheck;
+  if (imageCheck.info.width !== maskCheck.info.width || imageCheck.info.height !== maskCheck.info.height) {
+    return { error: "mask dimensions must match image dimensions", code: "EDIT_MASK_DIMENSION_MISMATCH" };
+  }
+  return { mask: maskCheck.b64, maskBytes: maskCheck.buffer.length };
+}
+
 export function registerEditRoutes(app, ctx) {
   app.post("/api/edit", async (req, res) => {
     const requestId = typeof req.body?.requestId === "string" ? req.body.requestId : req.id;
@@ -28,12 +70,15 @@ export function registerEditRoutes(app, ctx) {
       const {
         prompt,
         image: imageB64,
+        mask: rawMask,
         quality: rawQuality = "medium",
         size = "1024x1024",
         moderation = "low",
         provider = "oauth",
         mode: promptMode = "auto",
         model: rawModel,
+        reasoningEffort: rawReasoningEffort,
+        webSearchEnabled: rawWebSearchEnabled = true,
       } = req.body;
       const { quality, warnings: qualityWarnings } = normalizeOAuthParams({ provider, quality: rawQuality });
       const modelCheck = normalizeImageModel(ctx, rawModel);
@@ -44,6 +89,15 @@ export function registerEditRoutes(app, ctx) {
         return res.status(modelCheck.status).json({ error: modelCheck.error, code: modelCheck.code });
       }
       const imageModel = modelCheck.model;
+      const reasoningCheck = normalizeReasoningEffort(ctx, rawReasoningEffort);
+      if (reasoningCheck.error) {
+        finishStatus = "error";
+        finishHttpStatus = reasoningCheck.status;
+        finishErrorCode = reasoningCheck.code;
+        return res.status(reasoningCheck.status).json({ error: reasoningCheck.error, code: reasoningCheck.code });
+      }
+      const reasoningEffort = reasoningCheck.effort;
+      const webSearchEnabled = rawWebSearchEnabled !== false;
       const sessionId = typeof req.body?.sessionId === "string" ? req.body.sessionId : null;
       const normalizedPromptMode = promptMode === "direct" ? "direct" : "auto";
 
@@ -57,7 +111,6 @@ export function registerEditRoutes(app, ctx) {
           quality,
           model: imageModel,
           size,
-          styleSheetApplied: false,
         },
       });
 
@@ -67,6 +120,13 @@ export function registerEditRoutes(app, ctx) {
         finishErrorCode = "INVALID_EDIT_INPUT";
         return res.status(400).json({ error: "Prompt and image are required" });
       }
+      const maskCheck = validateEditMask(imageB64, rawMask);
+      if (maskCheck.error) {
+        finishStatus = "error";
+        finishHttpStatus = 400;
+        finishErrorCode = maskCheck.code;
+        return res.status(400).json({ error: maskCheck.error, code: maskCheck.code });
+      }
       const moderationCheck = validateModeration(ctx, moderation);
       if (moderationCheck.error) {
         finishStatus = "error";
@@ -81,21 +141,6 @@ export function registerEditRoutes(app, ctx) {
         return res.status(403).json({ error: "API key provider is disabled. Use OAuth (Codex login).", code: "APIKEY_DISABLED" });
       }
 
-      let effectivePrompt = prompt;
-      let styleSheetApplied = null;
-      if (sessionId) {
-        try {
-          const data = getStyleSheet(sessionId);
-          if (data && data.enabled && data.styleSheet) {
-            const prefix = renderStyleSheetPrefix(data.styleSheet);
-            if (prefix) {
-              effectivePrompt = `${prefix} ${prompt}`.slice(0, 4000);
-              styleSheetApplied = data.styleSheet;
-            }
-          }
-        } catch {}
-      }
-
       logEvent("edit", "request", {
         requestId,
         client: req.get("x-ima2-client") || "ui",
@@ -107,12 +152,14 @@ export function registerEditRoutes(app, ctx) {
         sessionId,
         promptChars: typeof prompt === "string" ? prompt.length : 0,
         promptMode: normalizedPromptMode,
-        styleSheetApplied: !!styleSheetApplied,
+        webSearchEnabled,
         inputImageChars: typeof imageB64 === "string" ? imageB64.length : 0,
+        maskPresent: Boolean(maskCheck.mask),
+        maskBytes: maskCheck.maskBytes ?? 0,
       });
       const startTime = Date.now();
-      const { b64: resultB64, usage, revisedPrompt } = await editViaOAuth(
-        effectivePrompt,
+      const { b64: resultB64, usage, revisedPrompt, webSearchCalls = 0 } = await editViaOAuth(
+        prompt,
         imageB64,
         quality,
         size,
@@ -120,7 +167,7 @@ export function registerEditRoutes(app, ctx) {
         normalizedPromptMode,
         ctx,
         requestId,
-        { model: imageModel },
+        { model: imageModel, reasoningEffort, webSearchEnabled, mask: maskCheck.mask },
       );
 
       const elapsed = ((Date.now() - startTime) / 1000).toFixed(1);
@@ -132,8 +179,6 @@ export function registerEditRoutes(app, ctx) {
         userPrompt: prompt,
         revisedPrompt: revisedPrompt || null,
         promptMode: normalizedPromptMode,
-        effectivePrompt: styleSheetApplied ? effectivePrompt : undefined,
-        styleSheetApplied: styleSheetApplied || undefined,
         quality,
         size,
         moderation,
@@ -143,7 +188,8 @@ export function registerEditRoutes(app, ctx) {
         kind: "edit",
         createdAt: Date.now(),
         usage: usage || null,
-        webSearchCalls: 0,
+        webSearchCalls,
+        webSearchEnabled,
       };
       await writeFile(join(ctx.config.storage.generatedDir, filename + ".json"), JSON.stringify(meta)).catch(() => {});
       finishHttpStatus = 200;
@@ -166,6 +212,8 @@ export function registerEditRoutes(app, ctx) {
         warnings: qualityWarnings,
         revisedPrompt: revisedPrompt || null,
         promptMode: normalizedPromptMode,
+        webSearchCalls,
+        webSearchEnabled,
       });
     } catch (err) {
       const fallbackCode = err.code || classifyUpstreamError(err.message);

package/routes/generate.js

@@ -4,12 +4,10 @@ import { randomBytes } from "crypto";
 import { validateAndNormalizeRefs } from "../lib/refs.js";
 import { classifyUpstreamError } from "../lib/errorClassify.js";
 import { normalizeOAuthParams } from "../lib/oauthNormalize.js";
-import { normalizeImageModel } from "../lib/imageModels.js";
+import { normalizeImageModel, normalizeReasoningEffort } from "../lib/imageModels.js";
 import { generateViaOAuth } from "../lib/oauthProxy.js";
 import { isNonRetryableGenerationError, normalizeGenerationFailure } from "../lib/generationErrors.js";
 import { startJob, finishJob } from "../lib/inflight.js";
-import { getStyleSheet } from "../lib/sessionStore.js";
-import { renderStyleSheetPrefix } from "../lib/styleSheet.js";
 import { logEvent, logError } from "../lib/logger.js";
 import { embedImageMetadataBestEffort } from "../lib/imageMetadataStore.js";
 
@@ -41,6 +39,8 @@ export function registerGenerateRoutes(app, ctx) {
         references = [],
         mode: promptMode = "auto",
         model: rawModel,
+        reasoningEffort: rawReasoningEffort,
+        webSearchEnabled: rawWebSearchEnabled = true,
       } = req.body;
       const { quality, warnings: qualityWarnings } = normalizeOAuthParams({ provider, quality: rawQuality });
       const modelCheck = normalizeImageModel(ctx, rawModel);
@@ -51,6 +51,15 @@ export function registerGenerateRoutes(app, ctx) {
         return res.status(modelCheck.status).json({ error: modelCheck.error, code: modelCheck.code });
       }
       const imageModel = modelCheck.model;
+      const reasoningCheck = normalizeReasoningEffort(ctx, rawReasoningEffort);
+      if (reasoningCheck.error) {
+        finishStatus = "error";
+        finishHttpStatus = reasoningCheck.status;
+        finishErrorCode = reasoningCheck.code;
+        return res.status(reasoningCheck.status).json({ error: reasoningCheck.error, code: reasoningCheck.code });
+      }
+      const reasoningEffort = reasoningCheck.effort;
+      const webSearchEnabled = rawWebSearchEnabled !== false;
       const normalizedPromptMode = promptMode === "direct" ? "direct" : "auto";
 
       if (!prompt) return res.status(400).json({ error: "Prompt is required" });
@@ -58,25 +67,10 @@ export function registerGenerateRoutes(app, ctx) {
       if (moderationCheck.error) return res.status(400).json({ error: moderationCheck.error });
       const count = Math.min(Math.max(parseInt(n) || 1, 1), 8);
 
-      let effectivePrompt = prompt;
-      let styleSheetApplied = null;
-      if (sessionId) {
-        try {
-          const data = getStyleSheet(sessionId);
-          if (data && data.enabled && data.styleSheet) {
-            const prefix = renderStyleSheetPrefix(data.styleSheet);
-            if (prefix) {
-              effectivePrompt = `${prefix} ${prompt}`.slice(0, 4000);
-              styleSheetApplied = data.styleSheet;
-            }
-          }
-        } catch {}
-      }
-
       startJob({
         requestId,
         kind: "classic",
-        prompt: effectivePrompt,
+        prompt,
         meta: {
           kind: "classic",
           sessionId,
@@ -86,7 +80,6 @@ export function registerGenerateRoutes(app, ctx) {
           model: imageModel,
           size,
           n: count,
-          styleSheetApplied: !!styleSheetApplied,
         },
       });
 
@@ -124,7 +117,7 @@ export function registerGenerateRoutes(app, ctx) {
         clientNodeId,
         promptChars: typeof prompt === "string" ? prompt.length : 0,
         promptMode: normalizedPromptMode,
-        styleSheetApplied: !!styleSheetApplied,
+        webSearchEnabled,
       });
       const startTime = Date.now();
 
@@ -138,7 +131,7 @@ export function registerGenerateRoutes(app, ctx) {
         for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
           try {
             const r = await generateViaOAuth(
-              effectivePrompt,
+              prompt,
               quality,
               size,
               moderation,
@@ -146,7 +139,7 @@ export function registerGenerateRoutes(app, ctx) {
               requestId,
               normalizedPromptMode,
               ctx,
-              { model: imageModel },
+              { model: imageModel, reasoningEffort, webSearchEnabled },
             );
             if (r.b64) return r;
             lastErr = new Error("Empty response (safety refusal)");
@@ -180,8 +173,6 @@ export function registerGenerateRoutes(app, ctx) {
             userPrompt: prompt,
             revisedPrompt: r.value.revisedPrompt || null,
             promptMode: normalizedPromptMode,
-            effectivePrompt: styleSheetApplied ? effectivePrompt : undefined,
-            styleSheetApplied: styleSheetApplied || undefined,
             quality,
             size,
             format,
@@ -191,6 +182,7 @@ export function registerGenerateRoutes(app, ctx) {
             createdAt: Date.now(),
             usage: r.value.usage || null,
             webSearchCalls: r.value.webSearchCalls || 0,
+            webSearchEnabled,
             refsCount: refCheck.refs.length,
           };
           const rawBuffer = Buffer.from(r.value.b64, "base64");
@@ -263,6 +255,7 @@ export function registerGenerateRoutes(app, ctx) {
         warnings: qualityWarnings,
         revisedPrompt: firstRevised,
         promptMode: normalizedPromptMode,
+        webSearchEnabled,
       };
 
       if (count === 1) {

package/routes/history.js

@@ -1,5 +1,5 @@
 import { listHistoryRows } from "../lib/historyList.js";
-import { trashAsset, restoreAsset } from "../lib/assetLifecycle.js";
+import { trashAsset, restoreAsset, deleteAssetPermanent } from "../lib/assetLifecycle.js";
 import { getSessionTitleMap } from "../lib/sessionStore.js";
 import { logError, logEvent } from "../lib/logger.js";
 import { getDb } from "../lib/db.js";
@@ -88,6 +88,16 @@ export function registerHistoryRoutes(app, ctx) {
     }
   });
 
+  app.delete("/api/history/:filename/permanent", async (req, res) => {
+    try {
+      const filename = decodeURIComponent(req.params.filename);
+      const result = await deleteAssetPermanent(ctx.rootDir, filename);
+      res.json(result);
+    } catch (err) {
+      res.status(err.status || 500).json({ error: err.message, code: err.code });
+    }
+  });
+
   app.delete("/api/history/:filename", async (req, res) => {
     try {
       const filename = decodeURIComponent(req.params.filename);

package/routes/index.js

@@ -4,20 +4,30 @@ import { registerSessionRoutes } from "./sessions.js";
 import { registerEditRoutes } from "./edit.js";
 import { registerNodeRoutes } from "./nodes.js";
 import { registerGenerateRoutes } from "./generate.js";
+import { registerMultimodeRoutes } from "./multimode.js";
 import { registerStorageRoutes } from "./storage.js";
 import { registerCardNewsRoutes } from "./cardNews.js";
 import { registerMetadataRoutes } from "./metadata.js";
 import { registerPromptRoutes } from "./prompts.js";
+import { registerPromptImportRoutes } from "./promptImport.js";
+import { registerAnnotationRoutes } from "./annotations.js";
+import { registerCanvasVersionRoutes } from "./canvasVersions.js";
+import { registerComfyRoutes } from "./comfy.js";
 
 export function configureRoutes(app, ctx) {
   registerHealthRoutes(app, ctx);
   registerStorageRoutes(app, ctx);
   registerMetadataRoutes(app, ctx);
   registerHistoryRoutes(app, ctx);
+  registerAnnotationRoutes(app, ctx);
+  registerCanvasVersionRoutes(app, ctx);
+  registerComfyRoutes(app, ctx);
   registerSessionRoutes(app, ctx);
   registerEditRoutes(app, ctx);
   registerNodeRoutes(app, ctx);
   if (ctx.config.features.cardNews) registerCardNewsRoutes(app, ctx);
+  registerMultimodeRoutes(app, ctx);
   registerGenerateRoutes(app, ctx);
   registerPromptRoutes(app, ctx);
+  registerPromptImportRoutes(app, ctx);
 }

package/routes/multimode.js

@@ -0,0 +1,281 @@
+import { mkdir, writeFile } from "fs/promises";
+import { join } from "path";
+import { randomBytes } from "crypto";
+import { validateAndNormalizeRefs } from "../lib/refs.js";
+import { classifyUpstreamError } from "../lib/errorClassify.js";
+import { normalizeOAuthParams } from "../lib/oauthNormalize.js";
+import { normalizeImageModel, normalizeReasoningEffort } from "../lib/imageModels.js";
+import { generateMultimodeViaOAuth } from "../lib/oauthProxy.js";
+import { startJob, finishJob } from "../lib/inflight.js";
+import { logEvent, logError } from "../lib/logger.js";
+import { embedImageMetadataBestEffort } from "../lib/imageMetadataStore.js";
+
+function sendSse(res, event, data) {
+  res.write(`event: ${event}\n`);
+  res.write(`data: ${JSON.stringify(data)}\n\n`);
+}
+
+function validateModeration(ctx, moderation) {
+  if (typeof moderation !== "string" || !ctx.config.oauth.validModeration.has(moderation)) {
+    return { error: "moderation must be one of: auto, low" };
+  }
+  return { moderation };
+}
+
+function normalizeMaxImages(value) {
+  return Math.min(8, Math.max(1, Math.trunc(Number(value) || 1)));
+}
+
+function sequenceStatus(returned, requested) {
+  if (returned <= 0) return "empty";
+  if (returned < requested) return "partial";
+  return "complete";
+}
+
+export function registerMultimodeRoutes(app, ctx) {
+  app.post("/api/generate/multimode", async (req, res) => {
+    const requestId = typeof req.body?.requestId === "string" ? req.body.requestId : req.id;
+    let finishStatus = "completed";
+    let finishHttpStatus = 200;
+    let finishErrorCode;
+    let finishMeta = {};
+
+    res.setHeader("Content-Type", "text/event-stream; charset=utf-8");
+    res.setHeader("Cache-Control", "no-cache, no-transform");
+    res.setHeader("Connection", "keep-alive");
+    res.flushHeaders?.();
+
+    try {
+      const {
+        prompt,
+        quality: rawQuality = "medium",
+        size = "1024x1024",
+        format = "png",
+        moderation = "low",
+        provider = "auto",
+        references = [],
+        mode: promptMode = "auto",
+        model: rawModel,
+        reasoningEffort: rawReasoningEffort,
+        webSearchEnabled: rawWebSearchEnabled = true,
+      } = req.body;
+      const maxImages = normalizeMaxImages(req.body?.maxImages);
+      const normalizedPromptMode = promptMode === "direct" ? "direct" : "auto";
+      const { quality, warnings: qualityWarnings } = normalizeOAuthParams({ provider, quality: rawQuality });
+      const modelCheck = normalizeImageModel(ctx, rawModel);
+      if (modelCheck.error) {
+        finishStatus = "error";
+        finishHttpStatus = modelCheck.status;
+        finishErrorCode = modelCheck.code;
+        sendSse(res, "error", { error: modelCheck.error, code: modelCheck.code, status: modelCheck.status, requestId });
+        return;
+      }
+      const imageModel = modelCheck.model;
+      const reasoningCheck = normalizeReasoningEffort(ctx, rawReasoningEffort);
+      if (reasoningCheck.error) {
+        finishStatus = "error";
+        finishHttpStatus = reasoningCheck.status;
+        finishErrorCode = reasoningCheck.code;
+        sendSse(res, "error", { error: reasoningCheck.error, code: reasoningCheck.code, status: reasoningCheck.status, requestId });
+        return;
+      }
+      const reasoningEffort = reasoningCheck.effort;
+      const webSearchEnabled = rawWebSearchEnabled !== false;
+      if (!prompt) {
+        finishStatus = "error";
+        finishHttpStatus = 400;
+        finishErrorCode = "PROMPT_REQUIRED";
+        sendSse(res, "error", { error: "Prompt is required", code: finishErrorCode, status: 400, requestId });
+        return;
+      }
+      const moderationCheck = validateModeration(ctx, moderation);
+      if (moderationCheck.error) {
+        finishStatus = "error";
+        finishHttpStatus = 400;
+        finishErrorCode = "INVALID_MODERATION";
+        sendSse(res, "error", { error: moderationCheck.error, code: finishErrorCode, status: 400, requestId });
+        return;
+      }
+      if (provider === "api") {
+        finishStatus = "error";
+        finishHttpStatus = 403;
+        finishErrorCode = "APIKEY_DISABLED";
+        sendSse(res, "error", {
+          error: "API key provider is disabled. Use OAuth (Codex login).",
+          code: finishErrorCode,
+          status: 403,
+          requestId,
+        });
+        return;
+      }
+
+      const refCheck = validateAndNormalizeRefs(references);
+      if (refCheck.error) {
+        finishStatus = "error";
+        finishHttpStatus = 400;
+        finishErrorCode = refCheck.code;
+        sendSse(res, "error", { error: refCheck.error, code: refCheck.code, status: 400, requestId });
+        return;
+      }
+
+      startJob({
+        requestId,
+        kind: "multimode",
+        prompt,
+        meta: { kind: "multimode", quality, model: imageModel, size, maxImages },
+      });
+
+      logEvent("multimode", "request", {
+        requestId,
+        quality,
+        model: imageModel,
+        size,
+        moderation,
+        maxImages,
+        refs: refCheck.refs.length,
+        promptChars: typeof prompt === "string" ? prompt.length : 0,
+        webSearchEnabled,
+      });
+
+      const startTime = Date.now();
+      const mimeMap = { png: "image/png", jpeg: "image/jpeg", webp: "image/webp" };
+      const mime = mimeMap[format] || "image/png";
+      const sequenceId = `seq_${Date.now().toString(36)}_${randomBytes(4).toString("hex")}`;
+      await mkdir(ctx.config.storage.generatedDir, { recursive: true });
+
+      sendSse(res, "phase", { phase: "streaming", requestId, sequenceId, maxImages });
+      const generated = await generateMultimodeViaOAuth(
+        prompt,
+        quality,
+        size,
+        moderation,
+        refCheck.refDetails || refCheck.refs,
+        requestId,
+        normalizedPromptMode,
+        ctx,
+        {
+          model: imageModel,
+          maxImages,
+          reasoningEffort,
+          webSearchEnabled,
+          onPartialImage: (partial) =>
+            sendSse(res, "partial", {
+              image: `data:${mime};base64,${partial.b64}`,
+              requestId,
+              sequenceId,
+              index: partial.index,
+            }),
+        },
+      );
+
+      const returned = generated.images.length;
+      const status = sequenceStatus(returned, maxImages);
+      const elapsed = ((Date.now() - startTime) / 1000).toFixed(1);
+      const images = [];
+
+      for (const [index, image] of generated.images.entries()) {
+        const rand = randomBytes(ctx.config.ids.generatedHexBytes).toString("hex");
+        const filename = `${Date.now()}_${rand}_multimode_${index}.${format}`;
+        const meta = {
+          kind: "multimode-image",
+          generationStrategy: "one-call-text-sequence",
+          sequenceId,
+          sequenceIndex: index + 1,
+          sequenceTotalRequested: maxImages,
+          sequenceTotalReturned: returned,
+          sequenceStatus: status,
+          stageLabel: String.fromCharCode(65 + index),
+          requestId,
+          prompt,
+          userPrompt: prompt,
+          revisedPrompt: image.revisedPrompt || null,
+          promptMode: normalizedPromptMode,
+          quality,
+          size,
+          format,
+          moderation,
+          model: imageModel,
+          provider: "oauth",
+          createdAt: Date.now(),
+          usage: generated.usage || null,
+          webSearchCalls: generated.webSearchCalls || 0,
+          webSearchEnabled,
+          refsCount: refCheck.refs.length,
+        };
+        const rawBuffer = Buffer.from(image.b64, "base64");
+        const embedded = await embedImageMetadataBestEffort(rawBuffer, format, meta, {
+          version: ctx.packageVersion,
+        });
+        await writeFile(join(ctx.config.storage.generatedDir, filename), embedded.buffer);
+        await writeFile(join(ctx.config.storage.generatedDir, filename + ".json"), JSON.stringify(meta)).catch(() => {});
+        const item = {
+          image: `data:${mime};base64,${image.b64}`,
+          filename,
+          revisedPrompt: image.revisedPrompt || null,
+          sequenceId,
+          sequenceIndex: index + 1,
+          sequenceTotalRequested: maxImages,
+          sequenceTotalReturned: returned,
+          sequenceStatus: status,
+        };
+        images.push(item);
+        sendSse(res, "image", item);
+      }
+
+      finishMeta = { sequenceId, imageCount: returned, maxImages, status };
+      finishHttpStatus = 200;
+      sendSse(res, "done", {
+        ok: true,
+        requestId,
+        sequenceId,
+        requested: maxImages,
+        returned,
+        status,
+        elapsed,
+        images,
+        provider: "oauth",
+        quality,
+        size,
+        moderation,
+        model: imageModel,
+        usage: generated.usage || null,
+        webSearchCalls: generated.webSearchCalls || 0,
+        webSearchEnabled,
+        warnings: qualityWarnings,
+        extraIgnored: generated.extraIgnored || 0,
+        promptMode: normalizedPromptMode,
+      });
+      logEvent("multimode", "saved", {
+        requestId,
+        sequenceId,
+        imageCount: returned,
+        maxImages,
+        status,
+        elapsedMs: Date.now() - startTime,
+      });
+    } catch (err) {
+      const fallbackCode = err.code || classifyUpstreamError(err.message);
+      finishStatus = "error";
+      finishHttpStatus = err.status || 500;
+      finishErrorCode = fallbackCode || "MULTIMODE_GENERATE_FAILED";
+      logError("multimode", "error", err, { requestId, code: finishErrorCode });
+      sendSse(res, "error", {
+        error: err.message,
+        code: finishErrorCode,
+        status: finishHttpStatus,
+        requestId,
+        upstreamCode: err.upstreamCode || null,
+        upstreamType: err.upstreamType || null,
+        upstreamParam: err.upstreamParam || null,
+      });
+    } finally {
+      finishJob(requestId, {
+        status: finishStatus,
+        httpStatus: finishHttpStatus,
+        errorCode: finishErrorCode,
+        meta: finishMeta,
+      });
+      res.end();
+    }
+  });
+}