ima2-gen 2.0.1 → 2.0.2

package/docs/migration/runtime-test-inventory.md

@@ -4,7 +4,7 @@ Generated by `npm run test:inventory` (script: `scripts/classify-tests.mjs`).
 
 _Tests considered "runtime-importing" if they import from `../lib/`, `../routes/`, `../bin/`, `../server`, or `../config`._
 
-Total: 177 (runtime: 61, contract: 116)
+Total: 191 (runtime: 66, contract: 125)
 
 ## Runtime-importing tests
 - `tests/agent-mode-auto-planner-contract.test.ts`
@@ -18,10 +18,14 @@ Total: 177 (runtime: 61, contract: 116)
 - `tests/api-provider-parity.test.ts`
 - `tests/billing-source.test.ts`
 - `tests/card-news-contract.test.ts`
+- `tests/card-news-template.test.ts`
+- `tests/classic-generate-async.test.ts`
 - `tests/cli-error-hints.test.ts`
 - `tests/cli-lib.test.ts`
 - `tests/comfy-bridge-contract.test.ts`
 - `tests/error-classify.test.ts`
+- `tests/event-bus.test.ts`
+- `tests/events-channel-contract.test.ts`
 - `tests/generate-route-validation-error.test.ts`
 - `tests/generated-static-privacy.test.ts`
 - `tests/generation-errors.test.ts`
@@ -36,6 +40,7 @@ Total: 177 (runtime: 61, contract: 116)
 - `tests/image-metadata-route.test.ts`
 - `tests/image-metadata-xmp.test.ts`
 - `tests/image-model.test.ts`
+- `tests/inflight-guard-contract.test.ts`
 - `tests/inflight-persistence.test.ts`
 - `tests/inflight.test.ts`
 - `tests/local-import-contract.test.ts`
@@ -75,6 +80,9 @@ Total: 177 (runtime: 61, contract: 116)
 - `tests/agent-mode-right-sidebar-contract.test.js`
 - `tests/agent-mode-tool-folding-contract.test.js`
 - `tests/app-weight-splitting-contract.test.js`
+- `tests/async-capacity-retry-behavior.test.ts`
+- `tests/async-capacity-retry-contract.test.js`
+- `tests/async-stream-subscribe-order.test.js`
 - `tests/background-cleanup-brush-rasterize.test.js`
 - `tests/background-cleanup-mask-compose.test.js`
 - `tests/bin.test.js`
@@ -123,8 +131,12 @@ Total: 177 (runtime: 61, contract: 116)
 - `tests/current-image-actions-readiness-contract.test.js`
 - `tests/direct-mode-visual-contract.test.js`
 - `tests/edit-mask-api-contract.test.js`
+- `tests/frontend-connection-state-contract.test.js`
+- `tests/frontend-sse-risk-contract.test.js`
+- `tests/gallery-hang-regression-contract.test.ts`
 - `tests/gallery-load-older-contract.test.js`
 - `tests/gallery-navigation-ux-contract.test.js`
+- `tests/gallery-selection-during-generation-contract.test.js`
 - `tests/gallery-session-scope-contract.test.js`
 - `tests/gallery-shortcuts-behavior.test.js`
 - `tests/gallery-shortcuts-visible-domain-contract.test.js`
@@ -145,6 +157,7 @@ Total: 177 (runtime: 61, contract: 116)
 - `tests/multimode-backend-contract.test.js`
 - `tests/multimode-concurrent-store-contract.test.js`
 - `tests/multimode-ui-contract.test.js`
+- `tests/node-async-eventbus-contract.test.js`
 - `tests/node-batch-contract.test.js`
 - `tests/node-child-refs-contract.test.js`
 - `tests/node-child-refs-payload.test.js`
@@ -184,5 +197,6 @@ Total: 177 (runtime: 61, contract: 116)
 - `tests/toast-stack-contract.test.js`
 - `tests/ui-error-code-contract.test.js`
 - `tests/video-continuity-ui-contract.test.js`
+- `tests/video-gallery-refresh-contract.test.ts`
 - `tests/vite-dev-port-contract.test.js`
 - `tests/web-search-toggle-contract.test.js`

package/lib/agentImageVideoGen.js

@@ -0,0 +1,261 @@
+import { randomBytes } from "node:crypto";
+import { mkdir, readFile, unlink, writeFile } from "node:fs/promises";
+import { atomicWriteJson } from "./atomicWrite.js";
+import { join } from "node:path";
+import { ulid } from "ulid";
+import { embedImageMetadataBestEffort } from "./imageMetadataStore.js";
+import { invalidateHistoryIndex } from "./historyIndex.js";
+import { logEvent } from "./logger.js";
+import { detectImageMimeFromB64 } from "./refs.js";
+import { resolveProviderOptions } from "./providerOptions.js";
+import { generateViaResponses } from "./responsesImageAdapter.js";
+import { generateViaGrok } from "./grokImageAdapter.js";
+import { generateViaAgy } from "./agyImageAdapter.js";
+import { generateVideoViaGrok } from "./grokVideoAdapter.js";
+import { parseVideoParams } from "./agentGenerationPlanner.js";
+import { appendAgentTurn, getAgentImages, getAgentSession, importAgentImage, } from "./agentStore.js";
+import { errInfo } from "./errInfo.js";
+import { forceImagePrompt, isTextOnlyResult, textOnlyError, notFound } from "./agentRuntime.js";
+export async function generateAgentImageWithRetry(ctx, sessionId, prompt, manifest, webSearchEnabled, options) {
+    let lastError = null;
+    for (let attempt = 0; attempt < 2; attempt++) {
+        try {
+            const forcedPrompt = attempt === 0 ? prompt : forceImagePrompt(prompt);
+            const result = await generateAgentImage(ctx, sessionId, forcedPrompt, manifest, webSearchEnabled, options);
+            if (result.image)
+                return result;
+        }
+        catch (error) {
+            lastError = error;
+            if (!isTextOnlyResult(error))
+                throw error;
+            if (attempt === 1)
+                break;
+            appendAgentTurn({
+                sessionId,
+                role: "tool",
+                text: "ima2.generate_image retry: text-only result rejected",
+                status: "error",
+            });
+        }
+    }
+    throw textOnlyError(lastError);
+}
+async function generateAgentImage(ctx, sessionId, prompt, manifest, webSearchEnabled, options) {
+    const requestId = options.requestId ?? `agent_${ulid()}`;
+    const providerOptions = resolveProviderOptions(ctx, {
+        provider: options.provider ?? "oauth",
+        rawModel: options.model,
+        rawReasoningEffort: options.reasoningEffort,
+        rawSize: options.size ?? "1024x1024",
+        rawWebSearchEnabled: webSearchEnabled,
+        searchMode: webSearchEnabled ? "on" : "off",
+    });
+    if (providerOptions.error) {
+        const err = new Error(providerOptions.error);
+        err.code = providerOptions.code;
+        err.status = providerOptions.status;
+        throw err;
+    }
+    const activeProvider = providerOptions.provider;
+    const effectiveModel = activeProvider === "grok" && options.quality === "high"
+        ? "grok-imagine-image-quality"
+        : providerOptions.model;
+    const response = activeProvider === "agy"
+        ? await generateViaAgy(`${manifest}\n\nUser request:\n${prompt}`, {
+            requestId,
+            signal: options.signal ?? undefined,
+        })
+        : activeProvider === "grok"
+            ? await generateViaGrok(`${manifest}\n\nUser request:\n${prompt}`, ctx, {
+                model: effectiveModel,
+                size: providerOptions.size,
+                requestId,
+                signal: options.signal ?? undefined,
+                references: await loadAgentCurrentImageReferences(ctx, sessionId),
+            })
+            : await generateViaResponses(activeProvider, `${manifest}\n\nUser request:\n${prompt}`, options.quality ?? "medium", providerOptions.size, options.moderation ?? "low", [], requestId, "auto", ctx, {
+                model: providerOptions.model,
+                reasoningEffort: providerOptions.reasoningEffort,
+                webSearchEnabled,
+                signal: options.signal,
+            });
+    const format = activeProvider === "grok" || activeProvider === "agy"
+        ? imageFormatFromMime(("mime" in response ? response.mime : undefined) || detectImageMimeFromB64(response.b64) || "image/jpeg")
+        : options.format ?? "png";
+    const image = await persistAgentImage(ctx, sessionId, prompt, format, requestId, response, {
+        provider: String(activeProvider),
+        model: String(effectiveModel),
+    });
+    const responseText = "text" in response && typeof response.text === "string" ? response.text : null;
+    return { image, webSearchCalls: response.webSearchCalls || 0, text: responseText, provider: activeProvider };
+}
+async function loadAgentCurrentImageReferences(ctx, sessionId) {
+    const session = getAgentSession(sessionId);
+    const currentImage = session?.lastImageId
+        ? getAgentImages(sessionId).find((image) => image.id === session.lastImageId)
+        : null;
+    if (!currentImage?.filename)
+        return [];
+    try {
+        const b64 = (await readFile(join(ctx.config.storage.generatedDir, currentImage.filename))).toString("base64");
+        const mime = detectImageMimeFromB64(b64);
+        return [{ b64, declaredMime: mime, detectedMime: mime }];
+    }
+    catch (error) {
+        const err = errInfo(error);
+        logEvent("agent", "grok_ref_missing", { sessionId, filename: currentImage.filename, code: err.code, message: err.message });
+        return [];
+    }
+}
+function imageFormatFromMime(mime) {
+    if (mime === "image/jpeg")
+        return "jpeg";
+    if (mime === "image/webp")
+        return "webp";
+    return "png";
+}
+async function persistAgentImage(ctx, sessionId, prompt, format, requestId, response, generation) {
+    await mkdir(ctx.config.storage.generatedDir, { recursive: true });
+    const rand = randomBytes(ctx.config.ids.generatedHexBytes).toString("hex");
+    const filename = `${Date.now()}_${rand}_agent.${format}`;
+    const meta = {
+        kind: "agent",
+        requestId,
+        sessionId,
+        prompt,
+        userPrompt: prompt,
+        revisedPrompt: response.revisedPrompt ?? null,
+        provider: generation.provider,
+        model: generation.model,
+        createdAt: Date.now(),
+        usage: response.usage ?? null,
+        webSearchCalls: response.webSearchCalls ?? 0,
+    };
+    const embedded = await embedImageMetadataBestEffort(Buffer.from(response.b64, "base64"), format, meta, {
+        version: ctx.packageVersion,
+    });
+    const filePath = join(ctx.config.storage.generatedDir, filename);
+    await writeFile(filePath, embedded.buffer);
+    try {
+        await atomicWriteJson(`${filePath}.json`, meta);
+    }
+    catch (err) {
+        await unlink(filePath).catch(() => { });
+        throw err;
+    }
+    invalidateHistoryIndex();
+    logEvent("agent", "saved", { requestId, sessionId, filename });
+    return importAgentImage(sessionId, {
+        id: `ai_${ulid()}`,
+        filename,
+        url: `/generated/${filename}`,
+        prompt,
+        revisedPrompt: response.revisedPrompt ?? null,
+        createdAt: Date.now(),
+    });
+}
+export async function runAgentVideoGeneration(ctx, sessionId, prompt, options = {}) {
+    const session = getAgentSession(sessionId);
+    if (!session)
+        throw notFound(sessionId);
+    if (!options.skipUserTurn) {
+        appendAgentTurn({ sessionId, role: "user", text: prompt, status: "complete" });
+    }
+    const requestId = options.requestId ?? `agent_video_${ulid()}`;
+    const startedAt = Date.now();
+    // Auto I2V: if session has a last image, use it as source
+    let sourceImage;
+    let mode = "text-to-video";
+    if (session.lastImageId) {
+        const images = getAgentImages(sessionId);
+        const lastImage = images.find((img) => img.id === session.lastImageId);
+        if (lastImage?.filename && !lastImage.filename.endsWith(".mp4")) {
+            try {
+                const { loadAssetB64 } = await import("./nodeStore.js");
+                sourceImage = await loadAssetB64(ctx.rootDir, lastImage.filename, ctx.config.storage.generatedDir);
+                mode = "image-to-video";
+            }
+            catch { /* fallback to T2V */ }
+        }
+    }
+    const videoParams = parseVideoParams(prompt);
+    const result = await generateVideoViaGrok(prompt, ctx, {
+        model: "grok-imagine-video",
+        mode,
+        sourceImage,
+        duration: videoParams.duration ?? 5,
+        resolution: videoParams.resolution ?? "480p",
+        aspectRatio: (videoParams.aspectRatio ?? "auto"),
+        requestId,
+        signal: options.signal ?? undefined,
+    });
+    const video = await persistAgentVideo(ctx, sessionId, prompt, requestId, result);
+    const finishedAt = Date.now();
+    const toolCall = {
+        id: `tc_video_${ulid()}`,
+        name: "ima2.generate_video",
+        status: "complete",
+        startedAt,
+        finishedAt,
+        durationMs: finishedAt - startedAt,
+        requestId,
+        inputSummary: prompt,
+        outputSummary: `Generated video ${video.filename}.`,
+        imageIds: [video.id],
+    };
+    appendAgentTurn({
+        sessionId,
+        role: "tool",
+        text: "ima2.generate_video",
+        imageIds: [video.id],
+        status: "complete",
+        raw: { toolCalls: [toolCall] },
+    });
+    const assistantTurn = appendAgentTurn({
+        sessionId,
+        role: "assistant",
+        text: `Generated 1 video artifact. ${result.revisedPrompt}`,
+        imageIds: [video.id],
+        status: "complete",
+    });
+    return { assistantTurn, imageIds: [video.id], webFindingIds: [] };
+}
+async function persistAgentVideo(ctx, sessionId, prompt, requestId, result) {
+    await mkdir(ctx.config.storage.generatedDir, { recursive: true });
+    const rand = randomBytes(ctx.config.ids.generatedHexBytes).toString("hex");
+    const filename = `${Date.now()}_${rand}_agent.mp4`;
+    const meta = {
+        kind: "agent",
+        mediaType: "video",
+        requestId,
+        sessionId,
+        prompt,
+        userPrompt: prompt,
+        revisedPrompt: result.revisedPrompt,
+        provider: "grok",
+        model: "grok-imagine-video",
+        createdAt: Date.now(),
+        usage: result.usage,
+        webSearchCalls: result.webSearchCalls,
+    };
+    const filePath = join(ctx.config.storage.generatedDir, filename);
+    await writeFile(filePath, result.videoBuffer);
+    try {
+        await atomicWriteJson(`${filePath}.json`, meta);
+    }
+    catch (err) {
+        await unlink(filePath).catch(() => { });
+        throw err;
+    }
+    invalidateHistoryIndex();
+    logEvent("agent", "video_saved", { requestId, sessionId, filename });
+    return importAgentImage(sessionId, {
+        id: `ai_${ulid()}`,
+        filename,
+        url: `/generated/${filename}`,
+        prompt,
+        revisedPrompt: result.revisedPrompt,
+        createdAt: Date.now(),
+    });
+}

package/lib/agentRuntime.js

@@ -1,19 +1,7 @@
-import { randomBytes } from "node:crypto";
-import { mkdir, readFile, unlink, writeFile } from "node:fs/promises";
-import { atomicWriteJson } from "./atomicWrite.js";
-import { join } from "node:path";
 import { ulid } from "ulid";
-import { embedImageMetadataBestEffort } from "./imageMetadataStore.js";
-import { invalidateHistoryIndex } from "./historyIndex.js";
-import { logEvent } from "./logger.js";
-import { detectImageMimeFromB64 } from "./refs.js";
-import { resolveProviderOptions } from "./providerOptions.js";
-import { generateViaResponses } from "./responsesImageAdapter.js";
-import { generateViaGrok } from "./grokImageAdapter.js";
-import { generateViaAgy } from "./agyImageAdapter.js";
-import { generateVideoViaGrok } from "./grokVideoAdapter.js";
-import { parseVideoParams } from "./agentGenerationPlanner.js";
-import { appendAgentTurn, buildImageContextManifest, getAgentImages, getAgentSession, importAgentImage, recordAgentWebFinding, restartAgentRuntimeSession, } from "./agentStore.js";
+import { generateAgentImageWithRetry } from "./agentImageVideoGen.js";
+import { runAgentVideoGeneration } from "./agentImageVideoGen.js";
+import { appendAgentTurn, buildImageContextManifest, getAgentSession, recordAgentWebFinding, restartAgentRuntimeSession, } from "./agentStore.js";
 import { AGENT_ALLOWED_TOOLS, } from "./agentTypes.js";
 import { errInfo } from "./errInfo.js";
 export function assertAgentAllowedTools(tools) {
@@ -190,249 +178,6 @@ export function isRuntimeRestartableError(error) {
         code.includes("PROTOCOL") ||
         err.message.toLowerCase().includes("protocol wedge"));
 }
-async function generateAgentImageWithRetry(ctx, sessionId, prompt, manifest, webSearchEnabled, options) {
-    let lastError = null;
-    for (let attempt = 0; attempt < 2; attempt++) {
-        try {
-            const forcedPrompt = attempt === 0 ? prompt : forceImagePrompt(prompt);
-            const result = await generateAgentImage(ctx, sessionId, forcedPrompt, manifest, webSearchEnabled, options);
-            if (result.image)
-                return result;
-        }
-        catch (error) {
-            lastError = error;
-            if (!isTextOnlyResult(error))
-                throw error;
-            if (attempt === 1)
-                break;
-            appendAgentTurn({
-                sessionId,
-                role: "tool",
-                text: "ima2.generate_image retry: text-only result rejected",
-                status: "error",
-            });
-        }
-    }
-    throw textOnlyError(lastError);
-}
-async function generateAgentImage(ctx, sessionId, prompt, manifest, webSearchEnabled, options) {
-    const requestId = options.requestId ?? `agent_${ulid()}`;
-    const providerOptions = resolveProviderOptions(ctx, {
-        provider: options.provider ?? "oauth",
-        rawModel: options.model,
-        rawReasoningEffort: options.reasoningEffort,
-        rawSize: options.size ?? "1024x1024",
-        rawWebSearchEnabled: webSearchEnabled,
-        searchMode: webSearchEnabled ? "on" : "off",
-    });
-    if (providerOptions.error) {
-        const err = new Error(providerOptions.error);
-        err.code = providerOptions.code;
-        err.status = providerOptions.status;
-        throw err;
-    }
-    const activeProvider = providerOptions.provider;
-    const effectiveModel = activeProvider === "grok" && options.quality === "high"
-        ? "grok-imagine-image-quality"
-        : providerOptions.model;
-    const response = activeProvider === "agy"
-        ? await generateViaAgy(`${manifest}\n\nUser request:\n${prompt}`, {
-            requestId,
-            signal: options.signal ?? undefined,
-        })
-        : activeProvider === "grok"
-            ? await generateViaGrok(`${manifest}\n\nUser request:\n${prompt}`, ctx, {
-                model: effectiveModel,
-                size: providerOptions.size,
-                requestId,
-                signal: options.signal ?? undefined,
-                references: await loadAgentCurrentImageReferences(ctx, sessionId),
-            })
-            : await generateViaResponses(activeProvider, `${manifest}\n\nUser request:\n${prompt}`, options.quality ?? "medium", providerOptions.size, options.moderation ?? "low", [], requestId, "auto", ctx, {
-                model: providerOptions.model,
-                reasoningEffort: providerOptions.reasoningEffort,
-                webSearchEnabled,
-                signal: options.signal,
-            });
-    const format = activeProvider === "grok" || activeProvider === "agy"
-        ? imageFormatFromMime(("mime" in response ? response.mime : undefined) || detectImageMimeFromB64(response.b64) || "image/jpeg")
-        : options.format ?? "png";
-    const image = await persistAgentImage(ctx, sessionId, prompt, format, requestId, response, {
-        provider: String(activeProvider),
-        model: String(effectiveModel),
-    });
-    const responseText = "text" in response && typeof response.text === "string" ? response.text : null;
-    return { image, webSearchCalls: response.webSearchCalls || 0, text: responseText, provider: activeProvider };
-}
-async function loadAgentCurrentImageReferences(ctx, sessionId) {
-    const session = getAgentSession(sessionId);
-    const currentImage = session?.lastImageId
-        ? getAgentImages(sessionId).find((image) => image.id === session.lastImageId)
-        : null;
-    if (!currentImage?.filename)
-        return [];
-    try {
-        const b64 = (await readFile(join(ctx.config.storage.generatedDir, currentImage.filename))).toString("base64");
-        const mime = detectImageMimeFromB64(b64);
-        return [{ b64, declaredMime: mime, detectedMime: mime }];
-    }
-    catch (error) {
-        const err = errInfo(error);
-        logEvent("agent", "grok_ref_missing", { sessionId, filename: currentImage.filename, code: err.code, message: err.message });
-        return [];
-    }
-}
-function imageFormatFromMime(mime) {
-    if (mime === "image/jpeg")
-        return "jpeg";
-    if (mime === "image/webp")
-        return "webp";
-    return "png";
-}
-async function persistAgentImage(ctx, sessionId, prompt, format, requestId, response, generation) {
-    await mkdir(ctx.config.storage.generatedDir, { recursive: true });
-    const rand = randomBytes(ctx.config.ids.generatedHexBytes).toString("hex");
-    const filename = `${Date.now()}_${rand}_agent.${format}`;
-    const meta = {
-        kind: "agent",
-        requestId,
-        sessionId,
-        prompt,
-        userPrompt: prompt,
-        revisedPrompt: response.revisedPrompt ?? null,
-        provider: generation.provider,
-        model: generation.model,
-        createdAt: Date.now(),
-        usage: response.usage ?? null,
-        webSearchCalls: response.webSearchCalls ?? 0,
-    };
-    const embedded = await embedImageMetadataBestEffort(Buffer.from(response.b64, "base64"), format, meta, {
-        version: ctx.packageVersion,
-    });
-    const filePath = join(ctx.config.storage.generatedDir, filename);
-    await writeFile(filePath, embedded.buffer);
-    try {
-        await atomicWriteJson(`${filePath}.json`, meta);
-    }
-    catch (err) {
-        await unlink(filePath).catch(() => { });
-        throw err;
-    }
-    invalidateHistoryIndex();
-    logEvent("agent", "saved", { requestId, sessionId, filename });
-    return importAgentImage(sessionId, {
-        id: `ai_${ulid()}`,
-        filename,
-        url: `/generated/${filename}`,
-        prompt,
-        revisedPrompt: response.revisedPrompt ?? null,
-        createdAt: Date.now(),
-    });
-}
-export async function runAgentVideoGeneration(ctx, sessionId, prompt, options = {}) {
-    const session = getAgentSession(sessionId);
-    if (!session)
-        throw notFound(sessionId);
-    if (!options.skipUserTurn) {
-        appendAgentTurn({ sessionId, role: "user", text: prompt, status: "complete" });
-    }
-    const requestId = options.requestId ?? `agent_video_${ulid()}`;
-    const startedAt = Date.now();
-    // Auto I2V: if session has a last image, use it as source
-    let sourceImage;
-    let mode = "text-to-video";
-    if (session.lastImageId) {
-        const images = getAgentImages(sessionId);
-        const lastImage = images.find((img) => img.id === session.lastImageId);
-        if (lastImage?.filename && !lastImage.filename.endsWith(".mp4")) {
-            try {
-                const { loadAssetB64 } = await import("./nodeStore.js");
-                sourceImage = await loadAssetB64(ctx.rootDir, lastImage.filename, ctx.config.storage.generatedDir);
-                mode = "image-to-video";
-            }
-            catch { /* fallback to T2V */ }
-        }
-    }
-    const videoParams = parseVideoParams(prompt);
-    const result = await generateVideoViaGrok(prompt, ctx, {
-        model: "grok-imagine-video",
-        mode,
-        sourceImage,
-        duration: videoParams.duration ?? 5,
-        resolution: videoParams.resolution ?? "480p",
-        aspectRatio: (videoParams.aspectRatio ?? "auto"),
-        requestId,
-        signal: options.signal ?? undefined,
-    });
-    const video = await persistAgentVideo(ctx, sessionId, prompt, requestId, result);
-    const finishedAt = Date.now();
-    const toolCall = {
-        id: `tc_video_${ulid()}`,
-        name: "ima2.generate_video",
-        status: "complete",
-        startedAt,
-        finishedAt,
-        durationMs: finishedAt - startedAt,
-        requestId,
-        inputSummary: prompt,
-        outputSummary: `Generated video ${video.filename}.`,
-        imageIds: [video.id],
-    };
-    appendAgentTurn({
-        sessionId,
-        role: "tool",
-        text: "ima2.generate_video",
-        imageIds: [video.id],
-        status: "complete",
-        raw: { toolCalls: [toolCall] },
-    });
-    const assistantTurn = appendAgentTurn({
-        sessionId,
-        role: "assistant",
-        text: `Generated 1 video artifact. ${result.revisedPrompt}`,
-        imageIds: [video.id],
-        status: "complete",
-    });
-    return { assistantTurn, imageIds: [video.id], webFindingIds: [] };
-}
-async function persistAgentVideo(ctx, sessionId, prompt, requestId, result) {
-    await mkdir(ctx.config.storage.generatedDir, { recursive: true });
-    const rand = randomBytes(ctx.config.ids.generatedHexBytes).toString("hex");
-    const filename = `${Date.now()}_${rand}_agent.mp4`;
-    const meta = {
-        kind: "agent",
-        mediaType: "video",
-        requestId,
-        sessionId,
-        prompt,
-        userPrompt: prompt,
-        revisedPrompt: result.revisedPrompt,
-        provider: "grok",
-        model: "grok-imagine-video",
-        createdAt: Date.now(),
-        usage: result.usage,
-        webSearchCalls: result.webSearchCalls,
-    };
-    const filePath = join(ctx.config.storage.generatedDir, filename);
-    await writeFile(filePath, result.videoBuffer);
-    try {
-        await atomicWriteJson(`${filePath}.json`, meta);
-    }
-    catch (err) {
-        await unlink(filePath).catch(() => { });
-        throw err;
-    }
-    invalidateHistoryIndex();
-    logEvent("agent", "video_saved", { requestId, sessionId, filename });
-    return importAgentImage(sessionId, {
-        id: `ai_${ulid()}`,
-        filename,
-        url: `/generated/${filename}`,
-        prompt,
-        revisedPrompt: result.revisedPrompt,
-        createdAt: Date.now(),
-    });
-}
 function recordSearchFindings(sessionId, prompt, count, provider) {
     if (!count)
         return [];
@@ -446,14 +191,14 @@ function recordSearchFindings(sessionId, prompt, count, provider) {
         }),
     ];
 }
-function forceImagePrompt(prompt) {
+export function forceImagePrompt(prompt) {
     return [
         "The previous turn did not return an image artifact.",
         "Return a final image using ima2.generate_image/image_generation now.",
         `User request: ${prompt}`,
     ].join("\n");
 }
-function isTextOnlyResult(error) {
+export function isTextOnlyResult(error) {
     const err = errInfo(error);
     return [
         "EMPTY_RESPONSE",
@@ -462,7 +207,7 @@ function isTextOnlyResult(error) {
         "IMAGE_TOOL_COMPLETED_WITHOUT_RESULT",
     ].includes(err.code || "") || err.message.includes("No image data");
 }
-function textOnlyError(cause) {
+export function textOnlyError(cause) {
     const err = new Error("Agent result did not include an image artifact.");
     err.code = "AGENT_TEXT_ONLY_RESULT";
     err.status = 422;
@@ -488,7 +233,7 @@ function cleanParallelism(value) {
         return 2;
     return Math.max(1, Math.min(8, Math.round(numeric)));
 }
-function notFound(sessionId) {
+export function notFound(sessionId) {
     const err = new Error(`Agent session not found: ${sessionId}`);
     err.code = "AGENT_SESSION_NOT_FOUND";
     err.status = 404;